arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2409.08640cs.LGcs.DC
收藏

基于压缩动量滤波的拜占庭鲁棒且通信高效的分布式学习

Byzantine-Robust and Communication-Efficient Distributed Learning via Compressed Momentum Filtering

Digital Futures(数字未来机构);College of Computer Science, Sichuan University(四川大学计算机学院);Institute of Clinical Pathology, West China Hospital, Sichuan University(四川大学华西医院临床病理研究所)

Changxin Liu, Yanghao Li, Yuhao Yi, Karl H. Johansson

AI总结:

针对现有分布式学习方法拜占庭鲁棒性与通信效率不足、收敛邻域大的问题,提出基于Polyak动量的压缩动量滤波方法,可在压缩通信下抵御拜占庭攻击,收敛邻域更小且复杂度界匹配无拜占庭下界,经多任务实验验证有效。

摘要:

分布式学习已成为跨私有数据孤岛训练大规模机器学习模型的标准方法。尽管分布式学习提升了隐私保护能力与训练效率,但它仍面临拜占庭鲁棒性与通信缩减相关的关键挑战。现有的拜占庭鲁棒且通信高效的方法要么在每一轮迭代中,要么以一定概率在特定迭代中依赖完整梯度信息,且它们仅能收敛到解周围一个不必要的大邻域内。受这些问题启发,我们提出了一种全新的拜占庭鲁棒且通信高效的随机分布式学习方法,该方法对批次大小无要求,且相较于所有现有方法能收敛到最优解周围更小的邻域,与理论下界相契合。我们的核心创新是利用Polyak Momentum(Polyak动量)来缓解由有偏压缩器和随机梯度共同带来的噪声,从而在信息压缩场景下抵御拜占庭工作节点的攻击。我们针对非凸光滑损失函数场景下的算法给出了紧复杂度界的证明,证明这些界与无拜占庭场景下的下界相匹配。最后,我们通过一系列大量实验验证了算法的实际价值,在二分类和图像分类任务上对其性能进行了基准测试。

英文摘要:

Distributed learning has become the standard approach for training large-scale machine learning models across private data silos. While distributed learning enhances privacy preservation and training efficiency, it faces critical challenges related to Byzantine robustness and communication reduction. Existing Byzantine-robust and communication-efficient methods rely on full gradient information either at every iteration or at certain iterations with a probability, and they only converge to an unnecessarily large neighborhood around the solution. Motivated by these issues, we propose a novel Byzantine-robust and communication-efficient stochastic distributed learning method that imposes no requirements on batch size and converges to a smaller neighborhood around the optimal solution than all existing methods, aligning with the theoretical lower bound. Our key innovation is leveraging Polyak Momentum to mitigate the noise caused by both biased compressors and stochastic gradients, thus defending against Byzantine workers under information compression. We provide proof of tight complexity bounds for our algorithm in the context of non-convex smooth loss functions, demonstrating that these bounds match the lower bounds in Byzantine-free scenarios. Finally, we validate the practical significance of our algorithm through an extensive series of experiments, benchmarking its performance on both binary classification and image classification tasks.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑