联邦学习的模型训练方法、装置、设备及存储介质.pdfVIP

  • 1
  • 0
  • 约3.58万字
  • 约 30页
  • 2023-06-05 发布于四川
  • 举报

联邦学习的模型训练方法、装置、设备及存储介质.pdf

本申请实施例公开了一种联邦学习的模型训练方法、装置、设备及存储介质,属于机器学习技术领域。该方法包括:基于第i子模型的第i模型参数以及第i一阶梯度生成第i标量算子;基于第i标量算子向下一节点设备发送第i融合算子;基于获取到的二阶梯度标量、第i模型参数以及第i一阶梯度,确定第i子模型的第i二阶梯度下降方向;基于第i二阶梯度下降方向更新第i子模型。本申请实施例中,节点设备之间通过传递融合算子,联合计算子模型的二阶梯度下降方向,完成模型迭代训练,不需要依赖第三方节点就能够利用二阶梯度下降法训练机器学习

(19)中华人民共和国国家知识产权局 (12)发明专利 (10)授权公告号 CN 112733967 B (45)授权公告日 2021.06.29 (21)申请号 202110337283.9 G06N 3/08 (2006.01) (22)申请日 2

您可能关注的文档

文档评论(0)

1亿VIP精品文档

相关文档