English

Scalable and Resource-Efficient Second-Order Federated Learning via Over-the-Air Aggregation

Machine Learning 2025-01-15 v3

Abstract

Second-order federated learning (FL) algorithms offer faster convergence than their first-order counterparts by leveraging curvature information. However, they are hindered by high computational and storage costs, particularly for large-scale models. Furthermore, the communication overhead associated with large models and digital transmission exacerbates these challenges, causing communication bottlenecks. In this work, we propose a scalable second-order FL algorithm using a sparse Hessian estimate and leveraging over-the-air aggregation, making it feasible for larger models. Our simulation results demonstrate more than 67%67\% of communication resources and energy savings compared to other first and second-order baselines.

Keywords

Cite

@article{arxiv.2410.07662,
  title  = {Scalable and Resource-Efficient Second-Order Federated Learning via Over-the-Air Aggregation},
  author = {Abdulmomen Ghalkha and Chaouki Ben Issaid and Mehdi Bennis},
  journal= {arXiv preprint arXiv:2410.07662},
  year   = {2025}
}

Comments

6 pages, 1 figure, 4 subfigures, letter

R2 v1 2026-06-28T19:15:43.169Z