一种模型训练方法及装置与流程
本申请涉及人工智能领域，尤其涉及一种模型训练方法及装置。
背景技术：
人工智能(artificialintelligence，ai)是利用数字计算机或者数字计算机控制的机器模拟、延伸和扩展人的智能，感知环境、获取知识并使用知识获得最佳结果的理论、方法、技术及应用系统。换句话说，人工智能是计算机科学的一个分支，它企图了解智能的实质，并生产出一种新的能以人类智能相似的方式作出反应的智能机器。人工智能也就是研究各种智能机器的设计原理与实现方法，使机器具有感知、推理与决策的功能。
联合学习(federatedlearning)系统基于从大量用户与其设备(例如，智能电话等)交互生成的数据来训练机器学习模型，而不需要从设备中取出数据。例如，每个循环都选择在线设备的子集，并且机器学习模型的当前版本被发送到那些选定的设备。那些选定的设备中的每一个被赋予使用它们自己本地生成并本地存储的数据来计算模型的更新的任务。然后，模型更新被发送回服务器、进行平均、并应用于服务器的模型，以便为用户的下一次迭代(例如，设备的下一个子集)产生模型的新版本。
联邦学习分为模型下发和模型上传两个步骤，中心节点将模型通过网络下发至终端设备；各终端设备利用本地数据计算模型的梯度；各分布式节点将梯度加密后上传至中心节点；中心节点汇总各终端分布式节点的梯度，并采用参数平均算法更新中心节点模型的参数。
现有的联邦学习技术要求中心节点和各分布式节点的模型同构，但是在一些场景中，终端设备如手机等等因为数据处理能力的限制，无法运算与中心节点同等规模的大模型。
技术实现要素：
第一方面，本申请提供了一种模型训练方法，所述方法包括：
向第一设备发送第一子模型；其中，所述第一子模型为对待训练模型进行压缩得到的；
接收所述第一设备发送的第一梯度，所述第一梯度为所述第一设备训练所述第一子模型时得到的；
至少基于所述第一梯度，对所述待训练模型进行模型训练，以得到更新后的待训练模型。
本申请实施例提供了一种模型训练方法，包括：向第一设备发送第一子模型；其中，所述第一子模型为对待训练模型进行压缩得到的；接收所述第一设备发送的第一梯度，所述第一梯度为所述第一设备训练所述第一子模型时得到的；至少基于所述第一梯度，对所述待训练模型进行模型训练，以得到更新后的待训练模型。通过上述方式，服务器将待训练模型进行压缩后下发至终端设备，使得终端设备不需要训练和服务器同等规模的大模型。
在一种可能的实现中，存储所述第一子模型所需的存储资源小于存储所述待训练模型所需的存储资源。
在一种可能的实现中，运行所述第一子模型时所需的计算资源小于运行所述待训练模型时所需的计算资源。
在一种可能的实现中，所述计算资源至少包括如下的一种：内存资源、显存资源或cpu资源。
在一种可能的实现中，所述第一子模型为对待训练模型进行如下至少一种处理方式得到的：剪枝、知识蒸馏或量化。
在一种可能的实现中，所述至少基于所述第一梯度，对所述待训练模型进行模型训练，包括：
基于所述第一梯度，对所述第一子模型进行训练，得到更新后的第一子模型；
至少以所述更新后的第一子模型作为老师模型，所述待训练模型为学生模型，对所述待训练模型进行知识蒸馏，以得到更新后的待训练模型。
在一种可能的实现中，所述方法还包括：
获取所述第一设备的数据处理能力，所述数据处理能力包括以下至少一种：可使用的存储资源或可使用的计算资源；
基于第一映射关系，获取所述第一设备对应的第一子模型；其中，所述第一映射关系包括数据处理能力与子模型之间的对应关系，且在所述第一映射关系中，所述第一设备的数据处理能力对应于所述第一子模型。
在一种可能的实现中，所述方法还包括：
向第二设备发送第二子模型；其中，所述第二子模型为对待训练模型进行压缩得到的，且所述第二子模型与所述第一子模型不同；
接收所述第二设备发送的第二梯度，所述第二梯度为所述第二设备训练所述第二子模型时得到的；
所述至少基于所述第一梯度，对所述待训练模型进行模型训练，包括：
基于所述第一梯度和所述第二梯度，对所述待训练模型进行模型训练，以得到更新后的待训练模型。
在一种可能的实现中，所述第一设备与所述第二设备的数据处理能力不同，所述数据处理能力包括以下至少一种：可使用的存储资源或可使用的计算资源。
在一种可能的实现中，存储所述第一子模型和所述第二子模型所需的存储资源大小不同。
在一种可能的实现中，运行所述第一子模型和所述第二子模型时所需的计算资源大小不同。
在一种可能的实现中，所述第二子模型为对待训练模型进行如下至少一种处理方式得到的：剪枝、知识蒸馏或量化。
在一种可能的实现中，所述基于所述第一梯度和所述第二梯度，对所述待训练模型进行模型训练，包括：
基于所述第一梯度，对所述第一子模型进行训练，得到更新后的第一子模型；
基于所述第二梯度，对所述第二子模型进行训练，得到更新后的第二子模型；
以所述更新后的第一子模型和所述更新后的第二子模型作为老师模型，所述待训练模型为学生模型，对所述待训练模型进行知识蒸馏，以得到更新后的待训练模型。
第二方面，本申请提供了一种模型训练装置，所述装置包括：
发送模块，用于向第一设备发送第一子模型；其中，所述第一子模型为对待训练模型进行压缩得到的；
接收模块，用于接收所述第一设备发送的第一梯度，所述第一梯度为所述第一设备训练所述第一子模型时得到的；
模型训练模块，用于至少基于所述第一梯度，对所述待训练模型进行模型训练，以得到更新后的待训练模型。
在一种可能的实现中，存储所述第一子模型所需的存储资源小于存储所述待训练模型所需的存储资源。
在一种可能的实现中，运行所述第一子模型时所需的计算资源小于运行所述待训练模型时所需的计算资源。
在一种可能的实现中，所述计算资源至少包括如下的一种：内存资源、显存资源或cpu资源。
在一种可能的实现中，所述第一子模型为对待训练模型进行如下至少一种处理方式得到的：剪枝、知识蒸馏或量化。
在一种可能的实现中，所述模型训练模块，用于基于所述第一梯度，对所述第一子模型进行训练，得到更新后的第一子模型；至少以所述更新后的第一子模型作为老师模型，所述待训练模型为学生模型，对所述待训练模型进行知识蒸馏，以得到更新后的待训练模型。
在一种可能的实现中，所述装置还包括：
获取模块，用于获取所述第一设备的数据处理能力，所述数据处理能力包括以下至少一种：可使用的存储资源或可使用的计算资源；基于第一映射关系，获取所述第一设备对应的第一子模型；其中，所述第一映射关系包括数据处理能力与子模型之间的对应关系，且在所述第一映射关系中，所述第一设备的数据处理能力对应于所述第一子模型。
在一种可能的实现中，所述发送模块，用于向第二设备发送第二子模型；其中，所述第二子模型为对待训练模型进行压缩得到的，且所述第二子模型与所述第一子模型不同；
所述接收模块，用于接收所述第二设备发送的第二梯度，所述第二梯度为所述第二设备训练所述第二子模型时得到的；
所述模型训练模块，用于基于所述第一梯度和所述第二梯度，对所述待训练模型进行模型训练，以得到更新后的待训练模型。
在一种可能的实现中，所述第一设备与所述第二设备的数据处理能力不同，所述数据处理能力包括以下至少一种：可使用的存储资源或可使用的计算资源。
在一种可能的实现中，存储所述第一子模型和所述第二子模型所需的存储资源大小不同。
在一种可能的实现中，运行所述第一子模型和所述第二子模型时所需的计算资源大小不同。
在一种可能的实现中，所述第二子模型为对待训练模型进行如下至少一种处理方式得到的：剪枝、知识蒸馏或量化。
在一种可能的实现中，所述模型训练模块，用于基于所述第一梯度，对所述第一子模型进行训练，得到更新后的第一子模型；基于所述第二梯度，对所述第二子模型进行训练，得到更新后的第二子模型；以所述更新后的第一子模型和所述更新后的第二子模型作为老师模型，所述待训练模型为学生模型，对所述待训练模型进行知识蒸馏，以得到更新后的待训练模型。
第三方面，本申请提供了一种系统，包括第一设备和服务器；其中，
所述服务器，用于向所述第一设备发送第一子模型；其中，所述第一子模型为对待训练模型进行压缩得到的；
所述第一设备，用于训练所述第一子模型，以得到第一梯度；向所述服务器发送所述第一梯度；
所述服务器，用于至少基于所述第一梯度，对所述待训练模型进行模型训练，以得到更新后的待训练模型。
在一种可能的实现中，存储所述第一子模型所需的存储资源小于存储所述待训练模型所需的存储资源。
在一种可能的实现中，运行所述第一子模型时所需的计算资源小于运行所述待训练模型时所需的计算资源。
在一种可能的实现中，所述计算资源至少包括如下的一种：内存资源、显存资源或cpu资源。
在一种可能的实现中，所述第一子模型为对待训练模型进行如下至少一种处理方式得到的：剪枝、知识蒸馏或量化。
在一种可能的实现中，所述服务器，用于基于所述第一梯度，对所述第一子模型进行训练，得到更新后的第一子模型；
至少以所述更新后的第一子模型作为老师模型，所述待训练模型为学生模型，对所述待训练模型进行知识蒸馏，以得到更新后的待训练模型。
在一种可能的实现中，所述服务器，用于获取所述第一设备的数据处理能力，所述数据处理能力包括以下至少一种：可使用的存储资源或可使用的计算资源；
基于第一映射关系，获取所述第一设备对应的第一子模型；其中，所述第一映射关系包括数据处理能力与子模型之间的对应关系，且在所述第一映射关系中，所述第一设备的数据处理能力对应于所述第一子模型。
在一种可能的实现中，所述系统还包括：第二设备，所述服务器，用于向第二设备发送第二子模型；其中，所述第二子模型为对待训练模型进行压缩得到的，且所述第二子模型与所述第一子模型不同；
所述第二设备用于训练所述第二子模型，以得到第二梯度；向所述服务器发送所述第二梯度；
所述服务器，用于所述至少基于所述第一梯度，对所述待训练模型进行模型训练，包括：
基于所述第一梯度和所述第二梯度，对所述待训练模型进行模型训练，以得到更新后的待训练模型。
在一种可能的实现中，所述第一设备与所述第二设备的数据处理能力不同，所述数据处理能力包括以下至少一种：可使用的存储资源或可使用的计算资源不同。
在一种可能的实现中，存储所述第一子模型和所述第二子模型所需的存储资源大小不同。
在一种可能的实现中，运行所述第一子模型和所述第二子模型时所需的计算资源大小不同。
在一种可能的实现中，所述第二子模型为对待训练模型进行如下至少一种处理方式得到的：剪枝、知识蒸馏或量化。
在一种可能的实现中，所述服务器，用于基于所述第一梯度，对所述第一子模型进行训练，得到更新后的第一子模型；
基于所述第二梯度，对所述第二子模型进行训练，得到更新后的第二子模型；
以所述更新后的第一子模型和所述更新后的第二子模型作为老师模型，所述待训练模型为学生模型，对所述待训练模型进行知识蒸馏，以得到更新后的待训练模型。
第四方面，本申请实施例提供了一种数据处理装置，可以包括存储器、处理器以及总线系统，其中，存储器用于存储程序，处理器用于执行存储器中的程序，以执行如上述第一方面及第一方面任一可选的方法。
第五方面，本申请实施例提供了一种计算机可读存储介质，所述计算机可读存储介质中存储有计算机程序，当其在计算机上运行时，使得计算机执行上述第一方面或其任一可选的方法。
第六方面，本申请实施例提供了一种计算机程序，当其在计算机上运行时，使得计算机执行上述第一方面及其任一可选的方法。
第七方面，本申请提供了一种芯片系统，该芯片系统包括处理器，用于支持执行设备实现上述方面中所涉及的功能，例如，发送或处理上述方法中所涉及的数据；或，信息。在一种可能的设计中，所述芯片系统还包括存储器，所述存储器，用于保存执行设备或训练设备必要的程序指令和数据。该芯片系统，可以由芯片构成，也可以包括芯片和其他分立器件。
本申请实施例提供了一种模型训练方法，包括：向第一设备发送第一子模型；其中，所述第一子模型为对待训练模型进行压缩得到的；接收所述第一设备发送的第一梯度，所述第一梯度为所述第一设备训练所述第一子模型时得到的；至少基于所述第一梯度，对所述待训练模型进行模型训练，以得到更新后的待训练模型。通过上述方式，服务器将待训练模型进行压缩后下发至终端设备，使得终端设备不需要训练和服务器同等规模的大模型。
附图说明
图1为人工智能主体框架的一种结构示意图；
图2为本申请实施例中执行模型训练的计算系统的示意；
图3是本申请实施例提供的一种系统架构的示意图；
图4为本申请实施例提供的一种模型训练方法的架构示意；
图5为本申请实施例提供的一种模型训练方法的流程示意；
图6和图7为本申请实施例提供的一种模型训练方法的示例；
图8为本申请实施例提供的一种数据处理装置的示意；
图9为本申请实施例提供的一种系统的示意；
图10为本申请实施例提供的执行设备的一种结构示意图；
图11是本申请实施例提供的训练设备一种结构示意图；
图12为本申请实施例提供的芯片的一种结构示意图。
具体实施方式
下面结合本发明实施例中的附图对本发明实施例进行描述。本发明的实施方式部分使用的术语仅用于对本发明的具体实施例进行解释，而非旨在限定本发明。
下面结合附图，对本申请的实施例进行描述。本领域普通技术人员可知，随着技术的发展和新场景的出现，本申请实施例提供的技术方案对于类似的技术问题，同样适用。
本申请的说明书和权利要求书及上述附图中的术语“第一”、“第二”等是用于区别类似的对象，而不必用于描述特定的顺序或先后次序。应该理解这样使用的术语在适当情况下可以互换，这仅仅是描述本申请的实施例中对相同属性的对象在描述时所采用的区分方式。此外，术语“包括”和“具有”以及他们的任何变形，意图在于覆盖不排他的包含，以便包含一系列单元的过程、方法、系统、产品或设备不必限于那些单元，而是可包括没有清楚地列出的或对于这些过程、方法、产品或设备固有的其它单元。
首先对人工智能系统总体工作流程进行描述，请参见图1，图1示出的为人工智能主体框架的一种结构示意图，下面从“智能信息链”(水平轴)和“it价值链”(垂直轴)两个维度对上述人工智能主题框架进行阐述。其中，“智能信息链”反映从数据的获取到处理的一列过程。举例来说，可以是智能信息感知、智能信息表示与形成、智能推理、智能决策、智能执行与输出的一般过程。在这个过程中，数据经历了“数据—信息—知识—智慧”的凝练过程。“it价值链”从人智能的底层基础设施、信息(提供和处理技术实现)到系统的产业生态过程，反映人工智能为信息技术产业带来的价值。
(1)基础设施
基础设施为人工智能系统提供计算能力支持，实现与外部世界的沟通，并通过基础平台实现支撑。通过传感器与外部沟通；计算能力由智能芯片(cpu、npu、gpu、asic、fpga等硬件加速芯片)提供；基础平台包括分布式计算框架及网络等相关的平台保障和支持，可以包括云存储和计算、互联互通网络等。举例来说，传感器和外部沟通获取数据，这些数据提供给基础平台提供的分布式计算系统中的智能芯片进行计算。
(2)数据
基础设施的上一层的数据用于表示人工智能领域的数据来源。数据涉及到图形、图像、语音、文本，还涉及到传统设备的物联网数据，包括已有系统的业务数据以及力、位移、液位、温度、湿度等感知数据。
(3)数据处理
数据处理通常包括数据训练，机器学习，深度学习，搜索，推理，决策等方式。
其中，机器学习和深度学习可以对数据进行符号化和形式化的智能信息建模、抽取、预处理、训练等。
推理是指在计算机或智能系统中，模拟人类的智能推理方式，依据推理控制策略，利用形式化的信息进行机器思维和求解问题的过程，典型的功能是搜索与匹配。
决策是指智能信息经过推理后进行决策的过程，通常提供分类、排序、预测等功能。
(4)通用能力
对数据经过上面提到的数据处理后，进一步基于数据处理的结果可以形成一些通用的能力，比如可以是算法或者一个通用系统，例如，翻译，文本的分析，计算机视觉的处理，语音识别，图像的识别等等。
(5)智能产品及行业应用
智能产品及行业应用指人工智能系统在各领域的产品和应用，是对人工智能整体解决方案的封装，将智能信息决策产品化、实现落地应用，其应用领域主要包括：智能终端、智能交通、智能医疗、自动驾驶、平安城市等。
图2为本申请实施例中执行模型训练的计算系统的示意。计算系统包括通过网络通信地耦合的终端设备102(下文中可以称之为第一设备和第二设备)和服务器130(也可以称之为中心节点)。其中，终端设备102可以是任何类型的计算设备，诸如，例如个人计算设备(例如，膝上型计算机或台式计算机)、移动计算设备(例如，智能电话或平板计算机)、游戏控制台或控制器、可穿戴计算设备、嵌入式计算设备或任何其他类型的计算设备。
终端设备102可以包括处理器112和存储器114。处理器112可以是任何合适的处理设备(例如，处理器核、微处理器、特殊应用集成电路(applicationspecificintegratedcircuit，asic)、现场可编程门阵列(fieldprogrammablegatearray，fpga)、控制器、微控制器等)。存储器114可以包括但不限于是随机存储记忆体(randomaccessmemory，ram)、只读存储器(read-onlymemory，rom)、可擦除可编程只读存储器(erasableprogrammablereadonlymemory，eprom)、或便携式只读存储器(compactdiscread-onlymemory，cd-rom)。存储器114可以存储由处理器112执行的数据116和指令118，以使得终端设备102执行操作。
在一些实施方式中，存储器114可以存储一个或多个模型120。例如，模型120可以是或可以另外包括各种机器学习模型，诸如神经网络(例如，深层神经网络)或其他类型的机器学习模型，包括非线性模型和/或线性模型。神经网络可以包括前馈神经网络、递归神经网络(例如，长短期记忆递归神经网络)、卷积神经网络或其他形式的神经网络。
在一些实施方式中，一个或多个模型120可以通过网络180从服务器130接收，存储在存储器114中，然后由一个或多个处理器112使用或另外实施。
终端设备102还可以包括接收用户输入的一个或多个用户输入组件122。例如，用户输入组件122可以是对用户输入对象(例如，手指或触笔)的触摸敏感的触敏组件(例如，触敏显示屏或触摸板)。触敏组件可以用来实施虚拟键盘。其他示例用户输入组件包括麦克风、传统键盘或用户可以提供用户输入的其他装置。
终端设备102还可以包括通信接口123，终端设备102可以通过通信接口123和服务器130通信连接，服务器130可以包括通信接口133，终端设备102可以通过通信接口123和服务器130的通信接口133通信连接，以此实现终端设备102和服务器130之间的数据交互。
服务器130可以包括处理器132和存储器134。处理器132可以是可以是任何合适的处理设备(例如，处理器核、微处理器、特殊应用集成电路(applicationspecificintegratedcircuit，asic)、现场可编程门阵列(fieldprogrammablegatearray，fpga)、控制器、微控制器等)。存储器134可以包括但不限于是随机存储记忆体(randomaccessmemory，ram)、只读存储器(read-onlymemory，rom)、可擦除可编程只读存储器(erasableprogrammablereadonlymemory，eprom)、或便携式只读存储器(compactdiscread-onlymemory，cd-rom)。存储器134可以存储由处理器132执行的数据136和指令138，以使得服务器130执行操作。
如上所述，存储器134可以存储一个或多个机器学习模型140。例如，模型140可以是或者可以另外包括各种机器学习模型。示例机器学习模型包括神经网络或其他多层非线性模型。示例神经网络包括前馈神经网络、深层神经网络、递归神经网络和卷积神经网络。
图3是本申请实施例提供的一种系统100架构的示意图，在图3中，执行设备110配置输入/输出(input/output，i/o)接口112，用于与外部设备进行数据交互，用户可以通过客户设备140向i/o接口112输入数据，所述输入数据在本申请实施例中可以包括：各个待调度任务，可调用资源以及其他参数。
在执行设备110对输入数据进行预处理，或者在执行设备110的计算模块111执行计算等相关的处理(比如进行本申请中神经网络的功能实现)过程中，执行设备110可以调用数据存储系统150中的数据、代码等以用于相应的处理，也可以将相应处理得到的数据、指令等存入数据存储系统150中。
最后，i/o接口112将处理结果返回给客户设备140，从而提供给用户。
值得说明的是，训练设备120可以针对不同的目标或称不同的任务，基于不同的训练数据生成相应的目标模型/规则，该相应的目标模型/规则即可以用于实现上述目标或完成上述任务，从而为用户提供所需的结果。
在图3中所示情况下，用户可以手动给定输入数据，该手动给定可以通过i/o接口112提供的界面进行操作。另一种情况下，客户设备140可以自动地向i/o接口112发送输入数据，如果要求客户设备140自动发送输入数据需要获得用户的授权，则用户可以在客户设备140中设置相应权限。用户可以在客户设备140查看执行设备110输出的结果，具体的呈现形式可以是显示、声音、动作等具体方式。客户设备140也可以作为数据采集端，采集如图所示输入i/o接口112的输入数据及输出i/o接口112的输出结果作为新的样本数据，并存入数据库130。当然，也可以不经过客户设备140进行采集，而是由i/o接口112直接将如图所示输入i/o接口112的输入数据及输出i/o接口112的输出结果，作为新的样本数据存入数据库130。
值得注意的是，图3仅是本申请实施例提供的一种系统架构的示意图，图中所示设备、器件、模块等之间的位置关系不构成任何限制，例如，在图3中，数据存储系统150相对执行设备110是外部存储器，在其它情况下，也可以将数据存储系统150置于执行设备110中。如图3所示，可以根据训练设备120训练得到神经网络。
本申请实施例还提供的一种芯片，该芯片包括神经网络处理器npu50。该芯片可以被设置在如图3所示的执行设备110中，用以完成计算模块111的计算工作。该芯片也可以被设置在如图3所示的训练设备120中，用以完成训练设备120的训练工作并输出目标模型/规则。
神经网络处理器npu40，npu作为协处理器挂载到主中央处理器(centralprocessingunit，cpu)(hostcpu)上，由主cpu分配任务。npu的核心部分为运算电路403，控制器404控制运算电路403提取存储器(权重存储器或输入存储器)中的数据并进行运算。
在一些实现中，运算电路403内部包括多个处理单元(processengine,pe)。在一些实现中，运算电路403是二维脉动阵列。运算电路403还可以是一维脉动阵列或者能够执行例如乘法和加法这样的数学运算的其它电子线路。在一些实现中，运算电路403是通用的矩阵处理器。
举例来说，假设有输入矩阵a，权重矩阵b，输出矩阵c。运算电路从权重存储器402中取矩阵b相应的数据，并缓存在运算电路中每一个pe上。运算电路从输入存储器401中取矩阵a数据与矩阵b进行矩阵运算，得到的矩阵的部分结果或最终结果，保存在累加器(accumulator)408中。
向量计算单元407可以对运算电路的输出做进一步处理，如向量乘，向量加，指数运算，对数运算，大小比较等等。例如，向量计算单元407可以用于神经网络中非卷积/非fc层的网络计算，如池化(pooling)，批归一化(batchnormalization)，局部响应归一化(localresponsenormalization)等。
在一些实现种，向量计算单元能407将经处理的输出的向量存储到统一缓存器406。例如，向量计算单元407可以将非线性函数应用到运算电路403的输出，例如累加值的向量，用以生成激活值。在一些实现中，向量计算单元407生成归一化的值、合并值，或二者均有。在一些实现中，处理过的输出的向量能够用作到运算电路403的激活输入，例如用于在神经网络中的后续层中的使用。
统一存储器406用于存放输入数据以及输出数据。
权重数据直接通过存储单元访问控制器405(directmemoryaccesscontroller，dmac)将外部存储器中的输入数据搬运到输入存储器401和/或统一存储器406、将外部存储器中的权重数据存入权重存储器402，以及将统一存储器406中的数据存入外部存储器。
总线接口单元(businterfaceunit，biu)410，用于通过总线实现主cpu、dmac和取指存储器409之间进行交互。
与控制器404连接的取指存储器(instructionfetchbuffer)409，用于存储控制器404使用的指令；
控制器404，用于调用指存储器409中缓存的指令，实现控制该运算加速器的工作过程。
一般地，统一存储器406，输入存储器401，权重存储器402以及取指存储器409均为片上(on-chip)存储器，外部存储器为该npu外部的存储器，该外部存储器可以为双倍数据率同步动态随机存储器(doubledataratesynchronousdynamicrandomaccessmemory，ddrsdram)、高带宽存储器(highbandwidthmemory，hbm)或其他可读可写的存储器。
由于本申请实施例涉及大量神经网络的应用，为了便于理解，下面先对本申请实施例涉及的相关术语及神经网络等相关概念进行介绍。
(1)神经网络
神经网络可以是由神经单元组成的，神经单元可以是指以xs和截距1为输入的运算单元，该运算单元的输出可以为：
其中，s＝1、2、……n，n为大于1的自然数，ws为xs的权重，b为神经单元的偏置。f为神经单元的激活函数(activationfunctions)，用于将非线性特性引入神经网络中，来将神经单元中的输入信号转换为输出信号。该激活函数的输出信号可以作为下一层卷积层的输入。激活函数可以是sigmoid函数。神经网络是将许多个上述单一的神经单元联结在一起形成的网络，即一个神经单元的输出可以是另一个神经单元的输入。每个神经单元的输入可以与前一层的局部接受域相连，来提取局部接受域的特征，局部接受域可以是由若干个神经单元组成的区域。
神经网络中的每一层的工作可以用数学表达式
因为希望神经网络的输出尽可能的接近真正想要预测的值，所以可以通过比较当前网络的预测值和真正想要的目标值，再根据两者之间的差异情况来更新每一层神经网络的权重向量(当然，在第一次更新之前通常会有初始化的过程，即为神经网络中的各层预先配置参数)，比如，如果网络的预测值高了，就调整权重向量让它预测低一些，不断的调整，直到神经网络能够预测出真正想要的目标值。因此，就需要预先定义“如何比较预测值和目标值之间的差异”，这便是损失函数(lossfunction)或目标函数(objectivefunction)，它们是用于衡量预测值和目标值的差异的重要方程。其中，以损失函数举例，损失函数的输出值(loss)越高表示差异越大，那么神经网络的训练就变成了尽可能缩小这个loss的过程。
(2)反向传播算法
神经网络可以采用误差反向传播(backpropagation，bp)算法在训练过程中修正初始的神经网络模型中参数的大小，使得神经网络模型的重建误差损失越来越小。具体地，前向传递输入信号直至输出会产生误差损失，通过反向传播误差损失信息来更新初始的神经网络模型中参数，从而使误差损失收敛。反向传播算法是以误差损失为主导的反向传播运动，旨在得到最优的神经网络模型的参数，例如权重矩阵。
参见图4，图4为本申请实施例提供的一种模型训练方法的架构示意，如图4所示，本申请实施例提供的架构包括：云侧中心节点，例如可以是云侧的服务器。a1、a2、…为类型为a的分布式节点，如用户持有的手机产品。b1、b2、…为类型为b的分布式节点，如用户持有的个人电脑。在经过分布式节点的管理员(如手机、电脑的用户)同意后，分布式节点的管理员自愿在隐私得到保护的情况下共享其日常使用设备的过程中产生的数据，加入到模型训练计划，设备成为架构中的分布式节点。本实施例中的系统也可以包含更多类型的分布式节点，如智能手表等等。为保护数据隐私，分布式节点不会将数据上传至中心节点，仅在本地保存数据。分布式节点通过通信网络与云服务器连接。云侧中心节点可以运行大模型，而各分布式节点受硬件能力限制只能运行小模型，且a和b可以拥有不同的数据处理能力。
参见图5，图5为本申请实施例提供的一种模型训练方法的流程示意，如图5所示，本申请实施例提供的模型训练方法包括：
501、服务器向第一设备发送第一子模型；其中，所述第一子模型为对待训练模型进行压缩得到的。
本申请实施例中，服务器(也可以称之为中心节点或云侧的训练设备)可以向第一设备发送第一子模型；其中，所述第一子模型为对待训练模型进行压缩得到的。其中，第一设备可以为终端设备，例如图4中示出的分布式节点(a1、a2、b1、b2)。
本申请实施例中，服务器可以获取待训练模型，待训练模型可以为后续联邦学习的模型训练对象，待训练模型可以包括神经网络或其他多层非线性模型，例如，神经网络可以包括前馈神经网络、深层神经网络、递归神经网络和卷积神经网络。
本申请实施例中，服务器可以对待训练模型进行压缩，得到第一子模型。具体的，所述第一子模型可以为对待训练模型进行如下至少一种处理方式得到的：剪枝、知识蒸馏或量化。其中，在神经网络的权重矩阵中，将一部分元素置为无效的值，即为剪枝操作。在一种实现方式中，可以将需要剪枝的元素设置为零。比如，将权重矩阵中绝对值小于或者等于一定门限的权重值设置为零。关于知识蒸馏以及量化，可以使用现有技术中提供的知识蒸馏以及量化方法。
其中，第一子模型和待训练模型可以满足如下特点的至少一种：存储所述第一子模型所需的存储资源小于存储所述待训练模型所需的存储资源、运行所述第一子模型时所需的计算资源小于运行所述待训练模型时所需的计算资源，其中，所述计算资源可以至少包括如下的一种：内存资源、显存资源或cpu资源。
在一种实现中，存储所述第一子模型所需的存储资源小于存储所述待训练模型所需的存储资源，存储资源可以理解为存储空间大小；例如，终端设备存储第一子模型所需的存储空间为200mb，而存储待训练模型所需的存储空间为1gb。
在一种实现中，运行所述第一子模型时所需的计算资源小于运行所述待训练模型时所需的计算资源；所述计算资源可以至少包括如下的一种：内存资源、显存资源或cpu资源；需要说明的是，所需的计算资源可以指运行模型时所需的计算资源峰值，例如内存占用峰值，显存占用峰值或cpu占用峰值。
本申请实施例中，服务器可以参考第一设备的数据处理能力来选择第一子模型，进而使得第一设备有足够的资源来运行第一子模型；在一种实现中，服务器可以获取所述第一设备的数据处理能力，所述数据处理能力包括以下至少一种：可使用的存储资源或可使用的计算资源，服务器可以基于第一映射关系，获取所述第一设备对应的第一子模型；其中，所述第一映射关系包括数据处理能力与子模型之间的对应关系，且在所述第一映射关系中，所述第一设备的数据处理能力对应于所述第一子模型。
在一种实现中，第一设备的数据处理能力可以与可使用的计算资源有关，在一种实现中，第一设备的数据处理能力可以与可使用的计算资源有关，在一种实现中，第一设备的数据处理能力可以与可使用的计算资源和可使用的计算资源有关。
具体的，第一设备的数据处理能力可以是一个参数值，该参数值可以指示设备的数据处理能力。在一种实现中，终端设备可以基于第一设备的片上系统(systemonchip，soc)型号、图形处理器(graphicsprocessingunit，gpu)型号、嵌入式神经网络处理器(neural-networkprocessingunit，npu)型号、存储空间大小等等硬件能力信息来确定第一设备的数据处理能力。可选的，服务器可以维护了一个数据处理能力的计算函数，输入为上述soc型号、gpu型号、npu型号、存储空间大小等等，输出为计算得到的数据处理能力。
本申请实施例中，服务器可以基于第一映射关系，获取所述第一设备对应的第一子模型，其中，第一映射关系可以是数据处理能力区间-模型映射表，每个数据处理能力区间对应于一种模型，例如数据处理能力的区间为0～100分，数据处理能力区间为0～10，10～20，20～30，…，90～100，每个区间分别会映射到一个模型。本申请实施例中，服务器可以获取到第一设备的数据处理能力，并从第一映射关系中确定第一设备的数据处理能力所在的区间，其中，第一设备的数据处理能力所在的区间对应于第一子模型。
应理解，上述soc型号、gpu型号、npu型号、存储空间大小等等与第一设备相关的硬件能力信息可以是第一设备上传到服务器，可选的，第一设备可以向服务器发送设备注册请求，并携带上述硬件能力信息。
类似的，服务器还可以向第二设备发送第二子模型；其中，所述第二子模型为对待训练模型进行压缩得到的，且所述第二子模型与所述第一子模型不同；具体的，存储所述第一子模型和所述第二子模型所需的存储资源大小不同和/或运行所述第一子模型和所述第二子模型时所需的计算资源大小不同。所述第二子模型为对待训练模型进行如下至少一种处理方式得到的：剪枝、知识蒸馏或量化。
本申请实施例中，服务器针对于具有不同数据处理能力的设备，可以下发不同的模型，例如，第二设备和第一设备的数据处理能力不同，所述数据处理能力包括以下至少一种：可使用的存储资源或可使用的计算资源，相应的，服务器可以向第一设备下发第一子模型，向第二设备下发第二子模型，第二子模型与所述第一子模型不同，例如存储所述第一子模型和所述第二子模型所需的存储资源大小不同和/或运行所述第一子模型和所述第二子模型时所需的计算资源大小不同。
接下来，给出一个服务器下发模型到终端设备的应用实例：
首先，用户自愿将其持有的终端设备注册到大规模预训练语言模型的训练框架，让用户知悉该训练会对其日常使用设备造成的影响，且训练过程需要严格执行用户数据隐私保护的策略。用户可以选择其设备中可以参加到训练计划中的数据来源，例如：指定其在使用一些特定应用(如微博)时产生的文本数据可参与训练计划。用户侧的终端设备(例如上述实施例中的第一设备和第二设备)可以向服务器发送设备注册请求，并携带硬件能力信息(例如上述实施例中的soc型号、gpu型号、npu型号、存储空间大小等等硬件能力信息)。可选的：如果一个用户持有多个个人设备，且多个个人设备之间数据共享，该用户可以仅将一台设备注册为分布式节点。服务器可以获取终端设备的硬件能力信息，基于终端设备的硬件能力信息将该终端设备映射到特定组，不同的终端组对应着不同的终端硬件能力，每个组对应于一个压缩后的模型(例如上述实施例中，第一设备对应于第一子模型，第二设备对应于第二子模型)，具体的，服务器可以将终端设备的id与对应的分组id建立关联关系(例如在上述实施例中的第一映射关系中建立上述终端设备的id与对应的分组id的关联关系)。
用户在终端设备的日常使用过程中产生了足够用于训练的数据后，终端设备向云侧的服务器发送模型下载请求(携带终端设备的id)，云侧服务器根据终端设备的id，索引到对应的分组id，服务器可以将该分组id对应的模型通过网络发送至终端设备。示例性的，服务器可以将提供针对中心服务器存储的原模型bert的特定大小的压缩模型tinyberti，通过网络分发给终端设备。
502、接收所述第一设备发送的第一梯度，所述第一梯度为所述第一设备训练所述第一子模型时得到的。
本申请实施例中，第一设备在接收到服务器发送的第一子模型之后，可以利用本地数据对第一子模型进行训练，并在训练过程中得到第一梯度，之后可以将第一梯度上传至服务器，可选的，为了传输过程的信息隐私保护，第一设备可以将加密后的第一梯度上传至服务器。
类似的，第二设备在接收到服务器发送的第二子模型之后，可以利用本地数据对第二子模型进行训练，并在训练过程中得到第二梯度，之后可以将第二梯度上传至服务器，可选的，为了传输过程的信息隐私保护，第二设备可以将加密后的第二梯度上传至服务器。
本申请实施例中，服务器可以接收所述第一设备发送的第一梯度。类似的，服务器还可以接收所述第二设备发送的第二梯度。
503、至少基于所述第一梯度，对所述待训练模型进行模型训练，以得到更新后的待训练模型。
本申请实施例中，服务器可以至少基于所述第一梯度，对所述待训练模型进行模型训练，以得到更新后的待训练模型。具体的，服务器可以基于所述第一梯度，对所述第一子模型进行训练，得到更新后的第一子模型；至少以所述更新后的第一子模型作为老师模型，所述待训练模型为学生模型，对所述待训练模型进行知识蒸馏，以得到更新后的待训练模型。
本申请实施例中，服务器还可以基于所述第二梯度，对所述第二子模型进行训练，得到更新后的第二子模型；以所述更新后的第一子模型和所述更新后的第二子模型作为老师模型，所述待训练模型为学生模型，对所述待训练模型进行知识蒸馏，以得到更新后的待训练模型。
本申请实施例中，服务器可以按组分别汇总每一类型的终端设备上传的梯度，采用联邦梯度平均的算法更新对应的小模型。参照图6和图7，图6和图7为本申请实施例提供的一种模型训练方法的示例，其中，服务器的模型压缩模块可以将待训练模型压缩为第一子模型和第二子模型，之后服务器可以将第一子模型下发到节点a1，节点a2和节点a3，将第二子模型下发到节点b1，节点b2和节点b3。
之后，节点a1，节点a2和节点a3上传的梯度采用联邦梯度平均的方式更新第一子模型的参数，节点b1，节点b2和节点b3上传的梯度采用联邦梯度平均的方式更新第二子模型的参数，并采用多模型融合的方式，服务器利用各个类型的小模型，融合更新中心节点大模型。如第一子模型和第二子模型融合成待训练模型。融合的方式可以是多老师蒸馏。
示例性的，服务器可以汇总多个更新的压缩模型{tinyberti_update},采用多老师蒸馏的方式，将{tinyberti_update}当做多个老师模型，蒸馏得到学生模型bert_update。因为蒸馏过程不能使用用户的端侧数据，因此采用无数据蒸馏方式(datafreedistillation)、或者采用合成的数据进行蒸馏、或者使用通用的公开数据进行蒸馏。
本申请实施例提供了一种模型训练方法，包括：向第一设备发送第一子模型；其中，所述第一子模型为对待训练模型进行压缩得到的；接收所述第一设备发送的第一梯度，所述第一梯度为所述第一设备训练所述第一子模型时得到的；至少基于所述第一梯度，对所述待训练模型进行模型训练，以得到更新后的待训练模型。通过上述方式，服务器将待训练模型进行压缩后下发至终端设备，使得终端设备不需要训练和服务器同等规模的大模型。
本申请实施例中，服务器还可以向第二设备发送第二子模型；其中，所述第二子模型为对待训练模型进行压缩得到的，且所述第二子模型与所述第一子模型不同；接收所述第二设备发送的第二梯度，所述第二梯度为所述第二设备训练所述第二子模型时得到的；并基于所述第一梯度和所述第二梯度，对所述待训练模型进行模型训练，以得到更新后的待训练模型。通过上述方式，服务器可以针对于终端设备的不同，选择不同的模型下发到终端设备。
接下来从装置的角度对本申请实施例提供的数据处理装置进行描述，参照图8，图8为本申请实施例提供的一种数据处理装置的示意，如图8中示出的那样，本申请实施例提供的数据处理装置800，包括：
发送模块801，用于向第一设备发送第一子模型；其中，所述第一子模型为对待训练模型进行压缩得到的；
接收模块802，用于接收所述第一设备发送的第一梯度，所述第一梯度为所述第一设备训练所述第一子模型时得到的；
模型训练模块803，用于至少基于所述第一梯度，对所述待训练模型进行模型训练，以得到更新后的待训练模型。
在一种可能的实现中，存储所述第一子模型所需的存储资源小于存储所述待训练模型所需的存储资源。
在一种可能的实现中，运行所述第一子模型时所需的计算资源小于运行所述待训练模型时所需的计算资源。
在一种可能的实现中，所述计算资源至少包括如下的一种：内存资源、显存资源或cpu资源。
在一种可能的实现中，所述第一子模型为对待训练模型进行如下至少一种处理方式得到的：剪枝、知识蒸馏或量化。
在一种可能的实现中，所述模型训练模块，用于基于所述第一梯度，对所述第一子模型进行训练，得到更新后的第一子模型；至少以所述更新后的第一子模型作为老师模型，所述待训练模型为学生模型，对所述待训练模型进行知识蒸馏，以得到更新后的待训练模型。
在一种可能的实现中，所述装置还包括：
获取模块，用于获取所述第一设备的数据处理能力，所述数据处理能力包括以下至少一种：可使用的存储资源或可使用的计算资源；基于第一映射关系，获取所述第一设备对应的第一子模型；其中，所述第一映射关系包括数据处理能力与子模型之间的对应关系，且在所述第一映射关系中，所述第一设备的数据处理能力对应于所述第一子模型。
在一种可能的实现中，所述发送模块，用于向第二设备发送第二子模型；其中，所述第二子模型为对待训练模型进行压缩得到的，且所述第二子模型与所述第一子模型不同；
所述接收模块，用于接收所述第二设备发送的第二梯度，所述第二梯度为所述第二设备训练所述第二子模型时得到的；
所述模型训练模块，用于基于所述第一梯度和所述第二梯度，对所述待训练模型进行模型训练，以得到更新后的待训练模型。
在一种可能的实现中，所述第一设备与所述第二设备的数据处理能力不同，所述数据处理能力包括以下至少一种：可使用的存储资源或可使用的计算资源。
在一种可能的实现中，存储所述第一子模型和所述第二子模型所需的存储资源大小不同。
在一种可能的实现中，运行所述第一子模型和所述第二子模型时所需的计算资源大小不同。
在一种可能的实现中，所述第二子模型为对待训练模型进行如下至少一种处理方式得到的：剪枝、知识蒸馏或量化。
在一种可能的实现中，所述模型训练模块，用于基于所述第一梯度，对所述第一子模型进行训练，得到更新后的第一子模型；基于所述第二梯度，对所述第二子模型进行训练，得到更新后的第二子模型；以所述更新后的第一子模型和所述更新后的第二子模型作为老师模型，所述待训练模型为学生模型，对所述待训练模型进行知识蒸馏，以得到更新后的待训练模型。
参照图9，图9为本申请实施例提供的一种系统的示意，如图9中示出的那样，本申请实施例提供的系统900，包括：
第一设备901和服务器902；其中，所述服务器902，用于向所述第一设备901发送第一子模型；其中，所述第一子模型为对待训练模型进行压缩得到的；
所述第一设备901，用于训练所述第一子模型，以得到第一梯度；向所述服务器902发送所述第一梯度；
所述服务器902，用于至少基于所述第一梯度，对所述待训练模型进行模型训练，以得到更新后的待训练模型。
在一种可能的实现中，存储所述第一子模型所需的存储资源小于存储所述待训练模型所需的存储资源。
在一种可能的实现中，运行所述第一子模型时所需的计算资源小于运行所述待训练模型时所需的计算资源。
在一种可能的实现中，所述计算资源至少包括如下的一种：内存资源、显存资源或cpu资源。
在一种可能的实现中，所述第一子模型为对待训练模型进行如下至少一种处理方式得到的：剪枝、知识蒸馏或量化。
在一种可能的实现中，所述服务器902，用于基于所述第一梯度，对所述第一子模型进行训练，得到更新后的第一子模型；
至少以所述更新后的第一子模型作为老师模型，所述待训练模型为学生模型，对所述待训练模型进行知识蒸馏，以得到更新后的待训练模型。
在一种可能的实现中，所述服务器902，用于获取所述第一设备901的数据处理能力，所述数据处理能力包括以下至少一种：可使用的存储资源或可使用的计算资源；
基于第一映射关系，获取所述第一设备901对应的第一子模型；其中，所述第一映射关系包括数据处理能力与子模型之间的对应关系，且在所述第一映射关系中，所述第一设备901的数据处理能力对应于所述第一子模型。
在一种可能的实现中，所述系统还包括：第二设备903，所述服务器902，用于向第二设备903发送第二子模型；其中，所述第二子模型为对待训练模型进行压缩得到的，且所述第二子模型与所述第一子模型不同；
所述第二设备903用于训练所述第二子模型，以得到第二梯度；向所述服务器902发送所述第二梯度；
所述服务器902，用于所述至少基于所述第一梯度，对所述待训练模型进行模型训练，包括：
基于所述第一梯度和所述第二梯度，对所述待训练模型进行模型训练，以得到更新后的待训练模型。
在一种可能的实现中，所述第一设备901与所述第二设备903的数据处理能力不同，所述数据处理能力包括以下至少一种：可使用的存储资源或可使用的计算资源不同。
在一种可能的实现中，存储所述第一子模型和所述第二子模型所需的存储资源大小不同。
在一种可能的实现中，运行所述第一子模型和所述第二子模型时所需的计算资源大小不同。
在一种可能的实现中，所述第二子模型为对待训练模型进行如下至少一种处理方式得到的：剪枝、知识蒸馏或量化。
在一种可能的实现中，所述服务器902，用于基于所述第一梯度，对所述第一子模型进行训练，得到更新后的第一子模型；
基于所述第二梯度，对所述第二子模型进行训练，得到更新后的第二子模型；
以所述更新后的第一子模型和所述更新后的第二子模型作为老师模型，所述待训练模型为学生模型，对所述待训练模型进行知识蒸馏，以得到更新后的待训练模型。
接下来介绍本申请实施例提供的一种执行设备，请参阅图10，图10为本申请实施例提供的执行设备的一种结构示意图，执行设备1000具体可以表现为手机、平板、笔记本电脑、智能穿戴设备、服务器等，此处不做限定。其中，执行设备1000上可以部署有图10对应实施例中所描述的数据处理装置，用于实现图10对应实施例中数据处理的功能。具体的，执行设备1000包括：接收器1001、发射器1002、处理器1003和存储器1004(其中执行设备1000中的处理器1003的数量可以一个或多个，图10中以一个处理器为例)，其中，处理器1003可以包括应用处理器10031和通信处理器10032。在本申请的一些实施例中，接收器1001、发射器1002、处理器1003和存储器1004可通过总线或其它方式连接。
存储器1004可以包括只读存储器和随机存取存储器，并向处理器1003提供指令和数据。存储器1004的一部分还可以包括非易失性随机存取存储器(non-volatilerandomaccessmemory，nvram)。存储器1004存储有处理器和操作指令、可执行模块或者数据结构，或者它们的子集，或者它们的扩展集，其中，操作指令可包括各种操作指令，用于实现各种操作。
处理器1003控制执行设备的操作。具体的应用中，执行设备的各个组件通过总线系统耦合在一起，其中总线系统除包括数据总线之外，还可以包括电源总线、控制总线和状态信号总线等。但是为了清楚说明起见，在图中将各种总线都称为总线系统。
上述本申请实施例揭示的方法可以应用于处理器1003中，或者由处理器1003实现。处理器1003可以是一种集成电路芯片，具有信号的处理能力。在实现过程中，上述方法的各步骤可以通过处理器1003中的硬件的集成逻辑电路或者软件形式的指令完成。上述的处理器1003可以是通用处理器、数字信号处理器(digitalsignalprocessing，dsp)、微处理器或微控制器，还可进一步包括专用集成电路(applicationspecificintegratedcircuit，asic)、现场可编程门阵列(field-programmablegatearray，fpga)或者其他可编程逻辑器件、分立门或者晶体管逻辑器件、分立硬件组件。该处理器1003可以实现或者执行本申请实施例中的公开的各方法、步骤及逻辑框图。通用处理器可以是微处理器或者该处理器也可以是任何常规的处理器等。结合本申请实施例所公开的方法的步骤可以直接体现为硬件译码处理器执行完成，或者用译码处理器中的硬件及软件模块组合执行完成。软件模块可以位于随机存储器，闪存、只读存储器，可编程只读存储器或者电可擦写可编程存储器、寄存器等本领域成熟的存储介质中。该存储介质位于存储器1004，处理器1003读取存储器1004中的信息，结合其硬件完成上述方法的步骤。
接收器1001可用于接收输入的数字或字符信息，以及产生与执行设备的相关设置以及功能控制有关的信号输入。发射器1002可用于通过第一接口输出数字或字符信息；发射器1002还可用于通过第一接口向磁盘组发送指令，以修改磁盘组中的数据；发射器1002还可以包括显示屏等显示设备。
本申请实施例中，在一种情况下，处理器1003，用于执行图5对应实施例中的执行设备执行的方法。
本申请实施例还提供了一种训练设备，请参阅图11，图11是本申请实施例提供的训练设备一种结构示意图，训练设备1100上可以部署有图10对应实施例中所描述的神经网络训练装置，用于实现图10对应实施例中神经网络训练装置的功能，具体的，训练设备1100由一个或多个服务器实现，训练设备1100可因配置或性能不同而产生比较大的差异，可以包括一个或一个以上中央处理器(centralprocessingunits，cpu)1111(例如，一个或一个以上处理器)和存储器1132，一个或一个以上存储应用程序1142或数据1144的存储介质1130(例如一个或一个以上海量存储设备)。其中，存储器1132和存储介质1130可以是短暂存储或持久存储。存储在存储介质1130的程序可以包括一个或一个以上模块(图示没标出)，每个模块可以包括对训练设备中的一系列指令操作。更进一步地，中央处理器1111可以设置为与存储介质1130通信，在训练设备1100上执行存储介质1130中的一系列指令操作。
训练设备1100还可以包括一个或一个以上电源1126，一个或一个以上有线或无线网络接口1150，一个或一个以上输入输出接口1158；或，一个或一个以上操作系统1141，例如windowsservertm，macosxtm，unixtm,linuxtm，freebsdtm等等。
本申请实施例中，中央处理器1111，用于执行上述实施例中的训练方法相关的步骤。
本申请实施例中还提供一种包括计算机程序产品，当其在计算机上运行时，使得计算机执行如前述执行设备所执行的步骤，或者，使得计算机执行如前述训练设备所执行的步骤。
本申请实施例中还提供一种计算机可读存储介质，该计算机可读存储介质中存储有用于进行信号处理的程序，当其在计算机上运行时，使得计算机执行如前述执行设备所执行的步骤，或者，使得计算机执行如前述训练设备所执行的步骤。
本申请实施例提供的执行设备、训练设备或终端设备具体可以为芯片，芯片包括：处理单元和通信单元，所述处理单元例如可以是处理器，所述通信单元例如可以是输入/输出接口、管脚或电路等。该处理单元可执行存储单元存储的计算机执行指令，以使执行设备内的芯片执行上述实施例描述的数据处理方法，或者，以使训练设备内的芯片执行上述实施例描述的数据处理方法。可选地，所述存储单元为所述芯片内的存储单元，如寄存器、缓存等，所述存储单元还可以是所述无线接入设备端内的位于所述芯片外部的存储单元，如只读存储器(read-onlymemory，rom)或可存储静态信息和指令的其他类型的静态存储设备，随机存取存储器(randomaccessmemory，ram)等。
具体的，请参阅图12，图12为本申请实施例提供的芯片的一种结构示意图，所述芯片可以表现为神经网络处理器npu1200，npu1200作为协处理器挂载到主cpu(hostcpu)上，由hostcpu分配任务。npu的核心部分为运算电路1203，通过控制器1204控制运算电路1203提取存储器中的矩阵数据并进行乘法运算。
在一些实现中，运算电路1203内部包括多个处理单元(processengine,pe)。在一些实现中，运算电路1203是二维脉动阵列。运算电路1203还可以是一维脉动阵列或者能够执行例如乘法和加法这样的数学运算的其它电子线路。在一些实现中，运算电路1203是通用的矩阵处理器。
举例来说，假设有输入矩阵a，权重矩阵b，输出矩阵c。运算电路从权重存储器1202中取矩阵b相应的数据，并缓存在运算电路中每一个pe上。运算电路从输入存储器1201中取矩阵a数据与矩阵b进行矩阵运算，得到的矩阵的部分结果或最终结果，保存在累加器(accumulator)1208中。
统一存储器1206用于存放输入数据以及输出数据。权重数据直接通过存储单元访问控制器(directmemoryaccesscontroller，dmac)1205，dmac被搬运到权重存储器1202中。输入数据也通过dmac被搬运到统一存储器1206中。
biu为businterfaceunit即，总线接口单元1210，用于axi总线与dmac和取指存储器(instructionfetchbuffer，ifb)1209的交互。
总线接口单元1210(businterfaceunit，简称biu)，用于取指存储器1209从外部存储器获取指令，还用于存储单元访问控制器1205从外部存储器获取输入矩阵a或者权重矩阵b的原数据。
dmac主要用于将外部存储器ddr中的输入数据搬运到统一存储器1206或将权重数据搬运到权重存储器1202中或将输入数据数据搬运到输入存储器1201中。
向量计算单元1207包括多个运算处理单元，在需要的情况下，对运算电路1203的输出做进一步处理，如向量乘，向量加，指数运算，对数运算，大小比较等等。主要用于神经网络中非卷积/全连接层网络计算，如batchnormalization(批归一化)，像素级求和，对特征平面进行上采样等。
在一些实现中，向量计算单元1207能将经处理的输出的向量存储到统一存储器1206。例如，向量计算单元1207可以将线性函数；或，非线性函数应用到运算电路1203的输出，例如对卷积层提取的特征平面进行线性插值，再例如累加值的向量，用以生成激活值。在一些实现中，向量计算单元1207生成归一化的值、像素级求和的值，或二者均有。在一些实现中，处理过的输出的向量能够用作到运算电路1203的激活输入，例如用于在神经网络中的后续层中的使用。
控制器1204连接的取指存储器(instructionfetchbuffer)1209，用于存储控制器1204使用的指令；
统一存储器1206，输入存储器1201，权重存储器1202以及取指存储器1209均为on-chip存储器。外部存储器私有于该npu硬件架构。
其中，上述任一处提到的处理器，可以是一个通用中央处理器，微处理器，asic，或一个或多个用于控制上述程序执行的集成电路。
另外需说明的是，以上所描述的装置实施例仅仅是示意性的，其中所述作为分离部件说明的单元可以是或者也可以不是物理上分开的，作为单元显示的部件可以是或者也可以不是物理单元，即可以位于一个地方，或者也可以分布到多个网络单元上。可以根据实际的需要选择其中的部分或者全部模块来实现本实施例方案的目的。另外，本申请提供的装置实施例附图中，模块之间的连接关系表示它们之间具有通信连接，具体可以实现为一条或多条通信总线或信号线。
通过以上的实施方式的描述，所属领域的技术人员可以清楚地了解到本申请可借助软件加必需的通用硬件的方式来实现，当然也可以通过专用硬件包括专用集成电路、专用cpu、专用存储器、专用元器件等来实现。一般情况下，凡由计算机程序完成的功能都可以很容易地用相应的硬件来实现，而且，用来实现同一功能的具体硬件结构也可以是多种多样的，例如模拟电路、数字电路或专用电路等。但是，对本申请而言更多情况下软件程序实现是更佳的实施方式。基于这样的理解，本申请的技术方案本质上或者说对现有技术做出贡献的部分可以以软件产品的形式体现出来，该计算机软件产品存储在可读取的存储介质中，如计算机的软盘、u盘、移动硬盘、rom、ram、磁碟或者光盘等，包括若干指令用以使得一台计算机设备(可以是个人计算机，训练设备，或者网络设备等)执行本申请各个实施例所述的方法。
在上述实施例中，可以全部或部分地通过软件、硬件、固件或者其任意组合来实现。当使用软件实现时，可以全部或部分地以计算机程序产品的形式实现。
所述计算机程序产品包括一个或多个计算机指令。在计算机上加载和执行所述计算机程序指令时，全部或部分地产生按照本申请实施例所述的流程或功能。所述计算机可以是通用计算机、专用计算机、计算机网络、或者其他可编程装置。所述计算机指令可以存储在计算机可读存储介质中，或者从一个计算机可读存储介质向另一计算机可读存储介质传输，例如，所述计算机指令可以从一个网站站点、计算机、训练设备或数据中心通过有线(例如同轴电缆、光纤、数字用户线(dsl))或无线(例如红外、无线、微波等)方式向另一个网站站点、计算机、训练设备或数据中心进行传输。所述计算机可读存储介质可以是计算机能够存储的任何可用介质或者是包含一个或多个可用介质集成的训练设备、数据中心等数据存储设备。所述可用介质可以是磁性介质，(例如，软盘、硬盘、磁带)、光介质(例如，dvd)、或者半导体介质(例如固态硬盘(solidstatedisk，ssd))等。
- 还没有人留言评论。精彩留言会获得点赞！