基于联邦学习的人工智能模型训练方法、装置及系统
摘要:
本发明属于电力自动化领域,公开了一种基于联邦学习的人工智能模型训练方法、装置及系统,包括:获取初始人工智能模型并发送至各计算节点;接收中心节点发送的初始人工智能模型,作为本地人工智能模型;迭代进行更新步骤,至预设更新迭代阈值或当前人工智能模型满足预设条件时,将当前人工智能模型作为训练完成人工智能模型,并发送至各计算节点;迭代进行梯度计算步骤至预设梯度计算迭代阈值,并接收中心节点发送的训练完成人工智能模型,或者,迭代进行梯度计算步骤至接收到中心节点发送的训练完成人工智能模型,采用训练完成人工智能模型更新本地人工智能模型。通过异步模型参数更新方法提升高性能计算节点的算力利用率,提升模型训练速度。
0/0