基于外推自适应步长的压缩联邦学习优化算法
Extrapolation Adaptive Step Size for Compressed Federated Learning Optimization Algorithm
摘要: 由于联邦学习中的通信成本高昂,依赖通信压缩的方法正日益受到关注。本文基于外推自适应步长和压缩联邦学习框架提出了外推压缩联邦学习算法(ExpFedCom)。ExpFedCom算法支持通信压缩并在凸、强凸以及非凸条件下实现了快速收敛。数值实验表明ExpFedCom算法既实现了快速收敛又节省了通信成本,这在实际生活中有着较好的应用价值。
Abstract: The high communication costs in federated learning have motivated increasing interest in communication compression methods. This paper proposes Extrapolated Federated Compression (ExpFedCom), a novel algorithm that integrates extrapolation adaptive step sizes with a communication compression federated learning framework. The ExpFedCom algorithm supports communication compression while achieving faster convergence rates under convex, strongly convex, and nonconvex settings. Numerical experiments demonstrate that ExpFedCom simultaneously attains faster convergence and significant communication savings, exhibiting strong practical potential for real-world applications.
文章引用:张耀威. 基于外推自适应步长的压缩联邦学习优化算法[J]. 理论数学, 2025, 15(10): 25-39. https://doi.org/10.12677/pm.2025.1510246

参考文献

[1] Kairouz, P., McMahan, H.B., Avent, B., et al. (2021) Advances and Open Problems in Federated Learning. Foundations and Trends® in Machine Learning, 14, 1-210. [Google Scholar] [CrossRef
[2] Kliestik, T., Zvarikova, K. and Lazaroiu, G. (2022) Data-Driven Machine Learning and Neural Network Algorithms in the Retailing Environment: Consumer Engagement, Experience, and Purchase Behaviors. Economics, Management and Financial Markets, 17, 57-69. [Google Scholar] [CrossRef
[3] Liu, J., Huang, J., Zhou, Y., et al. (2022) From Distributed Machine Learning to Federated Learning: A Survey. Knowledge and Information Systems, 64, 885-917. [Google Scholar] [CrossRef
[4] McMahan, B., Moore, E., Ramage, D., et al. (2017) Communication-Efficient Learning of Deep Networks from Decentralized Data. Proceedings of the 20th International Conference on Artificial Intelligence and Statistics, 54, 1273-1282.
[5] Karimireddy, S.P., Kale, S., Mohri, M., et al. (2020) Scaffold: Stochastic Controlled Averaging for Federated Learning. Proceedings of the 37th International Conference on Machine Learning, 119, 5132-5143.
[6] Khirirat, S., Feyzmahdavian, H.R. and Johansson, M. (2018) Distributed Learning with Compressed Gradients. arXiv:1806.06573.
[7] Haddadpour, F., Kamani, M.M., Mokhtari, A. and Mahdavi, M. (2021) Federated Learning with Compression: Unified Analysis and Sharp Guarantees. Proceedings of the 24th International Conference on Artificial Intelligence and Statistics, 130, 2350-2358.
[8] 贾泽慧, 李登辉, 刘治宇, 等. 基于数据压缩和梯度追踪的方差缩减的联邦优化算法[J]. 南京理工大学学报, 2025, 49(2): 155-166.
[9] Reddi, S.J., Charles, Z., Zaheer, M., et al. (2020) Adaptive Federated Optimization. arXiv:2003.00295.
[10] Jhunjhunwala, D., Wang, S. and Joshi, G. (2023) FedExP: Speeding up Federated Averaging via Extrapolation. arXiv:2301.09604.