ISSN 1009-5624 CN 10-2021/TQ 主管:中国乐凯集团有限公司 主办:北京乐凯科技有限公司
【摘要】联邦学习算法已成功地应用于医疗、无人机编队任务规划等场景,目前大多数研究致力于异步联邦学习算法研究以及从算法层面解决优质客户端选择问题。 但是均存在以下问题:(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%。