Aggregation on Learnable Manifolds for Asynchronous Federated Optimization
Not on the map. Why?
Abstract (arXiv)
Asynchronous federated learning (FL) with heterogeneous clients faces two key issues: curvature-induced loss barriers encountered by standard linear parameter interpolation techniques (e.g. FedAvg) and interference from stale updates misaligned with the server's current optimisation state. To alleviate these issues, we introduce a geometric framework that casts aggregation as curve learning in a Riemannian model space and decouples trajectory selection from update conflict resolution. Within this, we propose AsyncBezier, which replaces linear aggregation with low-degree polynomial (Bezier) trajectories to bypass loss barriers, and OrthoDC, which projects delayed updates via inner product-based orthogonality to reduce interference. We establish framework-level convergence guarantees covering each variant given simple assumptions on their components. On three datasets spanning general-purpose and healthcare domains, including LEAF Shakespeare and FEMNIST, our approach consistently improves accuracy and client fairness over strong asynchronous baselines; finally, we show that these gains are preserved even when other methods are allocated a higher local compute budget.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 31 of its 44 references that are in the map. First posted on arXiv 18 Mar 2025.
Builds on
31 of 44 references.
References · 31
- 2016Communication-Efficient Learning of Deep Networks from Decentralized Datacited by 7,669
- 2018Federated Optimization in Heterogeneous Networkscited by 2,987
- 2020Gradient Surgery for Multi-Task Learningcited by 1,256
- 2018Averaging Weights Leads to Wider Optima and Better Generalizationcited by 1,209
- 2018ShuffleNet V2: Practical Guidelines for Efficient CNN Architecture Designcited by 1,136
- 2017Visualizing the Loss Landscape of Neural Netscited by 1,130
- 2020Adaptive Federated Optimizationcited by 1,046
- 2017ChestX-ray8: Hospital-scale Chest X-ray Database and Benchmarks on Weakly-Supervised Classification and Localization of Common Thorax Diseasescited by 972
- 2018LEAF: A Benchmark for Federated Settingscited by 792
- 2020The Future of Digital Health with Federated Learningcited by 633
- 2018Loss Surfaces, Mode Connectivity, and Fast Ensembling of DNNscited by 588
- 2020Federated Learning with Matched Averagingcited by 564
19 more
- 2019Agnostic Federated Learningcited by 456
- 2018Learning Continuous Hierarchies in the Lorentz Model of Hyperbolic Geometrycited by 265
- 2017EMNIST: an extension of MNIST to handwritten letterscited by 213
- 2021Federated Learning with Buffered Asynchronous Aggregationcited by 190
- 2016Asynchronous Stochastic Gradient Descent with Delay Compensationcited by 99
- 2022Improving Generalization in Federated Learning by Seeking Flat Minimacited by 98
- 2021Asynchronous Federated Learning on Heterogeneous Devices: A Surveycited by 93
- 2020Optimizing Mode Connectivity via Neuron Alignmentcited by 86
- 2021Loss Surface Simplexes for Mode Connecting Volumes and Fast Ensemblingcited by 75
- 2023Dynamic Regularized Sharpness Aware Minimization in Federated Learning: Approaching Global Consistency and Smooth Landscapecited by 46
- 2023Federated Learning for Computationally-Constrained Heterogeneous Devices: A Surveycited by 40
- 2022Stochastic Weight Averaging Revisitedcited by 26
- 2022AsyncFedED: Asynchronous Federated Learning with Euclidean Distance based Adaptive Weight Aggregationcited by 19
- 2022Federated Learning on Riemannian Manifoldscited by 13
- 2024A PAC-Bayesian Link Between Generalisation and Flat Minimacited by 13
- 2025Paths and Ambient Spaces in Neural Loss Landscapescited by 6
- 2024Ordered Momentum for Asynchronous SGDcited by 5
- 2023Mode Connectivity and Data Heterogeneity of Federated Learningcited by 3
- 2024Federated Learning over Connected Modescited by 3
Built on by
Citing papers per year: 2025: 1.
1 paper.