一种数据处理方法及相关装置与流程

文档序号:33700505发布日期:2023-03-31 18:34阅读:58来源:国知局
一种数据处理方法及相关装置与流程

1.本技术涉及人工智能领域,尤其涉及一种数据处理方法及相关装置。


背景技术:

2.人工智能(artificial intelligence,ai)是利用数字计算机或者数字计算机控制的机器模拟、延伸和扩展人的智能,感知环境、获取知识并使用知识获得最佳结果的理论、方法、技术及应用系统。换句话说,人工智能是计算机科学的一个分支,它企图了解智能的实质,并生产出一种新的能以人类智能相似的方式作出反应的智能机器。人工智能也就是研究各种智能机器的设计原理与实现方法,使机器具有感知、推理与决策的功能。
3.近年来,随着计算芯片技术和大规模计算集群的快速发展,越来越多的大模型开始出现,千亿甚至万亿级别的大模型成为最近甚至未来一段时间的主流技术趋势。但是,大模型最核心的问题在于,训练和使用成本极其高昂,通常需要数百块,甚至几千块gpu、tpu或者npu芯片等计算单元组成的超大规模集群,计算数周甚至数个月时间,成本以百万至千万美元计。这个核心问题的另一个问题是,成本如此高昂的模型,如何设计和训练,以充分发挥其价值,并在设计之初,就要为未来可能的未知的使用场景做出准备,因为成本如此高昂的模型,是不可能被一遍一遍的重新训练的,因此需要有某种形式能够保存以前训练的成果,并能够为未来的可能使用场景做准备。
4.在一种实现方案中,可以将大规模模型划分为多个网络单元,针对于不同的输入样本(例如不同的词向量),设置有对应的网络单元,在使用某一输入样本进行模型训练时,多个网络单元中仅该输入样本对应的网络单元被更新,且更新后的网络单元会被存储起来,以便下一次更新时被调用。
5.在现有技术中,通常使用计算单元中的内存来存储大模型,计算单元中的内存较小,限制了大规模模型自身的大小。


技术实现要素:

6.本技术提供了一种数据处理方法,可以提高大规模模型自身的大小,并增加大规模模型的可扩展性和灵活度。
7.第一方面,本技术实施例提供了一种数据处理方法,所述方法包括:根据目标映射关系,确定目标词向量token对应的一个或多个目标网络单元以及所述一个或多个目标网络单元的存储地址,所述存储地址对应的存储空间位于计算单元之外的存储空间;从所述存储地址对应的存储空间中获取所述一个或多个目标网络单元;所述一个或多个目标网络单元用于构建神经网络;根据所述目标词向量,通过所述计算单元执行所述神经网络对应的训练过程。
8.由于大规模模型中的目标网络单元存储在计算单元之外的存储位置,在当前的训练样本为对应的词向量时,从计算单元之外的存储位置中提取对应的一个或多个目标网络单元,由于计算单元之外的存储位置的存储空间可以设置的较大,通过存算分离,可以提高
大规模模型自身的大小,并增加大规模模型的可扩展性和灵活度。
9.在一种可能的实现中,所述目标网络单元为前向传播网络ffn。
10.在一种可能的实现中,所述计算单元为图形处理器gpu、张量处理器tpu或者神经网络处理器npu。
11.在一种可能的实现中,所述存储地址对应的存储空间位于内存、网络存储器或者硬盘。
12.在一种可能的实现中,所述目标映射关系包括第一目标映射关系,所述第一目标映射关系指示多个词向量以及每个词向量对应的一个或多个网络单元,所述第一目标映射关系具体为映射表。
13.在现有技术中,通过散列hash的方式来构建训练样本(词向量token)和网络单元之间的映射关系,由于散列hash中一个训练样本仅能对应唯一一个网络单元,且由于散列hash中存在数值范围限制,随着训练样本的增大,可能会出现大量不同的训练样本对应于同一个网络单元的情况,使得训练后的模型的精度受到影响。本技术中,通过映射表的方式来记载训练样本(词向量token)和网络单元之间的映射关系,可以实现一个训练样本对应多个网络单元,且由于没有数值范围限制,不会出现大量不同的训练样本对应于同一个网络单元的情况,进而可以增大训练后的模型的精度。
14.在一种可能的实现中,所述通过所述计算单元执行所述目标神经网络对应的训练过程用于得到更新后的目标神经网络,所述更新后的目标神经网络包括更新后的所述一个或多个目标网络单元,所述方法还包括:根据所述更新后的所述一个或多个目标网络单元,更新所述存储地址对应的存储空间中的数据。
15.在一种可能的实现中,在所述根据目标映射关系,确定目标词向量对应的一个或多个目标网络单元以及所述一个或多个目标网络单元的存储地址之前,所述方法还包括:接收到用户针对于第二目标映射关系的修改信息,其中,所述第二目标映射关系包括多个词向量以及每个词向量对应的网络单元;根据所述修改信息,修改第二目标映射关系,以得到所述目标映射关系。
16.在一种可能的实现中,所述修改信息用于指示如下的至少一种:对所述第二目标映射关系中至少一个词向量对应的网络单元的删除、替换或者增加;对所述第二目标映射关系中包括的映射关系的删除、替换或者增加,所述映射关系为词向量以及词向量对应的网络单元。
17.在一种可能的实现中,在所述根据目标映射关系,确定目标词向量对应的一个或多个目标网络单元以及所述一个或多个目标网络单元的存储地址之前,所述方法还包括:在通过所述目标词向量对所述神经网络执行训练过程中,基于训练损失的下降程度低于阈值,对第二目标映射关系中所述目标词向量对应的网络单元进行增加,以得到所述目标映射关系。在训练损失的下降程度较低时,可以表示模型当前的性能无法再继续提高了,这时候可以通过增加网络单元来改变神经网络的结构,使得训练的损失可以继续下降。
18.在一种可能的实现中,所述所述目标映射关系包括多个词单元以及每个词单元对应的网络单元,所述方法还包括:
19.根据收敛后所述神经网络,从所述目标映射关系中包括的多个词单元中确定部分词单元对应的网络单元,所述收敛后所述神经网络中所述部分词单元对应的网络单元用于
构建目标神经网络;所述目标神经网络用于进行模型推理。
20.第二方面,本技术提供了一种数据处理装置,所述装置包括:
21.处理模块,用于根据目标映射关系,确定目标词向量token对应的一个或多个目标网络单元以及所述一个或多个目标网络单元的存储地址,所述存储地址对应的存储空间位于计算单元之外的存储空间;
22.从所述存储地址对应的存储空间中获取所述一个或多个目标网络单元;所述一个或多个目标网络单元用于构建神经网络;
23.模型训练模块,用于根据所述目标词向量,通过所述计算单元执行所述神经网络对应的训练过程。
24.在一种可能的实现中,所述目标网络单元为前向传播网络ffn。
25.在一种可能的实现中,所述计算单元为图形处理器gpu、张量处理器tpu或者神经网络处理器npu。
26.在一种可能的实现中,所述存储地址对应的存储空间位于内存、网络存储器或者硬盘。
27.在一种可能的实现中,所述目标映射关系包括第一目标映射关系,所述第一目标映射关系指示多个词向量以及每个词向量对应的一个或多个网络单元,所述第一目标映射关系具体为映射表。
28.在一种可能的实现中,所述通过所述计算单元执行所述目标神经网络对应的训练过程用于得到更新后的目标神经网络,所述更新后的目标神经网络包括更新后的所述一个或多个目标网络单元,所述处理模块,还用于:
29.根据所述更新后的所述一个或多个目标网络单元,更新所述存储地址对应的存储空间中的数据。
30.在一种可能的实现中,在所述根据目标映射关系,确定目标词向量对应的一个或多个目标网络单元以及所述一个或多个目标网络单元的存储地址之前,所述处理模块,还用于:
31.接收到用户针对于第二目标映射关系的修改信息,其中,所述第二目标映射关系包括多个词向量以及每个词向量对应的网络单元;
32.根据所述修改信息,修改第二目标映射关系,以得到所述目标映射关系。
33.在一种可能的实现中,所述修改信息用于指示如下的至少一种:
34.对所述第二目标映射关系中至少一个词向量对应的网络单元的删除、替换或者增加;
35.对所述第二目标映射关系中包括的映射关系的删除、替换或者增加,所述映射关系为词向量以及词向量对应的网络单元。
36.在一种可能的实现中,在所述根据目标映射关系,确定目标词向量对应的一个或多个目标网络单元以及所述一个或多个目标网络单元的存储地址之前,所述处理模块,还用于:
37.在通过所述目标词向量对所述神经网络执行训练过程中,基于训练损失的下降程度低于阈值,对第二目标映射关系中所述目标词向量对应的网络单元进行增加,以得到所述目标映射关系。
38.在一种可能的实现中,所述所述目标映射关系包括多个词单元以及每个词单元对应的网络单元,所述处理模块,还用于:
39.根据收敛后所述神经网络,从所述目标映射关系中包括的多个词单元中确定部分词单元对应的网络单元,所述收敛后所述神经网络中所述部分词单元对应的网络单元用于构建目标神经网络;所述目标神经网络用于进行模型推理。
40.第三方面,本技术实施例提供了一种数据处理装置,可以包括存储器、处理器以及总线系统,其中,存储器用于存储程序,处理器用于执行存储器中的程序,以执行如上述第一方面任一可选的方法。
41.第四方面,本技术实施例提供了一种计算机可读存储介质,计算机可读存储介质中存储有计算机程序,当其在计算机上运行时,使得计算机执行上述第一方面及任一可选的方法。
42.第五方面,本技术实施例提供了一种计算机程序产品,包括代码,当代码被执行时,用于实现上述第一方面及任一可选的方法。
43.第六方面,本技术提供了一种芯片系统,该芯片系统包括处理器,用于支持数据处理装置实现上述方面中所涉及的功能,例如,发送或处理上述方法中所涉及的数据;或,信息。在一种可能的设计中,芯片系统还包括存储器,所述存储器,用于保存执行设备或训练设备必要的程序指令和数据。该芯片系统,可以由芯片构成,也可以包括芯片和其他分立器件。
附图说明
44.图1为人工智能主体框架的一种结构示意图;
45.图2为本发明的应用系统框架示意;
46.图3为本发明的应用系统框架示意;
47.图4a为一种网络的结构示意图;
48.图4b为一种网络的结构示意图;
49.图5为一种数据处理方法的流程示意;
50.图6为一种数据处理方法的流程示意;
51.图7为一种数据处理方法的流程示意;
52.图8为一种数据处理方法的流程示意;
53.图9为一种数据处理方法的流程示意;
54.图10为一种数据处理方法的流程示意;
55.图11为一种数据处理方法的流程示意;
56.图12为本技术实施例提供的一种数据处理装置的结构示意图;
57.图13为本技术实施例提供的一种执行设备的示意图;
58.图14为本技术实施例提供的一种训练设备的示意图;
59.图15为本技术实施例提供的一种芯片的示意图。
具体实施方式
60.下面结合本发明实施例中的附图对本发明实施例进行描述。本发明的实施方式部
分使用的术语仅用于对本发明的具体实施例进行解释,而非旨在限定本发明。
61.下面结合附图,对本技术的实施例进行描述。本领域普通技术人员可知,随着技术的发展和新场景的出现,本技术实施例提供的技术方案对于类似的技术问题,同样适用。
62.本技术的说明书和权利要求书及上述附图中的术语“第一”、“第二”等是用于区别类似的对象,而不必用于描述特定的顺序或先后次序。应该理解这样使用的术语在适当情况下可以互换,这仅仅是描述本技术的实施例中对相同属性的对象在描述时所采用的区分方式。此外,术语“包括”和“具有”以及他们的任何变形,意图在于覆盖不排他的包含,以便包含一系列单元的过程、方法、系统、产品或设备不必限于那些单元,而是可包括没有清楚地列出的或对于这些过程、方法、产品或设备固有的其它单元。
63.首先对人工智能系统总体工作流程进行描述,请参见图1,图1示出的为人工智能主体框架的一种结构示意图,下面从“智能信息链”(水平轴)和“it价值链”(垂直轴)两个维度对上述人工智能主题框架进行阐述。其中,“智能信息链”反映从数据的获取到处理的一列过程。举例来说,可以是智能信息感知、智能信息表示与形成、智能推理、智能决策、智能执行与输出的一般过程。在这个过程中,数据经历了“数据—信息—知识—智慧”的凝练过程。“it价值链”从人智能的底层基础设施、信息(提供和处理技术实现)到系统的产业生态过程,反映人工智能为信息技术产业带来的价值。
64.(1)基础设施
65.基础设施为人工智能系统提供计算能力支持,实现与外部世界的沟通,并通过基础平台实现支撑。通过传感器与外部沟通;计算能力由智能芯片(cpu、npu、gpu、asic、fpga等硬件加速芯片)提供;基础平台包括分布式计算框架及网络等相关的平台保障和支持,可以包括云存储和计算、互联互通网络等。举例来说,传感器和外部沟通获取数据,这些数据提供给基础平台提供的分布式计算系统中的智能芯片进行计算。
66.(2)数据
67.基础设施的上一层的数据用于表示人工智能领域的数据来源。数据涉及到图形、图像、语音、文本,还涉及到传统设备的物联网数据,包括已有系统的业务数据以及力、位移、液位、温度、湿度等感知数据。
68.(3)数据处理
69.数据处理通常包括数据训练,机器学习,深度学习,搜索,推理,决策等方式。
70.其中,机器学习和深度学习可以对数据进行符号化和形式化的智能信息建模、抽取、预处理、训练等。
71.推理是指在计算机或智能系统中,模拟人类的智能推理方式,依据推理控制策略,利用形式化的信息进行机器思维和求解问题的过程,典型的功能是搜索与匹配。
72.决策是指智能信息经过推理后进行决策的过程,通常提供分类、排序、预测等功能。
73.(4)通用能力
74.对数据经过上面提到的数据处理后,进一步基于数据处理的结果可以形成一些通用的能力,比如可以是算法或者一个通用系统,例如,翻译,文本的分析,计算机视觉的处理,语音识别,图像的识别等等。
75.(5)智能产品及行业应用
76.智能产品及行业应用指人工智能系统在各领域的产品和应用,是对人工智能整体解决方案的封装,将智能信息决策产品化、实现落地应用,其应用领域主要包括:智能终端、智能交通、智能医疗、自动驾驶、智慧城市等。
77.本技术实施例可以应用于信息推荐领域,该场景包括但不限于涉及电商产品推荐、搜索引擎结果推荐、应用市场推荐、音乐推荐、视频推荐等场景,各种不同应用场景中被推荐的物品也可以称为“对象”以方便后续描述,即在不同的推荐场景中,推荐对象可以是app,或者视频,或者音乐,或者某款商品(如线上购物平台的呈现界面,会根据用户的不同而显示不同的商品进行呈现,这实质也可以是通过推荐模型的推荐结果来进行呈现)。这些推荐场景通常涉及用户行为日志采集、日志数据预处理(例如,量化、采样等)、样本集训练以获得推荐模型、根据推荐模型对训练样本项对应的场景中所涉及的对象(如app、音乐等)进行分析处理、例如,推荐模型训练环节中所选择的样本来自于手机应用市场用户对于所推荐app的操作行为,则由此所训练出来的推荐模型则适用于上述手机app应用市场,或者可以用于其它的类型的终端的app应用市场进行终端app的推荐。推荐模型将最终计算出各个待推荐对象的推荐概率或者分值,推荐系统根据一定的选择规则选定的推荐结果,例如按照推荐概率或者分值进行排序,通过相应的应用或者终端设备呈现给用户、用户对推荐结果中的对象进行操作以生成用户行为日志等环节。
78.参照图4a,在推荐过程中,当一个用户与推荐系统进行交互会触发一个推荐请求,推荐系统会将该请求及其相关的特征信息输入到部署的推荐模型中,然后预测用户对所有候选对象的点击率。随后,根据预测的点击率对候选对象进行降序排列,按顺序将候选对象展示在不同的位置作为对用户的推荐结果。用户对展示的项目进行浏览并发生用户行为,如浏览、点击和下载等。这些用户行为会被存入日志中作为训练数据,通过离线训练模块不定期地更新推荐模型的参数,提高模型的推荐效果。
79.比如,用户打开手机应用市场即可触发应用市场的推荐模块,应用市场的推荐模块会根据用户的历史下载记录、用户点击记录,应用的自身特征,时间、地点等环境特征信息,预测用户对给定的各个候选应用的下载可能性。根据预测的结果,应用市场按照可能性降序展示,达到提高应用下载概率的效果。具体来说,将更有可能下载的应用排在靠前的位置,将不太可能下载的应用排列在靠后的位置。而用户的行为也会存入日志并通过离线训练模块对预测模型的参数进行训练和更新。
80.又比如,在终身伴侣相关的应用中,可以基于用户在视频、音乐、新闻等域的历史数据,通过各种模型和算法,仿照人脑机制,构建认知大脑,搭建用户终身学习系统框架。终身伴侣可以根据系统数据和应用数据等来记录用户过去发生的事件,理解用户的当前意图,预测用户未来的动作或行为,最终实现智能服务。在当前第一阶段,根据音乐app、视频app和浏览器app等获取用户的行为数据(包含端侧短信、照片、邮件事件等信息),一方面构建用户画像系统,另一方面实现基于用户信息过滤、关联分析、跨域推荐、因果推理等的学习与记忆模块,构建用户个人知识图谱。
81.接下来介绍本技术实施例的应用架构。
82.参见附图2,本发明实施例提供了一种推荐系统架构200。数据采集设备260用于采集样本,一个训练样本可以由多个特征信息(或者描述为属性信息,例如用户属性以及物品属性)组成,特征信息可以有多种,具体可以包括用户特征信息和对象特征信息以及标签特
征,用户特征信息用于表征用户的特征,例如性别,年龄,职业,爱好等,对象特征信息用于表征向用户所推送的对象的特征,不同的推荐系统对应不同的对象,不同的对象所需要提取的特征类型也不想同,例如app市场的训练样本中所提取的对象特征可以为,app的名称(标识),类型,大小等;而电商类app的训练样本中所提起的对象特征可以为,商品的名称,所属的类别,价格区间等;标签特征,则是用于表示这个样本是正例还是负例,通常样本的标签特征可以通过用户对所推荐对象的操作信息所获的,用户对所推荐对象有进行操作的样本为正例,用户对所推荐对象没有进行操作,或者仅浏览的样本为负例,例如当用户点击或者下载或者购买了所推荐的对象,则所述标签特征为1,表示该样本是正例,而如果用户没有对所推荐的对象进行任何操作,则所述标签特征为0,表示该样本是负例。样本在采集后可以保存在数据库230中,数据库230中的样本中的部分或全部特征信息也可以直接从客户设备240中获取,如用户特征信息,用户对对象的操作信息(用于确定类型标识),对象特征信息(如对象标识)等。训练设备220基于数据库230中样本训练获取模型参数矩阵用于生成推荐模型201。下面将更详细地描述训练设备220如何训练得到用于生成推荐模型201的模型参数矩阵,推荐模型201能够用于对大量对象进行评估从而得出各个待推荐对象的分值,进一步的还可以从大量对象的评估结果中推荐指定或者预设数目个对象,计算模块211基于推荐模型201的评估结果获取推荐结果,通过i/o接口212推荐给客户设备。
83.在本技术实施例中,该训练设备220可以从数据库230中样本集内选取正、负样本添加到所述训练集中,之后采用推荐模型对训练集中的样本进行训练从而得到训练后的推荐模型;计算模块211的实现细节可以参照图5所示的方法实施例的详细描述。
84.训练设备220基于样本训练获得模型参数矩阵后用于构建推荐模型201后,将推荐模型201发送给执行设备210,或者直接将模型参数矩阵发送给执行设备210,在执行设备210中构建推荐模型,用于进行相应系统的推荐,例如基于视频相关的样本训练获得的推荐模型可以用于视频网站或app中对用户进行视频的推荐,基于app相关的样本训练获得的推荐模型可以用于应用市场中对用户进行app的推荐。
85.执行设备210配置有i/o接口212,与外部设备进行数据交互,执行设备210可以通过i/o接口212从客户设备240获取用户特征信息,例如用户标识、用户身份、性别、职业、爱好等,此部分信息也可以从系统数据库中获取。推荐模型201基于用户特征信息和待推荐对象特征信息向用户推荐目标推荐对象。执行设备210可以设置在云端服务器中,也可以设置于用户客户端中。
86.执行设备210可以调用数据存储系统250中的数据、代码等,同时也可以将输出的数据存入数据存储系统250中。数据存储系统250可以设置于执行设备210中,也可以独立设置,或者设置于其他网络实体中,数量可以是一个也可以是多个。
87.计算模块211使用推荐模型201对用户特征信息,待推荐对象特征信息进行处理,例如,该计算模块211使用推荐模型201对用户特征信息,以及待推荐对象的特征信息进行分析处理,从而得出该待推荐对象的分值,对待推荐对象按照分值进行排序,其中,排序靠前的对象将作为推荐给客户设备240的对象。
88.最后,i/o接口212将推荐结果返回给客户设备240,呈现给用户。
89.更深层地,训练设备220可以针对不同的目标,基于不同的样本特征信息生成相应的推荐模型201,以给用户提供更佳的结果。
90.值得注意的,附图2仅是本发明实施例提供的一种系统架构的示意图,图中所示设备、器件、模块等之间的位置关系不构成任何限制,例如,在附图2中,数据存储系统250相对执行设备210是外部存储器,在其它情况下,也可将数据存储系统250置于执行设备210中。
91.在本技术实施例中,该训练设备220、执行设备210、客户设备240可以分别为三个不同的物理设备,也可能该训练设备220和执行设备210在同一个物理设备或者一个集群上,也可能该执行设备210与该客户设备240在同一个物理设备或者一个集群上。
92.参见附图3,是本发明实施例提的一种系统架构300。在此架构中执行设备210由一个或多个服务器实现,可选的,与其它计算设备配合,例如:数据存储、路由器、负载均衡器等设备;执行设备210可以布置在一个物理站点上,或者分布在多个物理站点上。执行设备210可以使用数据存储系统250中的数据,或者调用数据存储系统250中的程序代码实现对象推荐的功能,具体地,将待推荐的对象的信息输入到推荐模型中,推荐模型为每个待推荐对象生成预估分数,然后按照预估分数从高到低的顺序进行排序,按照排序结果向用户推荐该待推荐对象。例如,将排序结果中的前10个对象推荐给用户。
93.其中,数据存储系统250用于接收和存储训练设备发送的推荐模型的参数,以及用于存储通过推荐模型得到的推荐结果的数据,当然还可能包括该存储系统250正常运行所需的程序代码(或指令)。数据存储系统250可以为部署在执行设备210以外的一个设备或者多个设备构成的分布式存储集群,此时,当执行设备210需要使用存储系统250上的数据时,可以由存储系统250向执行设备210发送该执行设备所需的数据,相应地,该执行设备210接收并存储(或者缓存)该数据。当然数据存储系统250也可以部署在执行设备210内,当部署在执行设备210内时,该分布式存储系统可以包括一个或者多个存储器,可选的,存在多个存储器时,不同的存储器用于存储不同类型的数据,如通过训练设备生成的推荐模型的模型参数和通过推荐模型得到的推荐结果的数据可以分别存储在两个不同的存储器上。
94.用户可以操作各自的用户设备(例如本地设备301和本地设备302)与执行设备210进行交互。每个本地设备可以表示任何计算设备,例如个人计算机、计算机工作站、智能手机、平板电脑、智能摄像头、智能汽车或其他类型蜂窝电话、媒体消费设备、可穿戴设备、机顶盒、游戏机等。
95.每个用户的本地设备可以通过任何通信机制/通信标准的通信网络与执行设备210进行交互,通信网络可以是广域网、局域网、点对点连接等方式,或它们的任意组合。
96.在另一种实现中,执行设备210可以由本地设备实现,例如,本地设备301可以基于推荐模型实现执行设备210的的推荐功能获取用户特征信息并向用户反馈推荐结果,或者为本地设备302的用户提供服务。
97.由于本技术实施例涉及大量神经网络的应用,为了便于理解,下面先对本技术实施例涉及的相关术语及神经网络等相关概念进行介绍。
98.1、点击概率(click-throughrate,ctr)
99.点击概率又可以称为点击率,是指网站或者应用程序上推荐信息(例如,推荐物品)被点击次数和曝光次数之比,点击率通常是推荐系统中衡量推荐系统的重要指标。
100.2、个性化推荐系统
101.个性化推荐系统是指根据用户的历史数据(例如本技术实施例中的操作信息),利用机器学习算法进行分析,并以此对新请求进行预测,给出个性化的推荐结果的系统。
102.3、离线训练(offlinetraining)
103.离线训练是指在个性化推荐系统中,根据用户的历史数据(例如本技术实施例中的操作信息),对推荐模型参数按照器学习的算法进行迭代更新直至达到设定要求的模块。
104.4、在线预测(onlineinference)
105.在线预测是指基于离线训练好的模型,根据用户、物品和上下文的特征预测该用户在当前上下文环境下对推荐物品的喜好程度,预测用户选择推荐物品的概率。
106.例如,图4a是本技术实施例提供的推荐系统的示意图。如图4a所示,当一个用户进入统,会触发一个推荐的请求,推荐系统会将该请求及其相关信息(例如本技术实施例中的操作信息)输入到推荐模型,然后预测用户对系统内的物品的选择率。进一步,根据预测的选择率或基于该选择率的某个函数将物品降序排列,即推荐系统可以按顺序将物品展示在不同的位置作为对用户的推荐结果。用户浏览不同的处于位置的物品并发生用户行为,如浏览、选择以及下载等。同时,用户的实际行为会存入日志中作为训练数据,通过离线训练模块不断更新推荐模型的参数,提高模型的预测效果。
107.例如,用户打开智能终端(例如,手机)中的应用市场即可触发应用市场中的推荐系统。应用市场的推荐系统会根据用户的历史行为日志,例如,用户的历史下载记录、用户选择记录,应用市场的自身特征,比如时间、地点等环境特征信息,预测用户下载推荐的各个候选app的概率。根据计算的结果,应用市场的推荐系统可以按照预测的概率值大小降序展示候选app,从而提高候选app的下载概率。
108.示例性地,可以将预测的用户选择率较高的app展示在靠前的推荐位置,将预测的用户选择率较低的app展示在靠后的推荐位置。
109.上述推荐模型可以是神经网络模型,下面对本技术实施例可能涉及的神经网络的相关术语和概念进行介绍。
110.(1)神经网络
111.神经网络可以是由神经单元组成的,神经单元可以是指以xs(即输入数据)和截距1为输入的运算单元,该运算单元的输出可以为:
[0112][0113]
其中,s=1、2、
……
n,n为大于1的自然数,ws为xs的权重,b为神经单元的偏置。f为神经单元的激活函数(activation functions),用于将非线性特性引入神经网络中,来将神经单元中的输入信号转换为输出信号。该激活函数的输出信号可以作为下一层卷积层的输入,激活函数可以是sigmoid函数。神经网络是将多个上述单一的神经单元联结在一起形成的网络,即一个神经单元的输出可以是另一个神经单元的输入。每个神经单元的输入可以与前一层的局部接受域相连,来提取局部接受域的特征,局部接受域可以是由若干个神经单元组成的区域。
[0114]
(2)深度神经网络
[0115]
深度神经网络(deep neural network,dnn),也称多层神经网络,可以理解为具有很多层隐含层的神经网络,这里的“很多”并没有特别的度量标准。从dnn按不同层的位置划分,dnn内部的神经网络可以分为三类:输入层,隐含层,输出层。一般来说第一层是输入层,最后一层是输出层,中间的层数都是隐含层。层与层之间是全连接的,也就是说,第i层的任
意一个神经元一定与第i+1层的任意一个神经元相连。虽然dnn看起来很复杂,但是就每一层的工作来说,其实并不复杂,简单来说就是如下线性关系表达式:其中,是输入向量,是输出向量,是偏移向量,w是权重矩阵(也称系数),α()是激活函数。每一层仅仅是对输入向量经过如此简单的操作得到输出向量由于dnn层数多,则系数w和偏移向量的数量也就很多了。这些参数在dnn中的定义如下所述:以系数w为例:假设在一个三层的dnn中,第二层的第4个神经元到第三层的第2个神经元的线性系数定义为上标3代表系数w所在的层数,而下标对应的是输出的第三层索引2和输入的第二层索引4。总结就是:第l-1层的第k个神经元到第l层的第j个神经元的系数定义为需要注意的是,输入层是没有w参数的。在深度神经网络中,更多的隐含层让网络更能够刻画现实世界中的复杂情形。理论上而言,参数越多的模型复杂度越高,“容量”也就越大,也就意味着它能完成更复杂的学习任务。训练深度神经网络的也就是学习权重矩阵的过程,其最终目的是得到训练好的深度神经网络的所有层的权重矩阵(由很多层的向量w形成的权重矩阵)。
[0116]
(3)损失函数
[0117]
在训练深度神经网络的过程中,因为希望深度神经网络的输出尽可能的接近真正想要预测的值,所以可以通过比较当前网络的预测值和真正想要的目标值,再根据两者之间的差异情况来更新每一层神经网络的权重向量(当然,在第一次更新之前通常会有初始化的过程,即为深度神经网络中的各层预先配置参数),比如,如果网络的预测值高了,就调整权重向量让它预测低一些,不断的调整,直到深度神经网络能够预测出真正想要的目标值或与真正想要的目标值非常接近的值。因此,就需要预先定义“如何比较预测值和目标值之间的差异”,这便是损失函数(loss function)或目标函数(objective function),它们是用于衡量预测值和目标值的差异的重要方程。其中,以损失函数举例,损失函数的输出值(loss)越高表示差异越大,那么深度神经网络的训练就变成了尽可能缩小这个loss的过程。
[0118]
(4)反向传播算法
[0119]
可以采用误差反向传播(back propagation,bp)算法在训练过程中修正初始模型中参数的大小,使得模型的误差损失越来越小。具体地,前向传递输入信号直至输出会产生误差损失,通过反向传播误差损失信息来更新初始模型中的参数,从而使误差损失收敛。反向传播算法是以误差损失为主导的反向传播运动,旨在得到最优的模型参数,例如权重矩阵。
[0120]
(5)机器学习系统
[0121]
基于输入数据和标签,通过梯度下降等优化方法训练机器学习模型的参数,最终利用训练得到的模型来完成未知数据的预测。
[0122]
(6)个性化推荐系统
[0123]
根据用户的历史数据,利用机器学习算法进行分析和建模,并以此对新的用户请求进行预测,给出个性化的推荐结果的系统。
[0124]
(7)混合专家:在神经网络中,将每个线性层看做一个专家,多个线性层并行,将其中一个或者若干个专家的输出进行加权平均,并将结果输出到下一层。
[0125]
(8)计算图:将用户通过深度神经网络框架描述的运算构建成的一个计算的图结构,通常描述整个深度神经网络计算的流程,包括前向、反向传播以及参数更新。
[0126]
(9)记忆网络:将输入转化为网络里面内在的向量,并将向量保存在记忆矩阵里面。
[0127]
(10)路由:控制输入数据和某个网络模块映射关系的函数。
[0128]
(11)持续学习:也称终身学习(life-long learning)是解决此类问题的研究方向,它的目标是扩展模型适应能力,令模型能够在不同时刻学习不同任务的知识,同时不会遗忘先前任务的特征。
[0129]
(12)页表:一种数据结构,可以用于记载两种数据之间的映射关系。通常用于将虚拟地址转换为物理地址。分级页表也是页表的一种。页表被存储在存储介质中,页表的基址,也简称页表基址,是指存储该页表的物理地址的起始地址。页表的大小决定这个页表对应的虚拟地址的取值范围。页表中的虚拟地址是使用该页表的实例或者设备的虚拟地址,用于指示该页表中某个位置(或者说某个表项)的物理地址相对于页表基址的偏移量。而页表中某个位置存储的数据,就是使用这个页表的实例或者设备的物理地址空间中的物理地址。一个页表中包括多个表项,每个表项都可以指示一个虚拟地址与一个物理地址的映射关系,应理解的是,这里所说的表项是个功能性的描述,表项的具体实现方式本技术不做限定,例如一个虚拟地址与一个物理地址的映射关系可能是跨级的,一个虚拟地址与一个物理地址的映射关系也可能是间接的。本技术中的页表具体可以为多级页表。
[0130]
(13)transformer层
[0131]
参照图4b,图4b为一种transformer层的架构示意,如图4b所示,神经网络包括嵌入层和至少一个transformer层,至少一个transformer层可以为n个transformer层(n大于0的整数),其中,每个transformer层包括依次相邻的注意力层、加和与归一化(add&norm)层、前馈(feed forward)层和加和与归一化层。在嵌入层,对当前输入进行嵌入处理,得到多个特征向量;在所述注意力层,从所述transformer层的上一层获取p个输入向量,以p个输入向量中的任意的第一输入向量为中心,基于预设的注意力窗口范围内的各个输入向量与该第一输入向量之间的关联度,得到该第一输入向量对应的中间向量,如此确定出p个输入向量对应的p个中间向量;在所述池化层,将所述p个中间向量合并为q个输出向量,其中至少一个transformer层中最后一个transformer层得到的多个输出向量用作所述当前输入的特征表示。
[0132]
接下来,结合具体例子对上述各步骤进行具体介绍。
[0133]
首先,在所述嵌入层,对当前输入进行嵌入处理,得到多个特征向量。
[0134]
嵌入层可以称为输入嵌入(input embedding)层。当前输入可以为文本输入,例如可以为一段文本,也可以为一个句子。文本可以为中文文本,也可以为英文文本,还可以为其他语言文本。嵌入层在获取当前输入后,可以对该当前输入中各个词进行嵌入处理,可得到各个词的特征向量。在一些实施例中,如图4b所示,所述嵌入层包括输入嵌入层和位置编码(positional encoding)层。在输入嵌入层,可以对当前输入中的各个词进行词嵌入处理,从而得到各个词的词嵌入向量。在位置编码层,可以获取各个词在该当前输入中的位置,进而对各个词的位置生成位置向量。在一些示例中,各个词的位置可以为各个词在该当前输入中的绝对位置。以当前输入为“几号应还花呗”为例,其中的“几”的位置可以表示为
第一位,“号”的位置可以表示为第二位,
……
。在一些示例中,各个词的位置可以为各个词之间的相对位置。仍以当前输入为“几号应还款”为例,其中的“几”的位置可以表示为“号”之前,“号”的位置可以表示为“几”之后、“应”之前,
……
。当得到当前输入中各个词的词嵌入向量和位置向量时,可以将各个词的位置向量和对应的词嵌入向量进行组合,得到各个词特征向量,即得到该当前输入对应的多个特征向量。多个特征向量可以表示为具有预设维度的嵌入矩阵。可以设定该多个特征向量中的特征向量个数为m,预设维度为h维,则该多个特征向量可以表示为m
×
h的嵌入矩阵。
[0135]
其次,从所述第一transformer层的上一层获取p个输入向量,以p个输入向量中的任意的第一输入向量为中心,基于预设的注意力窗口范围内的各个输入向量与该第一输入向量之间的关联度,得到该第一输入向量对应的中间向量,如此确定出p个输入向量对应的p个中间向量。注意力层也可以称为多头注意力(multi-head attention)层。在一个例子中,注意力层可以为固定窗口多头注意力(fixed window multi-head attention)层。
[0136]
在一些实施例中,第一transformer层可以为上述嵌入层的下一层,p个输入向量为从嵌入层得到的所述多个特征向量。在一些实施例中,本说明书实施例提供的神经网络中的至少一个transformer层还包括第二transformer层。该第二transformer层为第一自注意力的上一层,则p个输入向量为第二transformer层输出的p个输出向量。在该神经网络中的最后一个transformer层,通过上述步骤的多个输出向量可用作当前输入的特征表示。该特征表示为为当前输入的一种适合计算机处理的特征表示。
[0137]
(14)注意力机制(attention mechanism)
[0138]
注意力机制模仿了生物观察行为的内部过程,即一种将内部经验和外部感觉对齐从而增加部分区域的观察精细度的机制,能够利用有限的注意力资源从大量信息中快速筛选出高价值信息。注意力机制可以快速提取稀疏数据的重要特征,因而被广泛用于自然语言处理任务,特别是机器翻译。而自注意力机制(self-attention mechanism)是注意力机制的改进,其减少了对外部信息的依赖,更擅长捕捉数据或特征的内部相关性。注意力机制的本质思想可以改写为如下公式:
[0139]
其中,lx=||source||代表source的长度,公式含义即将source中的构成元素想象成是由一系列的数据对构成,此时给定目标target中的某个元素query,通过计算query和各个key的相似性或者相关性,得到每个key对应value的权重系数,然后对value进行加权求和,即得到了最终的attention数值。所以本质上attention机制是对source中元素的value值进行加权求和,而query和key用来计算对应value的权重系数。从概念上理解,把attention可以理解为从大量信息中有选择地筛选出少量重要信息并聚焦到这些重要信息上,忽略大多不重要的信息。聚焦的过程体现在权重系数的计算上,权重越大越聚焦于其对应的value值上,即权重代表了信息的重要性,而value是其对应的信息。自注意力机制可以理解为内部attention(intra attention),attention机制发生在target的元素query和source中的所有元素之间,自注意力机制指的是在source内部元素之间或者target内部元素之间发生的attention机制,也可以理解为target=source这种特殊情况下的注意力计算机制,其具体计算过程是一样的,只是计算对象发生了变化而已。
[0140]
近年来,随着计算芯片技术和大规模计算集群的快速发展,越来越多的大模型开始出现,千亿甚至万亿级别的大模型成为最近甚至未来一段时间的主流技术趋势。但是,大
模型最核心的问题在于,训练和使用成本极其高昂,通常需要数百块,甚至几千块gpu、tpu或者npu芯片等计算单元组成的超大规模集群,计算数周甚至数个月时间,成本以百万至千万美元计。这个核心问题的另一个问题是,成本如此高昂的模型,如何设计和训练,以充分发挥其价值,并在设计之初,就要为未来可能的未知的使用场景做出准备,因为成本如此高昂的模型,是不可能被一遍一遍的重新训练的,因此需要有某种形式能够保存以前训练的成果,并能够为未来的可能使用场景做准备。
[0141]
在一种实现方案中,可以将大规模模型划分为多个网络单元,针对于不同的输入样本(例如不同的词向量),设置有对应的网络单元,在使用某一输入样本进行模型训练时,多个网络单元中仅该输入样本对应的网络单元被更新,且更新后的网络单元会被存储起来,以便下一次更新时被调用。
[0142]
在现有技术中,通常使用计算单元中的内存来存储大模型,计算单元中的内存较小,限制了大规模模型自身的大小。
[0143]
为了解决上述问题,本技术提供了一种数据处理方法。
[0144]
参照图5,图5为本技术实施例提供的一种数据处理方法的实施例示意,如图5示出的那样,本技术实施例提供的一种数据处理方法,包括:
[0145]
501、根据目标映射关系,确定目标词向量token对应的一个或多个目标网络单元以及所述一个或多个目标网络单元的存储地址,所述存储地址对应的存储空间位于计算单元之外的存储空间。
[0146]
本技术实施例中,步骤501的执行主体可以为终端设备,终端设备可以为便携式移动设备,例如但不限于移动或便携式计算设备(如智能手机)、个人计算机、服务器计算机、手持式设备(例如平板)或膝上型设备、多处理器系统、游戏控制台或控制器、基于微处理器的系统、机顶盒、可编程消费电子产品、移动电话、具有可穿戴或配件形状因子(例如,手表、眼镜、头戴式耳机或耳塞)的移动计算和/或通信设备、网络pc、小型计算机、大型计算机、包括上面的系统或设备中的任何一种的分布式计算环境等等。
[0147]
本技术实施例中,步骤501的执行主体可以为云侧的服务器。
[0148]
为了方便描述,以下不对执行主体的形态进行区分,都描述为训练设备。
[0149]
在一种可能的实现中,所述目标网络单元为前向传播网络ffn,目标网络单元在本技术实施例中也可以称之为专家网络。
[0150]
在一种可能的实现中,可以预先构建不同训练样本(词向量token)和目标网络单元之间的映射关系(或者可以称之为多值映射表或者多级映射表),大规模网络在针对不同训练样本(词向量token)进行模型训练时,可以将大规模网络中目标网络单元所在位置的网络结构设置成映射关系中所指示的网络单元或者网络单元的组合。
[0151]
其中,目标映射关系可以包括第一目标映射关系和第二目标映射关系,第一目标映射关系可以指示词向量和网络单元之间的映射关系,具体可以是词向量和网络单元对应的标识id之间的映射关系。标识:用于将一种或者一个事物与其他同类或不同类的事物做区分。比如,节点的标识,网络的标识,网卡的标识。标识可以是名称,编号,也可以是以某个有区分性的特征作为标识,例如用种类标识。本技术对各种标识的实现不做限制,只要可以做出区分即可。
[0152]
在现有技术中,通过散列hash的方式来构建训练样本(词向量token)和网络单元
之间的映射关系,由于散列hash中一个训练样本仅能对应唯一一个网络单元,且由于散列hash中存在数值范围限制,随着训练样本的增大,可能会出现大量不同的训练样本对应于同一个网络单元的情况,使得训练后的模型的精度受到影响。本技术中,通过映射表的方式来记载训练样本(词向量token)和网络单元之间的映射关系,可以实现一个训练样本对应多个网络单元,且由于没有数值范围限制,不会出现大量不同的训练样本对应于同一个网络单元的情况,进而可以增大训练后的模型的精度。
[0153]
在一种可能的实现中,可以将大规模网络(例如盘古alpha或者transformer中的ffn)替换为图6中的存储单元,即若干个并行排列的expert。这些专家的分组是由多值映射表决定的,每个多值映射表许多由键值对(key-value)构成,可选的,词表中的词(token)构成多值映射表的键(key),值(value)就是专家的id号,比如图中的e1到e11。构建好这个多值映射表后,在训练过程中,控制模块就可以根据数据单元的数据动态从存储单元中提取需要计算的计算单元高效计算。可选的,也可以根据loss值决定是否动态扩展存储单元的专家模块。
[0154]
关于上述多值映射表的构建:
[0155]
分组是在构建多值映射表的时候由用户决定的,例如,可以构建如下图7的多射映射表。用户先确定需要建模的分组数量g,以及每个分组初始化的专家个数k_g,对词表的token来说,属于哪个或哪些分组,是由用户自己的决定的,例如,用户可以根据token所属的语言分组,中文为一个专家组,英文为另一个专家组,有些token属于两个专家组,比如一些阿拉伯数字。分好分组以后,每个分组的专家数也是固定的,那么属于这个专家组的token就可以通过某个函数进行分配映射的专家,比如可以随机,也可以顺序取余。在训练开始之前,需要构建好一个初始版本的多值映射表,然后启动训练。
[0156]
在一种可能的实现中,在一种可能的实现中,所述计算单元为图形处理器gpu、张量处理器tpu或者神经网络处理器npu。
[0157]
在一种可能的实现中,所述存储地址对应的存储空间位于内存、网络存储器或者硬盘。
[0158]
在一种可能的实现中,所述目标映射关系包括第一目标映射关系,所述第一目标映射关系指示多个词向量以及每个词向量对应的一个或多个网络单元,所述第一目标映射关系具体为映射表。
[0159]
在一种可能的实现中,所述目标映射关系还可以包括第三目标映射关系,所述第三目标映射关系指示每个网络单元的存储位置。该存储位置可以通过存储空间对应的物理地址或者虚拟地址来表示,基于该虚拟地址可以唯一确定对应的物理地址,例如,地址可以包括起始地址以及偏置。
[0160]
由于大规模模型中的目标网络单元存储在计算单元之外的存储位置,在当前的训练样本为对应的词向量时,从计算单元之外的存储位置中提取对应的一个或多个目标网络单元,由于计算单元之外的存储位置的存储空间可以设置的较大,通过存算分离,可以提高大规模模型自身的大小,并增加大规模模型的可扩展性和灵活度。
[0161]
在一种可能的实现中,上述目标映射关系可以被用户修改。
[0162]
具体的,在一种可能的实现中,在所述根据目标映射关系,确定目标词向量对应的一个或多个目标网络单元以及所述一个或多个目标网络单元的存储地址之前,还可以接收
到用户针对于第二目标映射关系的修改信息,其中,所述第二目标映射关系包括多个词向量以及每个词向量对应的网络单元;根据所述修改信息,修改第二目标映射关系,以得到所述目标映射关系。
[0163]
修改信息可以用于修改第二目标映射关系中某个词向量所对应的网络单元进行修改,或者是对词向量与网络单元的对应关系进行修改。
[0164]
在一种可能的实现中,所述修改信息用于指示如下的至少一种:对所述第二目标映射关系中至少一个词向量对应的网络单元的删除、替换或者增加;对所述第二目标映射关系中包括的映射关系的删除、替换或者增加,所述映射关系为词向量以及词向量对应的网络单元。
[0165]
此外,系统还可以基于模型训练的情况自行进行映射关系的修改。
[0166]
在一种可能的实现中,在所述根据目标映射关系,确定目标词向量对应的一个或多个目标网络单元以及所述一个或多个目标网络单元的存储地址之前,可以在通过所述目标词向量对所述神经网络执行训练过程中,基于训练损失的下降程度低于阈值(在训练损失的下降程度较低时,可以表示模型当前的性能无法再继续提高了,这时候可以通过增加网络单元来改变神经网络的结构,使得训练的损失可以继续下降),对第二目标映射关系中所述目标词向量对应的网络单元进行增加,以得到所述目标映射关系。
[0167]
在一种可能的实现中,可以通过确定应映射关系和大容量的映射关系数据结构,可以支撑本实施例进行记忆网络的容量扩增和删减,并在同时取得持续学习的效果。持续学习指的是模型能够在学习了一段时间的数据后,继续学习新的数据,能够避免之前学到的知识被遗忘,同时能够学习到新的知识的技术。本实施例的一个可选的流程如下:
[0168]
如图9所示,步骤如下:
[0169]
步骤1、训练从配置文件中读取是否修改映射表,如果有修改,则按照配置修改多值映射表和相应的存储单元。这种配置是非常灵活的,可以是同一个专家组的扩充,如图扩充专家组的e4,也可以是扩充新的专家组,如图中扩充专家组3。然后继续训练。
[0170]
步骤2、读取配置文件中的扩增阈值k,针对每个专家组的loss,并时刻记录loss的最小值,如果最小值已经连续k步没有更新,则认为模型的loss已经不再降了,进入步骤3。否则就继续训练。
[0171]
步骤3、满足步骤2条件后,针对不降的loss对应的专家组,修改多值映射表,假设原始专家组为m个专家,则按照某种规则将多值映射表中1/m个token映射的专家,添加一个新的映射专家,如图中的e4。其中这个规则可以是最高频的1/m个token,或者是随机选取1/m个token。
[0172]
步骤4、框架实时调整计算图,并分配新的存储空间给新增添的专家e4,并继续训练。
[0173]
通过上述方式,可以允许模型的记忆模块进行伸缩,以针对不同场景,扩展和裁剪模型容量。
[0174]
502、从所述存储地址对应的存储空间中获取所述一个或多个目标网络单元;所述一个或多个目标网络单元用于构建神经网络。
[0175]
在一种可能的实现中,在获取到存储地址后,可以从存储地址对应的存储空间中获取所述一个或多个目标网络单元,并利用该一个或多个目标网络单元进行神经网络的构
建,具体的,一个或多个目标网络单元可以用于进行目标网络单元对应位置的网络结构的构建。
[0176]
具体流程的一个示意可以如下图8所示,主要由以下步骤组成:
[0177]
步骤1:当一个迭代的训练开始的时候,控制模块读取本次迭代的数据。
[0178]
步骤2:根据上一步读取的数据,在多值映射表中,以数据为key,查找对应的专家的编号。
[0179]
步骤3:分两步,先根据上一步查找到的专家的编号,找到对应专家的内存地址。然后3.2,根据找到的内存地址,将需要使用的专家提取出来,或者将不使用的专家冻结。这种提取可以是在新的内存空间,构建计算子图,将存储单元中对应的专家拷贝到新的计算空间,也可以是在存储单元中,用虚拟的内存映射管理存储单元中对应的专家做为计算单元。而这种冻结,会将冻结操从计算图上传播给对应的反向计算节点和优化器更新节点,确保在本轮中不需要计算的专家,正向反向和优化器更新都不需要参与计算。
[0180]
503、根据所述目标词向量,通过所述计算单元执行所述神经网络对应的训练过程。
[0181]
在一种可能的实现中,所述通过所述计算单元执行所述目标神经网络对应的训练过程用于得到更新后的目标神经网络,所述更新后的目标神经网络包括更新后的所述一个或多个目标网络单元,根据所述更新后的所述一个或多个目标网络单元,更新所述存储地址对应的存储空间中的数据,以便下一次调用。
[0182]
参照图8,本技术实施例的一个可选实现步骤可以包括:
[0183]
步骤4:利用上一步构建的计算单元,将步骤1读入的数据和对应的专家做实际的计算,前向转播算出loss,并做反向计算,以及优化器更新专家参数。
[0184]
步骤5:将计算单元中更新的专家参数,和存储单元中对应的专家参数同步。
[0185]
上述步骤的思想在于,通过内存管理,将稀疏化的专家记忆单元,也就是存储单元中的专家找出来,并确保只计算这一部分激活的专家,其余不需要计算的不参与任何前向,反向,优化器更新的计算,保证稀疏化的记忆网络在计算中没有冗余,从而提到训练效率。
[0186]
在一种可能的实现中,所述所述目标映射关系包括多个词单元以及每个词单元对应的网络单元,还可以根据收敛后所述神经网络,从所述目标映射关系中包括的多个词单元中确定部分词单元对应的网络单元,所述收敛后所述神经网络中所述部分词单元对应的网络单元用于构建目标神经网络;所述目标神经网络用于进行模型推理。
[0187]
上述过程可以称之为子模型的抽取,示例性的流程可以参照图10所示。
[0188]
本技术实施例提出了一种新型的可伸缩的记忆网络设计,同其高效训练算法。保证能够在高效训练由这样的网络模块构成的大模型的同时,使得模型通过记忆模块的伸缩,获得持续学习能力,保证模型一直处于最新的知识状态。系统架构如下图11所示:
[0189]
一是多值映射表(multi-value mapping)。该模块是本专利可行的数据结构基础。和上面相关工作中提到的散列不同,映射是一种直接存储的键值对关系,不需要通过散列函数去映射到一个固定的索引范围,因此没有碰撞问题。这就决定了用映射数据结构构建的关系,不会丢失关键信息。而这里采用多值映射,就是构建一个键值和集合的映射对,集合中可以是任意多的元素,这样就保证了我们的映射关系不仅仅是单一的。本专利中的多值映射表,支持多种操作,如查找,插入和删除等修改。这些操作和下面将要描述的控制模
块配合,实现了我们专利描述的训练算法和终身学习算法。
[0190]
二是控制模块。控制模块如下图虚线控制模块所示,控制模块是本系统的核心,具体有两个主要功能:
[0191]
一是为完成终身学习能力设计的功能,控制模块从损失或者某个度量函数获得信息后,如果判断需要增加记忆网络的数量,算法走入“是”分支,存储控制单元首先会修改多值映射表,增添或者删除映射内的键值对,然后存储控制单元会在存储子图部分开辟或者释放对应的内存空间。
[0192]
二是为了实现高性能训练设计的功能,如果判断为否,也就是说不需要进行记忆网络的增添和删除,算法将会走到“否”分支。在这个分支,控制模块的路由能够通过读取当前数据单元的数据,查询多值映射表中的映射关系,得知本次需要计算的记忆网络的编号,并根据这个查询结果,计算控制模块将需要计算的子图通过某种方式提取出来,可以是拉取,也可以是构建一个新的子图,也可以是在存储子图的本地,对不计算的部分进行冻结,确保只计算需要计算的部分。并且这个操作将会传播给计算图的所有相关计算部分,在大规模分布式场景中,将不进行对应的梯度传输,从而大大减少整个模型的计算量和通信量。
[0193]
三是记忆网络模块。就是图中的存储单元和计算单元,记忆模块由线性网络构成,一般有一个以上的线性网络构成,每个线性网络有一个编号,比如图中的a,b,c,d,用于区分不同的线性网络,这个编号也是多值映射表中的键值对中的值的元素,控制装置可以通过这个值找到对应的记忆网络的地址,并对对应的网络进行提取,拷贝,删除,新增等操作。每个网络记忆了当前状态训练实体对应的本层特征。
[0194]
四是数据单元。可以是训练数据,也可以是某种经过编码的中间结果。
[0195]
接下来从装置的角度介绍本技术实施例提供的一种数据处理装置,参照图12,图12为本技术实施例提供的一种数据处理装置的结构示意,如图12所示,本技术实施例提供的一种数据处理装置1200包括:
[0196]
处理模块1201,用于根据目标映射关系,确定目标词向量token对应的一个或多个目标网络单元以及所述一个或多个目标网络单元的存储地址,所述存储地址对应的存储空间位于计算单元之外的存储空间;
[0197]
从所述存储地址对应的存储空间中获取所述一个或多个目标网络单元;所述一个或多个目标网络单元用于构建神经网络;
[0198]
其中,关于处理模块1201的具体描述,可以参照上述实施例中步骤501和步骤502的介绍,这里不再赘述。
[0199]
模型训练模块1202,用于根据所述目标词向量,通过所述计算单元执行所述神经网络对应的训练过程。
[0200]
其中,关于模型训练模块1202的具体描述,可以参照上述实施例中步骤503的介绍,这里不再赘述。
[0201]
在一种可能的实现中,所述目标网络单元为前向传播网络ffn。
[0202]
在一种可能的实现中,所述计算单元为图形处理器gpu、张量处理器tpu或者神经网络处理器npu。
[0203]
在一种可能的实现中,所述存储地址对应的存储空间位于内存、网络存储器或者硬盘。
access memory,nvram)。存储器1304存储有处理器和操作指令、可执行模块或者数据结构,或者它们的子集,或者它们的扩展集,其中,操作指令可包括各种操作指令,用于实现各种操作。
[0219]
处理器1303控制执行设备的操作。具体的应用中,执行设备的各个组件通过总线系统耦合在一起,其中总线系统除包括数据总线之外,还可以包括电源总线、控制总线和状态信号总线等。但是为了清楚说明起见,在图中将各种总线都称为总线系统。
[0220]
上述本技术实施例揭示的方法可以应用于处理器1303中,或者由处理器1303实现。处理器1303可以是一种集成电路芯片,具有信号的处理能力。在实现过程中,上述方法的各步骤可以通过处理器1303中的硬件的集成逻辑电路或者软件形式的指令完成。上述的处理器1303可以是通用处理器、数字信号处理器(digital signal processing,dsp)、微处理器或微控制器、以及视觉处理器(vision processing unit,vpu)、张量处理器(tensor processing unit,tpu)等适用于ai运算的处理器,还可进一步包括专用集成电路(application specific integrated circuit,asic)、现场可编程门阵列(field-programmable gate array,fpga)或者其他可编程逻辑器件、分立门或者晶体管逻辑器件、分立硬件组件。该处理器1303可以实现或者执行本技术实施例中的公开的各方法、步骤及逻辑框图。通用处理器可以是微处理器或者该处理器也可以是任何常规的处理器等。结合本技术实施例所公开的方法的步骤可以直接体现为硬件译码处理器执行完成,或者用译码处理器中的硬件及软件模块组合执行完成。软件模块可以位于随机存储器,闪存、只读存储器,可编程只读存储器或者电可擦写可编程存储器、寄存器等本领域成熟的存储介质中。该存储介质位于存储器1304,处理器1303读取存储器1304中的信息,结合其硬件完成上述实施例中步骤501至步骤503的步骤。
[0221]
接收器1301可用于接收输入的数字或字符信息,以及产生与执行设备的相关设置以及功能控制有关的信号输入。发射器1302可用于通过第一接口输出数字或字符信息;发射器1302还可用于通过第一接口向磁盘组发送指令,以修改磁盘组中的数据;发射器1302还可以包括显示屏等显示设备。
[0222]
本技术实施例还提供了一种训练设备,请参阅图14,图14是本技术实施例提供的训练设备一种结构示意图,具体的,训练设备1400可以为上述实施例介绍的数据处理装置,训练设备1400由一个或多个服务器实现,训练设备1400可因配置或性能不同而产生比较大的差异,可以包括一个或一个以上中央处理器(central processing units,cpu)1414(例如,一个或一个以上处理器)和存储器1432,一个或一个以上存储应用程序1442或数据1444的存储介质1430(例如一个或一个以上海量存储设备)。其中,存储器1432和存储介质1430可以是短暂存储或持久存储。存储在存储介质1430的程序可以包括一个或一个以上模块(图示没标出),每个模块可以包括对训练设备中的一系列指令操作。更进一步地,中央处理器1414可以设置为与存储介质1430通信,在训练设备1400上执行存储介质1430中的一系列指令操作。
[0223]
训练设备1400还可以包括一个或一个以上电源1426,一个或一个以上有线或无线网络接口1450,一个或一个以上输入输出接口1458;或,一个或一个以上操作系统1441,例如windows servertm,mac os xtm,unixtm,linuxtm,freebsdtm等等。
[0224]
本技术实施例中还提供一种包括计算机可读指令的计算机程序产品,当其在计算
机上运行时,使得计算机执行如前述执行设备所执行的步骤,或者,使得计算机执行如前述训练设备所执行的步骤。
[0225]
本技术实施例中还提供一种计算机可读存储介质,该计算机可读存储介质中存储有用于进行信号处理的程序,当其在计算机上运行时,使得计算机执行如前述执行设备所执行的步骤,或者,使得计算机执行如前述训练设备所执行的步骤。
[0226]
本技术实施例提供的执行设备、训练设备或终端设备具体可以为芯片,芯片包括:处理单元和通信单元,所述处理单元例如可以是处理器,所述通信单元例如可以是输入/输出接口、管脚或电路等。该处理单元可执行存储单元存储的计算机执行指令,以使执行设备内的芯片执行上述实施例描述的模型训练方法,或者,以使训练设备内的芯片执行上述实施例中与模型训练相关的步骤。可选地,所述存储单元为所述芯片内的存储单元,如寄存器、缓存等,所述存储单元还可以是所述无线接入设备端内的位于所述芯片外部的存储单元,如只读存储器(read-only memory,rom)或可存储静态信息和指令的其他类型的静态存储设备,随机存取存储器(random access memory,ram)等。
[0227]
具体的,请参阅图15,图15为本技术实施例提供的芯片的一种结构示意图,所述芯片可以表现为神经网络处理器npu 1500,npu 1500作为协处理器挂载到主cpu(host cpu)上,由host cpu分配任务。npu的核心部分为运算电路1503,通过控制器1504控制运算电路1503提取存储器中的矩阵数据并进行乘法运算。
[0228]
在一些实现中,运算电路1503内部包括多个处理单元(process engine,pe)。在一些实现中,运算电路1503是二维脉动阵列。运算电路1503还可以是一维脉动阵列或者能够执行例如乘法和加法这样的数学运算的其它电子线路。在一些实现中,运算电路1503是通用的矩阵处理器。
[0229]
举例来说,假设有输入矩阵a,权重矩阵b,输出矩阵c。运算电路从权重存储器1502中取矩阵b相应的数据,并缓存在运算电路中每一个pe上。运算电路从输入存储器1501中取矩阵a数据与矩阵b进行矩阵运算,得到的矩阵的部分结果或最终结果,保存在累加器(accumulator)1508中。
[0230]
统一存储器1506用于存放输入数据以及输出数据。权重数据直接通过存储单元访问控制器(direct memory access controller,dmac)1505,dmac被搬运到权重存储器1502中。输入数据也通过dmac被搬运到统一存储器1506中。
[0231]
biu为bus interface unit即,总线接口单元1510,用于axi总线与dmac和取指存储器(instruction fetch buffer,ifb)1509的交互。
[0232]
总线接口单元1510(bus interface unit,简称biu),用于取指存储器1509从外部存储器获取指令,还用于存储单元访问控制器1505从外部存储器获取输入矩阵a或者权重矩阵b的原数据。
[0233]
dmac主要用于将外部存储器ddr中的输入数据搬运到统一存储器1506或将权重数据搬运到权重存储器1502中或将输入数据数据搬运到输入存储器1501中。
[0234]
向量计算单元1507包括多个运算处理单元,在需要的情况下,对运算电路的输出做进一步处理,如向量乘,向量加,指数运算,对数运算,大小比较等等。主要用于神经网络中非卷积/全连接层网络计算,如batch normalization(批归一化),像素级求和,对特征平面进行上采样等。
[0235]
在一些实现中,向量计算单元1507能将经处理的输出的向量存储到统一存储器1506。例如,向量计算单元1507可以将线性函数;或,非线性函数应用到运算电路1503的输出,例如对卷积层提取的特征平面进行线性插值,再例如累加值的向量,用以生成激活值。在一些实现中,向量计算单元1507生成归一化的值、像素级求和的值,或二者均有。在一些实现中,处理过的输出的向量能够用作到运算电路1503的激活输入,例如用于在神经网络中的后续层中的使用。
[0236]
控制器1504连接的取指存储器(instruction fetch buffer)1509,用于存储控制器1504使用的指令;
[0237]
统一存储器1506,输入存储器1501,权重存储器1502以及取指存储器1509均为on-chip存储器。外部存储器私有于该npu硬件架构。
[0238]
其中,上述任一处提到的处理器,可以是一个通用中央处理器,微处理器,asic,或一个或多个用于控制上述程序执行的集成电路。
[0239]
另外需说明的是,以上所描述的装置实施例仅仅是示意性的,其中所述作为分离部件说明的单元可以是或者也可以不是物理上分开的,作为单元显示的部件可以是或者也可以不是物理单元,即可以位于一个地方,或者也可以分布到多个网络单元上。可以根据实际的需要选择其中的部分或者全部模块来实现本实施例方案的目的。另外,本技术提供的装置实施例附图中,模块之间的连接关系表示它们之间具有通信连接,具体可以实现为一条或多条通信总线或信号线。
[0240]
通过以上的实施方式的描述,所属领域的技术人员可以清楚地了解到本技术可借助软件加必需的通用硬件的方式来实现,当然也可以通过专用硬件包括专用集成电路、专用cpu、专用存储器、专用元器件等来实现。一般情况下,凡由计算机程序完成的功能都可以很容易地用相应的硬件来实现,而且,用来实现同一功能的具体硬件结构也可以是多种多样的,例如模拟电路、数字电路或专用电路等。但是,对本技术而言更多情况下软件程序实现是更佳的实施方式。基于这样的理解,本技术的技术方案本质上或者说对现有技术做出贡献的部分可以以软件产品的形式体现出来,该计算机软件产品存储在可读取的存储介质中,如计算机的软盘、u盘、移动硬盘、rom、ram、磁碟或者光盘等,包括若干指令用以使得一台计算机设备(可以是个人计算机,训练设备,或者网络设备等)执行本技术各个实施例所述的方法。
[0241]
在上述实施例中,可以全部或部分地通过软件、硬件、固件或者其任意组合来实现。当使用软件实现时,可以全部或部分地以计算机程序产品的形式实现。
[0242]
所述计算机程序产品包括一个或多个计算机指令。在计算机上加载和执行所述计算机程序指令时,全部或部分地产生按照本技术实施例所述的流程或功能。所述计算机可以是通用计算机、专用计算机、计算机网络、或者其他可编程装置。所述计算机指令可以存储在计算机可读存储介质中,或者从一个计算机可读存储介质向另一计算机可读存储介质传输,例如,所述计算机指令可以从一个网站站点、计算机、训练设备或数据中心通过有线(例如同轴电缆、光纤、数字用户线(dsl))或无线(例如红外、无线、微波等)方式向另一个网站站点、计算机、训练设备或数据中心进行传输。所述计算机可读存储介质可以是计算机能够存储的任何可用介质或者是包含一个或多个可用介质集成的训练设备、数据中心等数据存储设备。所述可用介质可以是磁性介质,(例如,软盘、硬盘、磁带)、光介质(例如,dvd)、或
者半导体介质(例如固态硬盘(solid state disk,ssd))等。
当前第1页1 2 
网友询问留言 已有0条留言
  • 还没有人留言评论。精彩留言会获得点赞!
1