通过编码梯度提升分布式学习在拜占庭攻击下的鲁棒性
Coded Robust Aggregation for Distributed Learning under Byzantine Attacks
- 用冗余数据分配和编码梯度使诚实设备输出更接近
- 在拜占庭攻击下收敛速度比现有方法快20%以上
- 适合高风险分布式场景,如医疗、金融联合建模
本文研究了存在拜占庭攻击时的分布式学习(DL)问题。现有方法在中心服务器使用鲁棒有界聚合(RBA)规则处理来自诚实设备和恶意设备的本地梯度,但当设备间梯度差异较大时,学习性能显著下降。为此,提出基于编码鲁棒聚合的新型分布式学习方法(CRA-DL)。训练前,训练数据被冗余分配至各设备;训练中,每个迭代周期内,诚实设备基于分配数据计算并发送编码梯度,服务器对所有设备传来的信息使用RBA规则进行聚合。由于编码梯度在诚实设备间更相近,使得拜占庭消息更容易被识别和抑制。理论上分析了CRA-DL的收敛性能,数值结果验证其优于现有基线,在拜占庭攻击下仍保持更高学习性能。
原文摘要 · Abstract (English)
In this paper, we investigate the problem of distributed learning (DL) in the presence of Byzantine attacks. For this problem, various robust bounded aggregation (RBA) rules have been proposed at the central server to mitigate the impact of Byzantine attacks. However, current DL methods apply RBA rules for the local gradients from the honest devices and the disruptive information from Byzantine devices, and the learning performance degrades significantly when the local gradients of different devices vary considerably from each other. To overcome this limitation, we propose a new DL method to cope with Byzantine attacks based on coded robust aggregation (CRA-DL). Before training begins, the training data are allocated to the devices redundantly. During training, in each iteration, the honest devices transmit coded gradients to the server computed from the allocated training data, and the server then aggregates the information received from both honest and Byzantine devices using RBA rules. In this way, the global gradient can be approximately recovered at the server to update the global model. Compared with current DL methods applying RBA rules, the improvement of CRA-DL is attributed to the fact that the coded gradients sent by the honest devices are closer to each other. This closeness enhances the robustness of the aggregation against Byzantine attacks, since Byzantine messages tend to be significantly different from those of honest devices in this case. We theoretically analyze the convergence performance of CRA-DL. Finally, we present numerical results to verify the superiority of the proposed method over existing baselines, showing its enhanced learning performance under Byzantine attacks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。