Paper: 2603.23472 Authors: Rustem Islamov, Grigory Malinovsky, Alexander Gaponov, Aurelien Lucchi, Peter Richtárik, Eduard Gorbunov Categories: cs.LG, cs.CR, math.OC
Abstract
This paper tackles federated learning when both privacy and adversarial robustness matter. It proposes Byz-Clip21-SGD2M, which mixes robust aggregation, double momentum, and clipping. The method avoids relying on bounded gradients or extra server data, and the analysis gives high-probability convergence under standard smoothness plus sub-Gaussian gradient noise. It also matches strong rates without attackers and improves utility with Byzantine and DP constraints. Tests on MNIST with CNN and MLP models support the method.
Key Contributions
- Combines Byzantine robustness and differential privacy in one federated optimization method
- Uses robust aggregation, double momentum, and clipping
- Removes the need for bounded gradients or auxiliary server data
- Provides high-probability convergence guarantees under smoothness and sub-Gaussian gradient noise
- Shows practical results on MNIST with CNN and MLP models
Methodology
The core idea is to make federated optimization resilient on two fronts at once: adversarial clients and privacy-preserving noise. The proposed optimizer, Byz-Clip21-SGD2M, uses clipping to control the influence of extreme updates, double momentum to stabilize noisy optimization, and robust aggregation to reduce the impact of Byzantine participants.
A key advantage is that the analysis works under weaker assumptions than many prior methods. Instead of requiring bounded gradients or extra server-side data, the paper assumes smooth objectives and sub-Gaussian gradient noise. That makes the result easier to fit into realistic federated learning settings.
Results
The theoretical guarantees show convergence with high probability, and the rates remain strong even without attackers. When Byzantine and differential privacy constraints are both present, the method still improves utility relative to weaker baselines.
The MNIST experiments with CNN and MLP models provide a concrete sanity check. They show that the method is not only mathematically neat but also usable in a standard learning benchmark.
Takeaways
- Federated learning can be made robust to both malicious clients and privacy constraints in one method
- Clipping and robust aggregation are useful complements to momentum-based optimization
- Strong convergence guarantees are possible under weaker assumptions than bounded-gradient settings
- Byzantine robustness and differential privacy do not have to be treated as separate problems
- The method’s empirical results on MNIST support the theoretical analysis
论文: 2603.23472 作者: Rustem Islamov, Grigory Malinovsky, Alexander Gaponov, Aurelien Lucchi, Peter Richtárik, Eduard Gorbunov 分类: cs.LG, cs.CR, math.OC
摘要
本文研究联邦学习中同时需要隐私保护与对抗鲁棒性的情形。作者提出了Byz-Clip21-SGD2M,将鲁棒聚合、双动量和裁剪结合在一起。该方法不依赖有界梯度或额外的服务器数据,并在标准光滑性与亚高斯梯度噪声假设下给出高概率收敛分析。它在没有攻击者时也能达到很强的收敛率,并在拜占庭与差分隐私约束下改善效用。作者还在MNIST上使用CNN和MLP模型进行了实验验证。
主要贡献
- 在一个联邦优化方法中同时结合拜占庭鲁棒性与差分隐私
- 采用鲁棒聚合、双动量和裁剪机制
- 不再需要有界梯度或额外的服务器辅助数据
- 在光滑性与亚高斯梯度噪声假设下给出高概率收敛保证
- 在MNIST上的CNN和MLP实验中验证实用性
方法论
该方法的核心目标是在两个方向上同时增强联邦优化的稳定性:对抗客户端和隐私保护噪声。提出的优化器Byz-Clip21-SGD2M使用裁剪来限制极端更新的影响,使用双动量稳定噪声优化,并通过鲁棒聚合降低拜占庭参与者的破坏作用。
一个关键优势是分析所需的假设比许多已有方法更弱。论文不要求有界梯度或额外的服务器侧数据,而是只假设目标函数光滑且梯度噪声服从亚高斯分布。这让结果更贴近现实中的联邦学习场景。
结果
理论保证表明该方法具有高概率收敛性,并且即使没有攻击者时也保持很强的收敛率。当同时存在拜占庭攻击和差分隐私约束时,该方法仍然能比较弱的基线取得更好的效用。
在MNIST上使用CNN和MLP模型的实验提供了直观验证,说明该方法不仅理论漂亮,也能在标准学习基准上实际运行。
要点总结
- 联邦学习可以用一种方法同时兼顾恶意客户端鲁棒性和隐私约束
- 裁剪与鲁棒聚合是动量优化的有效补充
- 在比有界梯度设定更弱的假设下仍可获得强收敛保证
- 拜占庭鲁棒与差分隐私不必被拆成两个独立问题
- MNIST实验支持了理论分析