Unified Analyses for Hierarchical Federated Learning: Topology Selection under Data Heterogeneity
Ziyi Zhou, Yipeng Li, Xinchen Lyu
Abstract
Hierarchical Federated Learning (HFL) addresses critical scalability limitations in conventional federated learning by incorporating intermediate aggregation layers, yet optimal topology selection across varying data heterogeneity conditions and network conditions remains an open challenge. This paper establishes the first unified convergence framework for all four HFL topologies (Star-Star, Star-Ring, Ring-Star, and Ring-Ring) with full/partial client participation under nonconvex objectives and different intra/inter-group data heterogeneity. Our theoretical analysis reveals three fundamental principles for topology selection: (1) The top-tier aggregation topology exerts greater influence on convergence than the intra-group topology, with ring-based top-tier configurations generally outperforming star-based alternatives; (2) Optimal topology strongly depends on client grouping characteristics, where Ring-Star excels with numerous small groups while Star-Ring is superior for large, client-dense clusters; and (3) Inter-group heterogeneity dominates convergence dynamics across all topologies, necessitating clustering strategies that minimize inter-group divergence. Extensive experiments on CIFAR-10/CINIC-10/Fashion-MNIST/SST-2 with ResNet-18/VGG-9/ResNet-10/MLP validate these insights, and provide practitioners with theoretically grounded guidance for HFL system design in real-world deployments. * Xinchen Lyu is the corresponding author.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 4660c9a7-6b4d-4853-be05-cbaca1d725b5Builds on5
- A Unified Theory of Decentralized SGD with Changing Topology and Local UpdatesAnastasia Koloskova, Nicolas Loizou, Sadra Boreiri, Martin Jaggi et al.ICML 2020 · 623 citations
- Federated Learning Based on Dynamic RegularizationDurmus Alp Emre Acar, Yue Zhao, Ramon Matas Navarro, Matthew Mattina et al.ICLR 2021 · 114 citations
- Convergence Analysis of Sequential Federated Learning on Heterogeneous DataYipeng Li, Xinchen LyuNeurIPS 2023 · 53 citations
- Hierarchical Federated Learning with Multi-Timescale Gradient CorrectionWenzhi Fang, Dong-Jun Han, Evan Chen, Shiqiang Wang et al.NeurIPS 2024 · 34 citations
- Multi-Level Local SGD: Distributed SGD for Heterogeneous Hierarchical NetworksTimothy Castiglia, Anirban Das, Stacy PattersonICLR 2021 · 11 citations
Related papers
- Heterogeneity-Guided Client Sampling: Towards Fast and Efficient Non-IID Federated LearningHuancheng Chen, Haris VikaloNeurIPS 2024 · 15 citations
- Tackling System and Statistical Heterogeneity for Federated Learning with Adaptive Client SamplingBing Luo, Wenli Xiao, Shiqiang Wang, Jianwei Huang et al.INFOCOM 2022 · 224 citations
- Linear Convergence in Federated Learning: Tackling Client Heterogeneity and Sparse GradientsAritra Mitra, Rayana H. Jaafar, George J. Pappas, Hamed HassaniNeurIPS 2021 · 193 citations
- Understanding Convergence and Generalization in Federated Learning through Feature Learning TheoryWei Huang, Ye Shi, Zhongyi Cai, Taiji SuzukiICLR 2024 · 17 citations
- GeoFL: A Framework for Efficient Geo-Distributed Cross-Device Federated LearningMaolin Gan, Lanpeng Li, Samiul Alam, Li Liu et al.INFOCOM 2025 · 2 citations
