一种基于动量法的权重动态聚合联邦学习方法

AITNT-国内领先的一站式人工智能新闻资讯网站
# 热门搜索 #
一种基于动量法的权重动态聚合联邦学习方法
申请号:CN202410999650
申请日期:2024-07-24
公开号:CN118821977A
公开日期:2024-10-22
类型:发明专利
摘要
本发明涉及联邦学习技术领域,公开了一种基于动量法的权重动态聚合联邦学习方法,包括以下步骤:S1、初始化全局模型参数和聚合权重向量;S2、服务器将全局模型分发给被选中的客户端;S3、客户端替换本地模型并在本地数据集上测试目标损失;S4、客户端在本地数据集上运行梯度下降算法更新本地模型;S5、客户端将更新后的本地模型及目标损失发送给服务器;S6、服务器根据公式计算聚合权重,并进行加权聚合得到全局模型。通过使用动量法动态更新聚合权重策略,实现了联邦学习模型的公平性。以每轮通信过程中客户端的损失和以往客户端在全局模型中的性能表现作为度量方式,优化聚合权重,有效地实现了联邦学习中的公平性策略。
技术关键词
联邦学习方法 客户端 服务器更新 梯度下降算法 历史性能数据 联邦学习技术 联邦学习模型 更新模型参数 随机梯度下降 权重策略 因子 动态更新 有效性 度量