机器学习模型的联合训练方法、装置、设备及存储介质
技术领域
本公开涉及计算机技术领域,尤其涉及一种机器学习模型的联合训练方法、机器学习模型的联合训练装置、电子设备与计算机可读存储介质。
背景技术
机器学习作为人工智能领域的核心技术,其发展状况很大程度上决定了人工智能的发展程度。目前,随着机器学习与大数据的结合,机器学习模型日益复杂化,训练模型所需的数据量与运算量大大增加,在此情况下,出现了由多台服务器或终端进行联合训练的方法。然而,现有的联合训练方法普遍存在数据泄露的安全隐患,限制了其在机器学习技术上的应用。
需要说明的是,在上述背景技术部分公开的信息仅用于加强对本公开的背景的理解,因此可以包括不构成对本领域普通技术人员已知的现有技术的信息。
发明内容
本公开提供了一种机器学习模型的联合训练方法、机器学习模型的联合训练装置、电子设备与计算机可读存储介质,进而至少在一定程度上克服现有的联合训练方法存在安全隐患的问题。
本公开的其他特性和优点将通过下面的详细描述变得显然,或部分地通过本公开的实践而习得。
根据本公开的第一方面,提供一种机器学习模型的联合训练方法,应用于联合训练***,所述联合训练***包括数据节点和标签节点;所述方法包括:所述数据节点获取用于训练的样本数据;所述数据节点通过目标模型的第一部分处理所述样本数据,得到中间数据,并将所述中间数据发送至所述标签节点;所述标签节点通过所述目标模型的第二部分处理所述中间数据,并基于所述样本数据对应的标签数据,得到误差数据;所述标签节点将所述误差数据发送至所述数据节点;所述数据节点根据所述误差数据调整所述第一部分的参数,和/或所述标签节点根据所述误差数据调整所述第二部分的参数;其中,所述目标模型由所述第一部分和所述第二部分组成。
在本公开的一种示例性实施例中,所述得到误差数据之后,所述方法还包括:所述标签节点根据所述误差数据判断是否达到预设条件;如果达到所述预设条件,则所述标签节点向所述数据节点发送训练结束信息,以当前的所述第一部分和所述第二部分确定所述目标模型;如果未达到所述预设条件,则所述标签节点执行将所述误差数据发送至所述数据节点的步骤。
在本公开的一种示例性实施例中,所述方法还包括:所述标签节点向所述数据节点发送数据索引信息;所述数据节点获取用于训练的样本数据,包括:所述数据节点根据所述数据索引信息获取所述样本数据。
在本公开的一种示例性实施例中,所述数据节点通过目标模型的第一部分处理所述样本数据,得到中间数据,并将所述中间数据发送至所述标签节点,包括:所述数据节点通过目标模型的第一部分处理所述样本数据,得到中间数据,并将加密后的所述中间数据发送至所述标签节点。
在本公开的一种示例性实施例中,所述标签节点将所述误差数据发送至所述数据节点,包括:所述标签节点将添加噪声的误差数据发送至所述数据节点。
在本公开的一种示例性实施例中,所述中间数据采用同态加密。
根据本公开的第二方面,提供一种机器学习模型的联合训练方法,应用于联合训练***的数据节点,所述联合训练***还包括标签节点;所述方法包括:获取用于训练的样本数据;通过目标模型的第一部分处理所述样本数据,得到中间数据;将所述中间数据发送至所述标签节点;如果从所述标签节点接收到误差数据,则根据所述误差数据调整所述第一部分的参数。
在本公开的一种示例性实施例中,所述方法还包括:如果从所述标签节点接收到训练结束信息,则将当前的所述第一部分确定为所述目标模型最终的第一部分。
在本公开的一种示例性实施例中,所述方法还包括:从所述标签节点获取数据索引信息;所述获取用于训练的样本数据,包括:根据所述数据索引信息获取所述样本数据。
在本公开的一种示例性实施例中,所述将所述中间数据发送至所述标签节点,包括:将加密后的所述中间数据发送至所述标签节点。
在本公开的一种示例性实施例中,所述如果从所述标签节点接收到误差数据,则根据所述误差数据调整所述第一部分的参数,包括:如果从所述标签节点接收到包含噪声的误差数据,则根据所述包含噪声的误差数据调整所述第一部分的参数。
在本公开的一种示例性实施例中,所述中间数据采用同态加密。
根据本公开的第三方面,提供一种机器学习模型的联合训练方法,应用于联合训练***的标签节点,所述联合训练***还包括数据节点;所述方法包括:从所述数据节点接收中间数据;通过目标模型的第二部分处理所述中间数据,并基于数据索引信息对应的标签数据得到误差数据;将所述误差数据发送至所述数据节点;根据所述误差数据调整所述目标模型的第二部分的参数。
在本公开的一种示例性实施例中,所述基于数据索引信息对应的标签数据得到误差数据之后,所述方法还包括:根据所述误差数据判断是否达到预设条件;如果达到所述预设条件,则向所述数据节点发送训练结束信息,并将当前的所述第二部分确定为所述目标模型最终的第二部分;如果未达到所述预设条件,则执行将所述误差数据发送至所述数据节点的步骤。
在本公开的一种示例性实施例中,所述从所述数据节点接收中间数据之前,所述方法还包括:向所述数据节点发送所述数据索引信息,使所述数据节点通过所述目标模型的第一部分处理所述数据索引信息对应的样本数据以得到所述中间数据。
在本公开的一种示例性实施例中,所述从所述数据节点接收中间数据,包括:从所述数据节点接收加密后的所述中间数据。
在本公开的一种示例性实施例中,所述将所述误差数据发送至所述数据节点,包括:将包含噪声的误差数据发送至所述数据节点。
在本公开的一种示例性实施例中,所述中间数据采用同态加密。
根据本公开的第四方面,提供一种机器学习模型的联合训练装置,应用于联合训练***的数据节点,所述联合训练***还包括标签节点;所述装置包括:获取模块,用于获取用于训练的样本数据;处理模块,用于通过目标模型的第一部分处理所述样本数据,得到中间数据;发送模块,用于将所述中间数据发送至所述标签节点;调整模块,用于如果从所述标签节点接收到误差数据,则根据所述误差数据调整所述第一部分的参数。
在本公开的一种示例性实施例中,所述装置还包括:确定模块,用于如果从所述标签节点接收到训练结束信息,则将当前的所述第一部分确定为所述目标模型最终的第一部分。
在本公开的一种示例性实施例中,所述获取模块用于从所述标签节点获取数据索引信息,并根据所述数据索引信息获取所述样本数据。
在本公开的一种示例性实施例中,所述发送模块用于将加密后的所述中间数据发送至所述标签节点。
在本公开的一种示例性实施例中,所述调整模块用于如果从所述标签节点接收到包含噪声的误差数据,则根据所述包含噪声的误差数据调整所述第一部分的参数。
在本公开的一种示例性实施例中,所述中间数据采用同态加密。
根据本公开的第五方面,提供一种机器学习模型的联合训练装置,应用于联合训练***的标签节点,所述联合训练***还包括数据节点;所述装置包括:接收模块,用于从所述数据节点接收中间数据;处理模块,用于通过目标模型的第二部分处理所述中间数据,并基于数据索引信息对应的标签数据得到误差数据;发送模块,用于将所述误差数据发送至所述数据节点;调整模块,用于根据所述误差数据调整所述目标模型的第二部分的参数。
在本公开的一种示例性实施例中,所述处理模块还用于根据所述误差数据判断是否达到预设条件;所述发送模块还用于如果达到所述预设条件,则向所述数据节点发送训练结束信息,并将当前的所述第二部分确定为所述目标模型最终的第二部分,以及如果未达到所述预设条件,则将所述误差数据发送至所述数据节点。
在本公开的一种示例性实施例中,所述发送模块还用于向所述数据节点发送所述数据索引信息,使所述数据节点通过所述目标模型的第一部分处理所述数据索引信息对应的样本数据以得到所述中间数据。
在本公开的一种示例性实施例中,所述接收模块用于从所述数据节点接收加密后的所述中间数据。
在本公开的一种示例性实施例中,所述发送模块用于将包含噪声的误差数据发送至所述数据节点。
在本公开的一种示例性实施例中,所述中间数据采用同态加密。
根据本公开的第六方面,提供一种电子设备,包括:处理器;以及存储器,用于存储所述处理器的可执行指令;其中,所述处理器配置为经由执行所述可执行指令来执行上述任意一项所述的方法。
根据本公开的第七方面,提供一种计算机可读存储介质,其上存储有计算机程序,所述计算机程序被处理器执行时实现上述任意一项所述的方法。
本公开的示例性实施例具有以下有益效果:
联合训练***的数据节点提供用于训练的样本数据,标签节点提供标签数据,数据节点通过目标模型的第一部分得到中间数据,标签节点通过目标模型的第二部分得到误差数据,两方节点再分别根据误差数据进行参数调整,以实现目标模型的联合训练。一方面,提出了一种有利于隐私保护的联合训练方法,由数据节点与标签节点分别持有训练所需的一个方面的数据以及目标模型的一个部分,且一方的数据以及模型对于另一方不可见,因此保障了数据的隐私性与安全性,可以满足多种业务场景的需求,提高联合训练方法的普遍适用性。另一方面,在训练过程中,数据节点与标签节点对于数据进行同步的分工处理,并分别根据误差数据进行参数调整,从而将传统由单个主机进行的模型训练任务分散到了两个或多个主机上,整体的运算能力得到提升,有利于提高效率。
应当理解的是,以上的一般描述和后文的细节描述仅是示例性和解释性的,并不能限制本公开。
附图说明
此处的附图被并入说明书中并构成本说明书的一部分,示出了符合本公开的实施例,并与说明书一起用于解释本公开的原理。显而易见地,下面描述中的附图仅仅是本公开的一些实施例,对于本领域普通技术人员来讲,在不付出创造性劳动的前提下,还可以根据这些附图获得其他的附图。
图1示出本示例性实施例的一种联合训练***的架构图;
图2示出本示例性实施例中一种联合训练方法的流程步骤图;
图3示出本示例性实施例中一种联合训练方法的流程示意图;
图4示出本示例性实施例中目标模型第一部分的示意图;
图5示出本示例性实施例中另一种联合训练方法的流程步骤图;
图6示出本示例性实施例中再一种联合训练方法的流程步骤图;
图7示出本示例性实施例中一种联合训练装置的结构框图;
图8示出本示例性实施例中另一种联合训练装置的结构框图;
图9示出本示例性实施例中一种用于实现上述方法的电子设备;
图10示出本示例性实施例中一种用于实现上述方法的计算机可读存储介质。
具体实施方式
现在将参考附图更全面地描述示例实施方式。然而,示例实施方式能够以多种形式实施,且不应被理解为限于在此阐述的范例;相反,提供这些实施方式使得本公开将更加全面和完整,并将示例实施方式的构思全面地传达给本领域的技术人员。所描述的特征、结构或特性可以以任何合适的方式结合在一个或更多实施方式中。
本公开的示例性实施例首先提供了一种机器学习模型的联合训练方法,图1示出了可以运行该方法的联合训练***的架构图。如图1所示,该联合训练***100可以包括数据节点110、标签节点120与网络130,数据节点110用于提供训练所需的样本数据,标签节点120用于提供样本数据对应的标签数据,在本文中,标签数据指样本数据的标签或标注。数据节点110与标签节点120之间通过网络130连接,进行数据交互。可见,数据节点110与标签节点120共同提供了训练所需的数据,任何一方无法单独实行训练,这也决定了联合训练***100是一种双侧的结构,例如:数据节点110是服务器,标签节点120是用户终端;数据节点110是公有云,标签节点120是私有云,等等。
应当理解,图1所示的各节点数目仅是示例性的,数据节点110或标签节点120都可以是由多个实体设备形成的集群,例如数据节点110为分布式的服务器集群,标签节点120为大量的用户终端,本公开对此不做特别限定。
图2示出了本示例性实施例的方法的流程步骤图,其执行主体可以是图1的联合训练***100。如图2所示,该联合训练方法可以包括步骤S210~S250:
步骤S210中,数据节点获取用于训练的样本数据。
其中,根据应用场景与实际需求的不同,样本数据可以是用户行为数据、图像数据、文本数据等各种类型的数据,用于在模型训练中作为输入数据。数据节点可以在相应的数据库中获取样本数据。
图3示出了一种联合训练方法的流程示意图,如图3所示,在一示例性实施例中,联合训练方法还可以包括步骤S205:标签节点向数据节点发送数据索引信息;相应的,步骤S210可以是:数据节点根据数据索引信息获取样本数据。
其中,数据索引信息用于标识训练所需的数据,对于数据节点,数据索引信息可以关联样本数据,对于标签节点,数据索引信息可以关联标签数据,因此,通过数据索引信息,可以在数据节点与标签节点之间建立样本数据与标签数据的关联对应。数据索引信息可以是用户唯一标识,如用户账号、移动终端的IMEI(International Mobile EquipmentIdentity,国际移动设备识别码)、手机号等,也可以是其他形式的数据标识等,举例而言,以用户唯一标识为数据索引信息,数据节点提供该用户的行为数据作为样本数据,标签节点提供该用户的真实标签作为标签数据,通过用户唯一标识使得样本数据对应于标签数据。
联合训练过程可以由数据节点和标签节点中的任一方发起,例如:数据节点启动对于机器学习模型的联合训练,向标签节点发送请求,标签节点响应于该请求,启动用于联合训练的进程,通过该进程向数据节点返回数据索引信息;或者标签节点首先启动用于联合训练的进程,向数据节点发送请求,该请求中可以包含数据索引信息,数据节点接收到该请求,从中提取出数据索引信息;或者联合训练过程由第三方发起,第三方同时向数据节点与标签节点发送数据索引信息,数据节点与标签节点同时启动用于联合训练的进程,进入图2所示的步骤。
在一示例性实施例中,为了保障信息的安全性,标签节点可以将数据索引信息加密后发送至数据节点,举例说明:可以预先在数据节点与标签节点上配置密钥,则数据节点通过解密得到数据索引信息;数据节点与标签节点配置相同的哈希加密算法,标签节点将数据索引信息通过哈希加密后发送至数据节点,数据节点获取哈希值后,在本地的数据库中查找出相同的哈希值所对应的数据索引信息,即为标签节点发送的数据索引信息。
步骤S220中,数据节点通过目标模型的第一部分处理样本数据,得到中间数据,并将中间数据发送至标签节点。
本示例性实施例中,目标模型为需要进行联合训练的模型,其由第一部分与第二部分组成,第一部分配置于数据节点上,通常为目标模型的前端部分,第二部分配置于标签节点上,通常为目标模型的后端部分,第一部分的输出可以与第二部分的输入相连接。可见,目标模型是由两方节点共同持有,任一方无法单独获得整个模型,从而在一定程度上保障了数据在两方节点之间的隐私性与安全性。
基于目标模型的第一部分与第二部分,对数据的处理过程也可以看作是两个阶段,第一部分的处理过程为第一阶段,第二部分的处理过程为第二阶段,第一阶段与第二阶段为连续的两个阶段。以目标模型是神经网络模型为例进行说明,如图4所示,第一部分可以是只有输入层的部分,也可以包括输入层以及若干中间层,即中间数据可以是样本数据本身,也可以是对样本数据进行特征提取后的数据,然后通过网络传输发送至标签节点。无论是哪种情况,对于标签节点而言,其并未获得目标模型第一部分的具体结构与参数,因此无法通过中间数据倒推得到样本数据,即标签节点无法获知样本数据。
步骤S230中,标签节点通过目标模型的第二部分处理中间数据,并基于样本数据对应的标签数据,得到误差数据。
其中,目标模型的第二部分包括输出端,将中间数据输入第二部分,可以得到对应的输出结果,计算输出结果与标签数据的差值,可以得到误差数据。标签数据为预先获取的,与样本数据具有对应关系,可以通过数据索引信息在两方节点之间统一样本数据与标签数据的关联,例如标签节点在启动用于联合训练的进程时,向数据节点发送数据索引信息,或者数据节点在发送中间数据时,将数据索引信息一起发送,又或者中间数据本身具有索引信息(该索引信息通常来自于样本数据),可通过索引信息确定与其关联的标签数据。需要说明的是,实际应用中标签数据的数量通常较多,每个标签数据对应于一组样本数据,在计算误差数据时,可以基于多个标签数据,通过以下交叉熵公式计算损失函数,以损失函数的形式表征误差。
Loss=(-∑iyilog(1/pi)),Loss为损失函数,pi表示第i个输出结果,是通过目标模型的第二部分处理第i组样本数据所得到的,yi表示第i个标签数据,与pi之间具有对应关系。
需要补充的是,除了上述输出结果与标签数据的差值或损失函数,误差数据还可以包括基于该差值或损失函数所计算的模型各参数的误差权重值、模型各部分的梯度值等,误差数据的具体内容与模型的具体训练方式相关,本公开对此不做特别限定。
步骤S240中,标签节点将误差数据发送至数据节点。
本示例性实施例中,标签节点可以将表示模型输出结果与标签数据差值的误差数据发送至数据节点,也可以将模型各参数的误差权重值、模型各部分的梯度值等发送至数据节点。
对于数据节点而言,由于其并未获得目标模型第二部分的具体结构与参数,根据误差数据无法推得标签数据,即数据节点无法获知标签数据。可见,本示例性实施例中,数据节点持有样本数据,标签节点持有标签数据,在联合训练的过程中,两方节点无法获知对方所持有的数据,从而保障了数据的隐私性与安全性。
步骤S250中,数据节点根据误差数据调整第一部分的参数,和/或标签节点根据误差数据调整第二部分的参数。
其中,调整参数即训练的过程,其目的是将目标模型的参数优化调整到一定的状态,使得误差数据小于预定值甚至为0。在一次调整中,可以仅仅调整第一部分或第二部分的参数,也可以同时调整两个部分的参数,这与目标模型训练的具体进度相关。根据目标模型与误差数据的具体形式,可以采用局部随机调整、梯度下降等方式进行参数调整,本公开对此不做特别限定。
需要说明的是,步骤S210~S250为联合训练过程中一次参数调整的完整过程,训练过程可能包含了多次参数调整的过程,则可以将多次执行其中的一个或多个步骤,例如:对于每一批(batch,如64或128组)样本数据和标签数据,执行一次步骤S220~S250,以进行一次参数调整。此外,训练过程通常还可以包括训练与验证的过程,例如:将样本数据划分训练集与验证集,对标签数据也同样划分训练集与验证集,利用训练集的样本数据与标签数据进行训练,执行一次或多次步骤S210~S250,再利用验证集的样本数据与标签数据进行验证,执行步骤S210~S230。应当理解,上述多种情况都属于本公开的保护范围。
在一示例性实施例中,参考上述图3所示,在步骤S230后,还可以包括以下步骤S231~S233:
步骤S231,标签节点根据误差数据判断是否达到预设条件;
步骤S232,如果达到预设条件,则标签节点向数据节点发送训练结束信息,以当前的第一部分和第二部分确定目标模型;
步骤S233,如果未达到预设条件,则标签节点执行步骤S240。
其中,预设条件是指判断目标模型是否训练完成的条件,例如可以是目标模型是否收敛、准确率是否达标等。可以根据误差数据判断是否达到预设条件,举例说明:判断误差数据是否小于预定值,如果小于,则目标模型的参数已经收敛;或者分别判断多个标签数据上的误差是否小于预定值,如果超过一定比例的误差小于预定值,则目标模型的准确率达标,等等。如果达到预设条件,表示目标模型训练完成,标签节点将该信息发送至数据节点,以告知数据节点训练完成,此时不再调整目标模型的参数,数据节点所持有的第一部分与标签节点所持有的第二部分共同构成最终的目标模型,任一方无法单独获得完整的模型。如果未达到预设条件,说明模型还需要进一步优化,可以执行步骤S240与S250以进行参数调整。
基于图3所示的循环形式的方法流程,可以实现机器学习模型联合训练的完整过程。下面对本实施例的应用场景做示例性说明,当然,下述示例并不能对本方案的应用范围造成限定:
数据节点为平台方(例如电子金融平台)的服务器,提供用户的行为数据,标签节点为业务合作方(例如银行、保险公司、征信机构等与电子金融平台有业务或数据共享需求的组织)的服务器,提供用户的真实信用数据,双方可以在实际不共享数据的情况下,联合训练机器学习模型,以满足用户信用评估、行为预测、产品预测等方面的需求,特别适用于样本数据或标签数据涉及到用户交易信息、存款信息等敏感数据的情况。数据节点为服务器,提供用户的在线日志数据,标签节点为用户终端,提供自己的真实用户画像,双方可以联合训练机器学习模型,以满足不同用户个性化标签下的业务需求,为不同的用户提供个性化的智能服务,在此过程中服务器无法获取到用户的标签数据,有利于用户的隐私保护。
在本示例性实施例中,联合训练***的数据节点提供用于训练的样本数据,标签节点提供标签数据,数据节点通过目标模型的第一部分得到中间数据,标签节点通过目标模型的第二部分得到误差数据,两方节点再分别根据误差数据进行参数调整,以实现目标模型的联合训练。一方面,提出了一种有利于隐私保护的联合训练方法,由数据节点与标签节点分别持有训练所需的一个方面的数据以及目标模型的一个部分,且一方的数据以及模型对于另一方不可见,因此保障了数据的隐私性与安全性,可以满足多种业务场景的需求,提高联合训练方法的普遍适用性。另一方面,在训练过程中,数据节点与标签节点对于数据进行同步的分工处理,并分别根据误差数据进行参数调整,从而将传统由单个主机进行的模型训练任务分散到了两个或多个主机上,整体的运算能力得到提升,有利于提高效率。
为了进一步提高数据的安全性,在一示例性实施例中,步骤S220可以通过以下步骤实现:
数据节点通过目标模型的第一部分处理样本数据,得到中间数据,并将加密后的中间数据发送至标签节点。
其中,加密可以通过多种具体方式实现,例如:预先在数据节点与标签节点上配置密钥,数据节点对中间数据加密,然后传输至标签节点,标签节点对中间数据解密,这样能提高数据传输过程的安全性,防止数据被窃取。
当然,上述配置密钥的方式是基于在数据节点与标签节点之间公开中间数据,也可以采用对标签节点不公开中间数据的形式,在一示例性实施例中,上述加密的过程可以采用同态加密,则标签节点接收到经过同态加密的中间数据的密文,通过目标模型的第二部分对该密文进行计算处理,不影响处理结果。例如:由于大部分机器学习模型的参数运算以加法与数乘为主,可以采用满足加法与数乘同态的加密算法,如Paillier算法、Gentry算法等;对于其他类型的机器学习模型,也可以采用RSA等其他形式的同态加密算法。
基于同态加密的方式,为了得到有效的误差数据,可以具体采用以下几种处理方法:利用非对称加密,在标签节点上仅配置公钥,其可以通过公钥对标签数据加密,并且从数据节点获取中间数据的密文,经过目标模型第二部分的计算后,将计算结果与标签数据的密文做差以得到误差数据;在数据传输的过程中添加噪声,这种处理方法将在下文具体说明。
相对应的,在一示例性实施例中,标签节点也可以将误差数据加密后发送至数据节点,类似于中间数据加密的过程,以提高误差数据的安全性。
在一示例性实施例中,也可以采用噪声的形式进行安全性的优化,步骤S240可以通过以下步骤实现:
标签节点将添加噪声的误差数据发送至数据节点。
其中,噪声是一个干扰值,其可以与误差数据进行加减乘除等操作,目的是对误差数据进行数值的微小改变,使得数据节点无法获得真实的误差数据,且对于后续计算的影响较小。数据节点在接收到包含噪声的误差数据后,根据该数据进行参数调整,可以在一定程度上增加目标模型的泛化能力,防止过拟合,同时进一步保障数据的隐私性。
基于上述中间数据加密以及误差数据添加噪声的方法,利用数据节点与标签节点之间的信息不对称性,通过算法上的设计,可以在两方节点互相数据不透明的情况下实现各自的计算目标:标签节点得到误差数据以及针对目标模型第二部分参数的梯度,数据节点得到目标模型第一部分参数的梯度。下面以目标模型是神经网络模型为例,对中间数据及误差数据的处理过程做示例性说明。
(1)隐私保护的正向传播过程,计算残差:
数据节点将样本数据x输入神经网络模型的第一部分,进行多层非线性变换:a=Act(W1×x+b)(Act()为激活函数,W,b为神经网络参数),得到中间数据a,利用Paillier算法对a加密,以[a]表示加密后的数据,将其发送至标签节点;
标签节点得到加密的[a]后,输入神经网络模型的第二部分,计算第二部分的参数
(该参数包含了∈
w噪声,因此用
表示,下文将具体为何包含∈
w噪声)与[a]之间的加权和:
并加入噪声∈
z,将计算结果
回传至数据节点;
数据节点对
解密,并去除
中的∈
w噪声以进行补偿计算:
计算
并传输至标签节点;
标签节点去除
中∈
z噪声
计算Softmax值
从而得到残差δ=p-y,残差即神经网络模型中的误差数据,y为标签数据。
(2)隐私保护的反向传播过程,计算梯度:
标签节点利用残差分别计算权重梯度[dw]与传播梯度
将[dw]添加噪声∈
g:
发送
至数据节点;
数据节点对
进行解密,在解密后的
中加入噪声
(η为设定的模型学习速率)以得到含有二层噪声的权重梯度
并连同积累的加密的噪声[∈
w]一起发送至标签节点;同时对噪声∈
w进行累加:
标签节点去除
中∈
g噪声:
得到含有一层噪声的梯度
即神经网络模型第二部分参数的梯度,因此可以进行第二部分参数的更新调整:
可见,第二部分的参数包含了∈
w噪声,因此表示为
从
中去除噪声
得到加密后的传播梯度[dx],将其发送至数据节点;
数据节点对[dx]进行解密得到dx,可以继续进行反向传播,最终得到第一部分参数的梯度。
通过上述方法过程,实现了通过反向传播调整目标模型的参数,且在数据节点与标签节点的数据传输中,通过加密与添加噪声的方式,对数据的隐私性起到了强保护的作用,同时保证了计算结果的有效性。
对于目标模型为神经网络模型的情况,在一示例性实施例中,可以采用一定的稀疏化处理方式,简化运算过程,达到加速的目的,例如:利用非线性激活函数ReLU=Max(0,x)为神经网络模型的激活函数,对激活值进行稀疏化;加入Dropout层,对激活值进行稀疏化,如可以在第一部分的最后一层加入。此外,在包含噪声计算的情况中,也可以相应的对噪声进行稀疏化处理,例如:初始化稀疏噪声矩阵∈g与∈z,设定矩阵中有一定比例的数值为0;对于累加噪声,设定累加后噪声矩阵∈w有一定比例的数值为0。当然,以上稀疏化处理的具体方式仅是为了示例性说明,本公开对此不做特别限定。
本公开的示例性实施例还提供了一种机器学习模型的联合训练方法,可以应用于图1中联合训练***100的数据节点110。如图5所示,该方法可以包括步骤S510~S540:
步骤S510,获取用于训练的样本数据;
步骤S520,通过目标模型的第一部分处理样本数据,得到中间数据;
步骤S530,将中间数据发送至标签节点;
步骤S540,如果从标签节点接收到误差数据,则根据误差数据调整第一部分的参数。
其中,步骤S510与图2中的步骤S210相同,步骤S520与S530与步骤S220相同,步骤S540与步骤S250中数据节点所执行的步骤相同,即步骤S510~S540的方法步骤为图2的方法中数据节点所执行的方法步骤,通过该方法,数据节点可以在未获知标签数据以及目标模型第二部分的情况下对目标模型的第一部分进行参数调整,从而在保护数据隐私性的前提下实现模型的联合训练。
在一示例性实施例中,联合训练方法还可以包括以下步骤:
如果从标签节点接收到训练结束信息,则将当前的第一部分确定为目标模型最终的第一部分。该过程可以参见图3中的步骤S232,接收到训练结束信息说明标签节点判断目标模型已经训练完成,则当前的第一部分即为最终的第一部分,目标模型由数据节点当前所持有的第一部分与标签节点当前所持有的第二部分共同确定。
在一示例性实施例中,联合训练方法还可以包括以下步骤:从标签节点获取数据索引信息;相应的,步骤S510可以通过以下步骤实现:根据数据索引信息获取样本数据。该过程可以参见图3中的步骤S205与S210。
在一示例性实施例中,步骤S530可以是:将加密后的中间数据发送至标签节点。
在一示例性实施例中,步骤S530可以是:如果从标签节点接收到包含噪声的误差数据,则根据包含噪声的误差数据调整第一部分的参数。
在一示例性实施例中,上述对中间数据的加密可以采用同态加密。
本公开的示例性实施例还提供了一种机器学习模型的联合训练方法,可以应用于图1中联合训练***100的标签节点120。如图6所示,该方法可以包括步骤S610~S640:
步骤S610,从数据节点接收中间数据;
步骤S620,通过目标模型的第二部分处理中间数据,并基于数据索引信息对应的标签数据得到误差数据;
步骤S630,将误差数据发送至数据节点;
步骤S640,根据误差数据调整目标模型的第二部分的参数。
其中,步骤S610与S620图2中的步骤S230相同,步骤S630与步骤S240相同,步骤S640与步骤S250中标签节点所执行的步骤相同,即步骤S610~S640的方法步骤为图2的方法中标签节点所执行的方法步骤,通过该方法,标签节点可以在未获知样本数据以及目标模型第一部分的情况下对目标模型的第二部分进行参数调整,从而在保护数据隐私性的前提下实现模型的联合训练。
需要说明的是,步骤S620中的数据索引信息用于使标签节点确定使用哪个或哪些标签数据,其可以由标签节点事先向数据节点发送,也可以是中间数据中所携带的信息,还可以是第三方在发起联合训练过程时,同时向数据节点与标签节点发送的索引信息。
在一示例性实施例中,步骤S620之后,联合训练方法还可以包括以下步骤:
根据误差数据判断是否达到预设条件;
如果达到预设条件,则向数据节点发送训练结束信息,并将当前的第二部分确定为目标模型最终的第二部分;
如果未达到预设条件,则继续执行步骤S630与S640。
该过程可以参见图3中的步骤S231~S233,如果达到预设条件,说明目标模型已经训练完成,将该信息以训练结束信息的形式通知到数据节点,标签节点上当前的第二部分即为最终的第二部分,目标模型由数据节点当前所持有的第一部分与标签节点当前所持有的第二部分共同确定。
在一示例性实施例中,步骤S610之前,联合训练方法还可以包括以下步骤:向数据节点发送数据索引信息,使数据节点通过目标模型的第一部分处理数据索引信息对应的样本数据以得到中间数据。该过程可以参见图3中的步骤S205与S210。
在一示例性实施例中,步骤S610可以是:从数据节点接收加密后的中间数据。
在一示例性实施例中,步骤S630可以是:将包含噪声的误差数据发送至数据节点。
在一示例性实施例中,上述中间数据可以通过同态加密算法进行加密。
本公开的示例性实施例还提供一种机器学习模型的联合训练装置,可以应用于图1中联合训练***100的数据节点110,联合训练***100还可以包括标签节点120;如图7所示,该装置700可以包括:获取模块710,用于获取用于训练的样本数据;处理模块720,用于通过目标模型的第一部分处理样本数据,得到中间数据;发送模块730,用于将中间数据发送至标签节点;调整模块740,用于如果从标签节点接收到误差数据,则根据误差数据调整第一部分的参数。
在一示例性实施例中,联合训练装置还可以包括:确定模块,用于如果从标签节点接收到训练结束信息,则将当前的第一部分确定为目标模型最终的第一部分。
在一示例性实施例中,获取模块可以用于从标签节点获取数据索引信息,并根据数据索引信息获取样本数据。
在一示例性实施例中,发送模块可以用于将加密后的中间数据发送至标签节点。
在一示例性实施例中,调整模块可以用于如果从标签节点接收到包含噪声的误差数据,则根据包含噪声的误差数据调整第一部分的参数。
在一示例性实施例中,上述中间数据可以采用同态加密。
根据本公开的第五方面,提供一种机器学习模型的联合训练装置,可以应用于图1中联合训练***100的标签节点120,联合训练***100还可以包括数据节点110;如图8所示,该装置800可以包括:接收模块810,用于从数据节点接收中间数据;处理模块820,用于通过目标模型的第二部分处理中间数据,并基于数据索引信息对应的标签数据得到误差数据;发送模块830,用于将误差数据发送至数据节点;调整模块840,用于根据误差数据调整目标模型的第二部分的参数。
在一示例性实施例中,处理模块还可以用于根据误差数据判断是否达到预设条件;发送模块还可以用于如果达到预设条件,则向数据节点发送训练结束信息,并将当前的第二部分确定为目标模型最终的第二部分,以及如果未达到预设条件,则将误差数据发送至数据节点。
在一示例性实施例中,发送模块还可以用于向数据节点发送数据索引信息,使数据节点通过目标模型的第一部分处理数据索引信息对应的样本数据以得到中间数据。
在一示例性实施例中,接收模块可以用于从数据节点接收加密后的中间数据。
在一示例性实施例中,发送模块可以用于将包含噪声的误差数据发送至数据节点。
在一示例性实施例中,上述中间数据可以采用同态加密。
上述装置中各模块的具体细节在方法部分的实施例中已经详细说明,因此不再赘述。
本公开的示例性实施例还提供了一种能够实现上述方法的电子设备。
所属技术领域的技术人员能够理解,本公开的各个方面可以实现为***、方法或程序产品。因此,本公开的各个方面可以具体实现为以下形式,即:完全的硬件实施方式、完全的软件实施方式(包括固件、微代码等),或硬件和软件方面结合的实施方式,这里可以统称为“电路”、“模块”或“***”。
下面参照图9来描述根据本公开的这种示例性实施例的电子设备900。图9显示的电子设备900仅仅是一个示例,不应对本公开实施例的功能和使用范围带来任何限制。
如图9所示,电子设备900以通用计算设备的形式表现。电子设备900的组件可以包括但不限于:上述至少一个处理单元910、上述至少一个存储单元920、连接不同***组件(包括存储单元920和处理单元910)的总线930、显示单元940。
其中,存储单元存储有程序代码,程序代码可以被处理单元910执行,使得处理单元910执行本说明书上述“示例性方法”部分中描述的根据本公开各种示例性实施方式的步骤。例如,处理单元910可以执行图2、图3、图5或图6所示的方法步骤等。
存储单元920可以包括易失性存储单元形式的可读介质,例如随机存取存储单元(RAM)921和/或高速缓存存储单元922,还可以进一步包括只读存储单元(ROM)923。
存储单元920还可以包括具有一组(至少一个)程序模块925的程序/实用工具924,这样的程序模块925包括但不限于:操作***、一个或者多个应用程序、其它程序模块以及程序数据,这些示例中的每一个或某种组合中可能包括网络环境的实现。
总线930可以为表示几类总线结构中的一种或多种,包括存储单元总线或者存储单元控制器、***总线、图形加速端口、处理单元或者使用多种总线结构中的任意总线结构的局域总线。
电子设备900也可以与一个或多个外部设备1100(例如键盘、指向设备、蓝牙设备等)通信,还可与一个或者多个使得用户能与该电子设备900交互的设备通信,和/或与使得该电子设备900能与一个或多个其它计算设备进行通信的任何设备(例如路由器、调制解调器等等)通信。这种通信可以通过输入/输出(I/O)接口960进行。并且,电子设备900还可以通过网络适配器960与一个或者多个网络(例如局域网(LAN),广域网(WAN)和/或公共网络,例如因特网)通信。如图所示,网络适配器960通过总线930与电子设备900的其它模块通信。应当明白,尽管图中未示出,可以结合电子设备900使用其它硬件和/或软件模块,包括但不限于:微代码、设备驱动器、冗余处理单元、外部磁盘驱动阵列、RAID***、磁带驱动器以及数据备份存储***等。
通过以上的实施方式的描述,本领域的技术人员易于理解,这里描述的示例实施方式可以通过软件实现,也可以通过软件结合必要的硬件的方式来实现。因此,根据本公开实施方式的技术方案可以以软件产品的形式体现出来,该软件产品可以存储在一个非易失性存储介质(可以是CD-ROM,U盘,移动硬盘等)中或网络上,包括若干指令以使得一台计算设备(可以是个人计算机、服务器、终端装置、或者网络设备等)执行根据本公开示例性实施例的方法。
本公开的示例性实施例还提供了一种计算机可读存储介质,其上存储有能够实现本说明书上述方法的程序产品。在一些可能的实施方式中,本公开的各个方面还可以实现为一种程序产品的形式,其包括程序代码,当程序产品在终端设备上运行时,程序代码用于使终端设备执行本说明书上述“示例性方法”部分中描述的根据本公开各种示例性实施方式的步骤。
参考图10所示,描述了根据本公开的示例性实施例的用于实现上述方法的程序产品1000,其可以采用便携式紧凑盘只读存储器(CD-ROM)并包括程序代码,并可以在终端设备,例如个人电脑上运行。然而,本公开的程序产品不限于此,在本文件中,可读存储介质可以是任何包含或存储程序的有形介质,该程序可以被指令执行***、装置或者器件使用或者与其结合使用。
程序产品可以采用一个或多个可读介质的任意组合。可读介质可以是可读信号介质或者可读存储介质。可读存储介质例如可以为但不限于电、磁、光、电磁、红外线、或半导体的***、装置或器件,或者任意以上的组合。可读存储介质的更具体的例子(非穷举的列表)包括:具有一个或多个导线的电连接、便携式盘、硬盘、随机存取存储器(RAM)、只读存储器(ROM)、可擦式可编程只读存储器(EPROM或闪存)、光纤、便携式紧凑盘只读存储器(CD-ROM)、光存储器件、磁存储器件、或者上述的任意合适的组合。
计算机可读信号介质可以包括在基带中或者作为载波一部分传播的数据信号,其中承载了可读程序代码。这种传播的数据信号可以采用多种形式,包括但不限于电磁信号、光信号或上述的任意合适的组合。可读信号介质还可以是可读存储介质以外的任何可读介质,该可读介质可以发送、传播或者传输用于由指令执行***、装置或者器件使用或者与其结合使用的程序。
可读介质上包含的程序代码可以用任何适当的介质传输,包括但不限于无线、有线、光缆、RF等等,或者上述的任意合适的组合。
可以以一种或多种程序设计语言的任意组合来编写用于执行本公开操作的程序代码,程序设计语言包括面向对象的程序设计语言—诸如Java、C++等,还包括常规的过程式程序设计语言—诸如“C”语言或类似的程序设计语言。程序代码可以完全地在用户计算设备上执行、部分地在用户设备上执行、作为一个独立的软件包执行、部分在用户计算设备上部分在远程计算设备上执行、或者完全在远程计算设备或服务器上执行。在涉及远程计算设备的情形中,远程计算设备可以通过任意种类的网络,包括局域网(LAN)或广域网(WAN),连接到用户计算设备,或者,可以连接到外部计算设备(例如利用因特网服务提供商来通过因特网连接)。
此外,上述附图仅是根据本公开示例性实施例的方法所包括的处理的示意性说明,而不是限制目的。易于理解,上述附图所示的处理并不表明或限制这些处理的时间顺序。另外,也易于理解,这些处理可以是例如在多个模块中同步或异步执行的。
应当注意,尽管在上文详细描述中提及了用于动作执行的设备的若干模块或者单元,但是这种划分并非强制性的。实际上,根据本公开的示例性实施例,上文描述的两个或更多模块或者单元的特征和功能可以在一个模块或者单元中具体化。反之,上文描述的一个模块或者单元的特征和功能可以进一步划分为由多个模块或者单元来具体化。
本领域技术人员在考虑说明书及实践这里公开的发明后,将容易想到本公开的其他实施例。本申请旨在涵盖本公开的任何变型、用途或者适应性变化,这些变型、用途或者适应性变化遵循本公开的一般性原理并包括本公开未公开的本技术领域中的公知常识或惯用技术手段。说明书和实施例仅被视为示例性的,本公开的真正范围和精神由权利要求指出。
应当理解的是,本公开并不局限于上面已经描述并在附图中示出的精确结构,并且可以在不脱离其范围进行各种修改和改变。本公开的范围仅由所附的权利要求来限。