English

Aggregation on Learnable Manifolds for Asynchronous Federated Optimization

Machine Learning 2025-10-13 v3

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.

Keywords

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

R2 v1 2026-06-28T22:25:29.845Z