Sep, 2023

DRAG: 基于差异的适应性聚合在非 IID 数据上的联邦学习

TL;DR基于梯度下降的分布式学习算法在通信效率方面取得了一定的成果,但由于异构数据分布引起的工作节点的‘客户端漂移’现象导致收敛速度减慢。为解决此问题,本文提出了一种称为‘分歧度量’的新指标,并通过该指标引入了一种自适应聚合算法,该算法可以在每一轮中动态地将收到的本地更新向参考方向‘拉拽’,而无需额外的通信开销。同时,本文还对该算法的收敛性进行了严格的分析,并通过实验证明了其管理‘客户端漂移’现象方面的优越性。此外,该算法还显著增强了对特定的拜占庭攻击的抵抗能力,在与 FL 服务器进行安全的数据共享的同时,通过综合实验对这些攻击进行了有效的应对。