[发明专利]联邦模型的训练方法、装置、设备、存储介质及程序

权利要求书

1.一种联邦模型的训练方法，其特征在于，应用于包括第一设备和第二设备的联邦系统，待训练的联邦模型的模型参数包括待训练的第一权重集合和待训练的第二权重集合，所述第一设备用于训练所述第一权重集合，所述第二设备用于训练所述第二权重集合，所述方法包括：

所述第一设备根据第一对象的第一训练样本和所述第一权重集合，生成第一子结果，所述第一权重集合包括所述第一训练样本中各特征的权重值；

所述第一设备接收所述第二设备发送的第二子结果，所述第二子结果为所述第二设备根据所述第一对象的第二训练样本和所述第二权重集合得到的，所述第二权重集合包括所述第二训练样本中各特征的权重值；

所述第一设备根据所述第一子结果和所述第二子结果确定所述第一对象的预测结果，并根据所述预测结果和所述第一训练样本对应的标记结果更新所述第一权重集合，所述预测结果和所述标记结果还用于更新所述第二权重集合。

2.根据权利要求1所述的联邦模型的训练方法，其特征在于，所述第一设备根据所述预测结果和所述第一训练样本对应的标记结果更新所述第一权重集合，包括：

所述第一设备根据所述预测结果和所述标记结果，确定目标梯度，所述目标梯度包括第一子梯度和第二子梯度；

所述第一设备向所述第二设备发送所述第一子梯度，并接收所述第二设备发送的第一子权重集合，所述第一子权重集合为所述第二设备根据所述第一子梯度确定得到的；

所述第一设备根据所述第二子梯度确定第二子权重集合；

所述第一设备根据所述第一子权重集合和所述第二子权重集合更新所述第一权重集合。

3.根据权利要求2所述的联邦模型的训练方法，其特征在于，所述第一设备根据所述预测结果和所述标记结果，确定目标梯度，包括：

所述第一设备根据所述预测结果确定第一子预测结果和第二子预测结果；

所述第一设备根据所述第二子预测结果和所述标记结果，确定第二损失数据；

所述第一设备向所述第二设备发送所述第一子预测结果，并接收所述第二设备发送的第一损失数据，所述第一损失数据为所述第二设备根据所述第一子预测结果确定得到的；

所述第一设备根据所述第一损失数据和所述第二损失数据，确定所述目标梯度。

4.根据权利要求3所述的联邦模型的训练方法，其特征在于，所述第一设备根据所述第一损失数据和所述第二损失数据，确定所述目标梯度，包括：

所述第一设备将所述第一损失数据和所述第二损失数据之和确定为目标损失数据；

所述第一设备根据所述目标损失数据确定所述目标梯度。

5.根据权利要求1至4任一项所述的联邦模型的训练方法，其特征在于，所述第一设备根据所述第一子结果和所述第二子结果确定所述第一对象的预测结果，包括：

所述第一设备将所述第一子结果和所述第二子结果之和确定为中间结果；

所述第一设备根据预设参数对所述中间结果进行处理，得到所述预测结果。

6.根据权利要求1至4任一项所述的联邦模型的训练方法，其特征在于，所述第一设备根据第一对象的第一训练样本和所述第一权重集合，生成第一子结果，包括：

所述第一设备确定所述第一训练样本对应的第一样本向量，以及确定所述第一权重集合对应的第一权重向量；

所述第一设备将所述第一样本向量和所述第一权重向量的乘积确定为所述第一子结果。