国内刊号:51-1267/TN
国际刊号:1001-893X
发布日期:
作者:张海波,任俊平,蔡磊,邹灿
单位:(1.重庆邮电大学 通信与信息工程学院,重庆 400065;2.泛在感知与互联重庆市重点实验室,重庆 400065;3.360数字安全科技集团有限公司, 北京 100015)
关键词:图像识别;异步联邦学习;数据异构;资源异构;优化引导;
基金:国家自然科学基金资助项目(62271094);重庆市留创计划创新类资助项目(cx2020059)
针对在数据异构和资源异构的无线网络中联邦学习训练效率低及训练能耗高的问题,面向图像识别任务,提出了基于优化引导的异步联邦学习算法AFedGuide。利用较高样本多样性的客户端模型的引导作用,提高单轮聚合有效性。采用基于训练状态的模型增量异步更新机制,提高模型更新实时性以及信息整合能力。设计基于模型差异性的训练决策,修正优化方向。仿真结果显示,相较于对比算法,AFedGuide的训练时长平均减少67.78%,系统能耗平均节省65.49%,客户端的准确率方差平均减少25.5%,说明在客户端数据异构和资源异构的无线网络下,AFedGuide可以在较短的训练时间内以较小的训练能耗完成训练目标,并维持较高的训练公平性和模型适用性。
来源:2024年第6期
《电讯技术》期刊编辑部