Topology Learning for Heterogeneous Decentralized Federated Learning over Unreliable D2D Networks (2312.13611v2)
Abstract: With the proliferation of intelligent mobile devices in wireless device-to-device (D2D) networks, decentralized federated learning (DFL) has attracted significant interest. Compared to centralized federated learning (CFL), DFL mitigates the risk of central server failures due to communication bottlenecks. However, DFL faces several challenges, such as the severe heterogeneity of data distributions in diverse environments, and the transmission outages and package errors caused by the adoption of the User Datagram Protocol (UDP) in D2D networks. These challenges often degrade the convergence of training DFL models. To address these challenges, we conduct a thorough theoretical convergence analysis for DFL and derive a convergence bound. By defining a novel quantity named unreliable links-aware neighborhood discrepancy in this convergence bound, we formulate a tractable optimization objective, and develop a novel Topology Learning method considering the Representation Discrepancy and Unreliable Links in DFL, named ToLRDUL. Intensive experiments under both feature skew and label skew settings have validated the effectiveness of our proposed method, demonstrating improved convergence speed and test accuracy, consistent with our theoretical findings.
- T. Yin, L. Li, W. Lin, T. Ni, Y. Liu, H. Xu, and Z. Han, “Joint client scheduling and wireless resource allocation for heterogeneous federated edge learning with non-iid data,” IEEE Transactions on Vehicular Technology, pp. 1–13, 2023.
- H. Zhu, J. Kuang, M. Yang, and H. Qian, “Client selection with staleness compensation in asynchronous federated learning,” IEEE Transactions on Vehicular Technology, vol. 72, no. 3, pp. 4124–4129, 2023.
- H. Ye, L. Liang, and G. Y. Li, “Decentralized federated learning with unreliable communications,” IEEE J. Sel. Top. Signal Process., vol. 16, no. 3, pp. 487–500, 2022.
- Z. Wu, X. Wu, and Y. Long, “Joint scheduling and robust aggregation for federated localization over unreliable wireless d2d networks,” IEEE Transactions on Network and Service Management, 2022.
- Z. Chen, W. Yi, and A. Nallanathan, “Exploring representativity in device scheduling for wireless federated learning,” IEEE Transactions on Wireless Communications, 2023.
- B. L. Bars, A. Bellet, M. Tommasi, E. Lavoie, and A. Kermarrec, “Refined convergence and topology learning for decentralized SGD with heterogeneous data,” in AISTATS, vol. 206 of Proceedings of Machine Learning Research, pp. 1672–1702, PMLR, 2023.
- W. Liu, L. Chen, and W. Zhang, “Decentralized federated learning: Balancing communication and computing costs,” IEEE Trans. Signal Inf. Process. over Networks, vol. 8, pp. 131–143, 2022.
- Z. Yan and D. Li, “Performance analysis for resource constrained decentralized federated learning over wireless networks,” arXiv preprint arXiv:2308.06496, 2023.
- C. Hu, Z. Chen, and E. G. Larsson, “Scheduling and aggregation design for asynchronous federated learning over wireless networks,” IEEE J. Sel. Areas Commun., vol. 41, no. 4, pp. 874–886, 2023.
- A. T. Nguyen, P. H. S. Torr, and S. N. Lim, “Fedsr: A simple and effective domain generalization method for federated learning,” in NeurIPS, 2022.
- Cambridge University Press Cambridge, MA, USA, 2022.
- A. B. de Luca, G. Zhang, X. Chen, and Y. Yu, “Mitigating data heterogeneity in federated learning with data augmentation,” arXiv preprint arXiv:2206.09979, 2022.
- Zheshun Wu (8 papers)
- Zenglin Xu (145 papers)
- Dun Zeng (15 papers)
- Junfan Li (10 papers)
- Jie Liu (492 papers)