[发明专利]模型训练方法、装置、设备和介质

权利要求书

1.一种模型训练方法，其特征在于，由第一方执行，所述方法包括：

根据拥有的标签数据和从第二方获取的第二方预测结果，确定残差原文；其中，所述第二方预测结果由所述第二方基于待训练的网络模型对第二方拥有的特征数据进行预测得到的；

对所述残差原文进行同态加密，得到残差密文；

向所述第二方发送所述残差密文，供所述第二方根据残差密文和第二方拥有的特征数据，确定第二方梯度密文；

对从所述第二方获取的第二方梯度密文进行同态解密，得到第二方梯度原文；

向第二方发送所述第二方梯度原文，供第二方根据所述第二方梯度原文继续对第二方的网络模型进行训练。

2.根据权利要求1所述的方法，其特征在于，根据拥有的标签数据和从第二方获取的第二方预测结果，确定残差原文，包括：

基于待训练的网络模型对第一方拥有的特征数据进行预测，得到第一方预测结果；

根据所述第一方预测结果和从第二方获取的第二方预测结果，确定综合预测结果；

根据拥有的标签数据和所述综合预测结果，确定残差原文。

3.根据权利要求1所述的方法，其特征在于，对所述残差原文进行同态加密，包括：

对所述残差原文进行同态加法加密。

4.根据权利要求1所述的方法，其特征在于，对所述残差原文进行同态加密，包括：采用放大系数对所述残差原文进行放大，得到放大残差；对所述放大残差进行同态加密；

相应地，对从所述第二方获取的第二方梯度密文进行同态解密，包括：

采用放大系数，对从所述第二方获取的第二方梯度密文进行同态解密。

5.根据权利要求1所述的方法，其特征在于，根据拥有的标签数据和从第二方获取的第二方预测结果，确定残差原文之后，还包括：

根据残差原文和第一方拥有的特征数据，确定第一方梯度原文；

根据所述第一方梯度原文，继续对第一方中的网络模型进行训练。

6.一种模型训练方法，其特征在于，由第二方执行，所述方法包括：

基于待训练的网络模型对第二方拥有的特征数据进行预测得到第二方预测结果；

向第一方发送所述第二方预测结果，供第一方执行如下：根据拥有的标签数据和所述第二方预测结果确定残差原文，并对所述残差原文进行同态加密得到残差密文；

根据从所述第一方获取的残差密文和第二方拥有的特征数据，确定第二方梯度密文；

向第一方发送所述第二方梯度密文，供第一方对所述第二方梯度密文进行同态解密得到第二方梯度原文；

根据从所述第一方获取的第二方梯度原文，继续对第二方的网络模型进行训练。

7.根据权利要求6所述的方法，其特征在于，根据从所述第一方获取的残差密文和第二方拥有的特征数据，确定第二方梯度密文，包括：

采用放大系数对第二方拥有的特征数据进行放大，得到特征放大数据；

根据从所述第一方获取的残差密文和所述特征放大数据，确定第二方梯度密文。

8.一种模型训练装置，其特征在于，配置于第一方，所述装置包括：

残差原文确定模块，用于根据拥有的标签数据和从第二方获取的第二方预测结果，确定残差原文；其中，所述第二方预测结果由所述第二方基于待训练的网络模型对第二方拥有的特征数据进行预测得到的；

残差密文获取模块，用于对所述残差原文进行同态加密，得到残差密文；

残差密文发送模块，用于向所述第二方发送所述残差密文，供所述第二方根据残差密文和第二方拥有的特征数据，确定第二方梯度密文；

第二方梯度原文获取模块，用于对从所述第二方获取的第二方梯度密文进行同态解密，得到第二方梯度原文；

第二方梯度原文发送模块，用于向第二方发送所述第二方梯度原文，供第二方根据所述第二方梯度原文继续对第二方的网络模型进行训练。