一种基于可信执行环境的模型训练方法和装置与流程

文档序号:22924321发布日期:2020-11-13 16:17阅读:来源:国知局

技术特征:

1.一种基于可信执行环境的模型训练方法,包括:

从加密训练样本集中获取若干加密目标样本;

将所述若干加密目标样本输入可信执行环境,在可信执行环境中对所述加密目标样本解密并将解密后的目标样本输入特征提取模型,得到样本特征;

将所述可信执行环境输出的样本特征确定为用于本次迭代的目标样本特征;

利用所述目标样本特征对目标模型进行本次迭代训练。

2.根据权利要求1所述方法,所述利用所述目标样本特征对目标模型进行本次迭代训练,包括:

根据所述目标模型的损失函数,确定本次迭代中所述目标模型对应的第一梯度值;

利用所述第一梯度值更新所述目标模型的模型参数;

所述方法还包括:

基于所述第一梯度值计算出本次迭代中所述特征提取模型对应的第二梯度值;

将所述第二梯度值输入到所述可信执行环境中,并利用所述第二梯度值更新所述可信执行环境中特征提取模型的模型参数。

3.根据权利要求1所述方法,所述将解密后的目标样本输入特征提取模型之前,所述方法还包括:

在可信执行环境中对所述解密后的目标样本进行特征增强处理。

4.根据权利要求1至3任一项所述方法,基于多个训练进程执行所述目标模型的迭代训练;

每个训练进程获取若干加密目标样本的过程,包括:

在首次迭代时,获取本训练进程对应的子训练样本集,并将所述子训练样本集中的加密训练样本确定为本次迭代的加密目标样本;

在非首次迭代时,与指定训练进程交换部分加密训练样本,得到本次迭代的加密目标样本。

5.根据权利要求4所述方法,所述与指定训练进程交换部分加密训练样本,包括:

每个训练进程将对应的子训练样本集中预设比例的加密训练样本发送给第一指定训练进程;

接收由第二指定训练进程发送的加密训练样本。

6.根据权利要求4所述方法,各个训练进程分别用于训练得到所述目标模型的指定模型参数。

7.根据权利要求4所述方法,所述利用所述目标样本特征对目标模型进行本次迭代训练之后,所述方法还包括:

针对每个训练进程,执行以下操作:

与指定训练进程交换部分目标样本特征,得到本训练进程用于下次迭代的交换后的目标样本特征,并在下次迭代中利用所述交换后的目标样本特征对所述目标模型进行训练。

8.一种基于可信执行环境的模型训练装置,包括:

获取单元,从加密训练样本集中获取若干加密目标样本;

输入单元,将所述若干加密目标样本输入可信执行环境,在可信执行环境中对所述加密目标样本解密并将解密后的目标样本输入特征提取模型,得到样本特征;

确定单元,将所述可信执行环境输出的样本特征确定为用于本次迭代的目标样本特征;

训练单元,利用所述目标样本特征对目标模型进行本次迭代训练。

9.根据权利要求8所述装置,所述训练单元:

根据所述目标模型的损失函数,确定本次迭代中所述目标模型对应的第一梯度值;

利用所述第一梯度值更新所述目标模型的模型参数;

所述训练单元还:

基于所述第一梯度值计算出本次迭代中所述特征提取模型对应的第二梯度值;

利用所述第二梯度值更新所述特征提取模型的模型参数。

10.根据权利要求8所述装置,所述输入单元在将解密后的目标样本输入特征提取模型之前,还:

在可信执行环境中对所述解密后的目标样本进行特征增强处理。

11.根据权利要求8至10任一项所述装置,基于多个训练进程执行所述目标模型的迭代训练;

每个训练进程执行所述获取单元中获取若干加密目标样本时:

在首次迭代时,获取本训练进程对应的子训练样本集,并将所述子训练样本集中的加密训练样本确定为本次迭代的加密目标样本;

在非首次迭代时,与指定训练进程交换部分加密训练样本,得到本次迭代的加密目标样本。

12.根据权利要求11所述装置,所述获取单元在与指定训练进程交换部分加密训练样本时:

每个训练进程将对应的子训练样本集中预设比例的加密训练样本发送给第一指定训练进程;

接收由第二指定训练进程发送的加密训练样本。

13.根据权利要求11所述装置,各个训练进程分别用于训练得到所述目标模型的指定模型参数。

14.根据权利要求11所述装置,所述训练单元,还:

针对每个训练进程,执行以下操作:

与指定训练进程交换部分目标样本特征,得到本训练进程用于下次迭代的交换后的目标样本特征,并在下次迭代中利用所述交换后的目标样本特征对所述目标模型进行训练。

15.一种基于可信执行环境的模型训练装置,包括:

处理器;

用于存储机器可执行指令的存储器;

其中,通过读取并执行所述存储器存储的与基于可信执行环境的模型训练逻辑对应的机器可执行指令,所述处理器被促使:

从加密训练样本集中获取若干加密目标样本;

将所述若干加密目标样本输入可信执行环境,在可信执行环境中对所述加密目标样本解密并将解密后的目标样本输入特征提取模型,得到样本特征;

将所述可信执行环境输出的样本特征确定为用于本次迭代的目标样本特征;

利用所述目标样本特征对目标模型进行本次迭代训练。


技术总结
本说明书披露一种基于可信执行环境的模型训练方法和装置。在一个实施例中,该方法,包括:从加密训练样本集中获取若干加密目标样本;将所述若干加密目标样本输入可信执行环境,在可信执行环境中对所述加密目标样本解密并将解密后的目标样本输入特征提取模型,得到样本特征;将所述可信执行环境输出的样本特征确定为用于本次迭代的目标样本特征;利用所述目标样本特征对目标模型进行本次迭代训练。

技术研发人员:刘永超;贾贝;金跃;杨成平;张尧
受保护的技术使用者:支付宝(杭州)信息技术有限公司
技术研发日:2020.09.23
技术公布日:2020.11.13
当前第2页1 2 
网友询问留言 已有0条留言
  • 还没有人留言评论。精彩留言会获得点赞!
1