ISSN 1009-5624 CN 10-2021/TQ    主管:中国乐凯集团有限公司    主办:北京乐凯科技有限公司

中国知网全文收录期刊
万方数据库收录期刊
RCCSE中文学术期刊
维普资讯网/超星域出版 全文收录
中国核心期刊(遴选)数据库收录期刊
首页 > 刊期 > 2024 > 1期 > 综合:探索与发现
DFSL:基于细粒度共享层激励机制的深度联邦学习算法
张浩宇,巫朝霞(通信作者)

【摘要】联邦学习算法已成功地应用于医疗、无人机编队任务规划等场景,目前大多数研究致力于异步联邦学习算法研究以及从算法层面解决优质客户端选择问题。 但是均存在以下问题:(1)难以针对不同的共享参数层来学习其对全局模型的贡献度;(2)难以确定客户端的选择标准,即在候选客户端中选择最有利于全局模型参数更新的客户端梯度。 基于此,本文提出基于细粒度共享层激励机制的深度联邦学习算法(deep federated learning algorithm based on fine⁃grained shared layer incentive mechanism,DFSL)。 该算法从不同的共享参数层量化其对全局模型准确率的贡献度,同时使用末位淘汰法去除潜在的异常客户端梯度,实现全局模型的稳定迭代更新。 实验结果表明,本文算法与其他 4 种典型算法在 CIFAR-10 以及 MNIST 两个数据集上的召回率以及 F1 值均达到最优。 在CIFAR-10 数据集上召回率以及 F1 值分别提升了 3. 73%以及 2. 12%;在 MNIST 数据集上分别提升了 1. 99%以及 2. 10%。


【关键字】联邦学习;优质客户端选择;激励机制;异常客户端