Aggregation on Learnable Manifolds for Asynchronous Federated Optimization
Abstract
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.
Cite
@article{arxiv.2503.14396,
title = {Aggregation on Learnable Manifolds for Asynchronous Federated Optimization},
author = {Archie Licudi and Anshul Thakur and Soheila Molaei and Danielle Belgrave and David Clifton},
journal= {arXiv preprint arXiv:2503.14396},
year = {2025}
}
Comments
18 pages, 5 figures. [v3] Updated from technical report to condensed conference paper under review at AISTATS 26