Scalable and Resource-Efficient Second-Order Federated Learning via Over-the-Air Aggregation
Ghalkha, Abdulmomen, Issaid, Chaouki Ben, Bennis, Mehdi
–arXiv.org Artificial Intelligence
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\%$ of communication resources and energy savings compared to other first and second-order baselines.
arXiv.org Artificial Intelligence
Jan-14-2025
- Country:
- Europe > Finland > Northern Ostrobothnia > Oulu (0.04)
- Genre:
- Research Report > New Finding (0.34)
- Industry:
- Information Technology > Security & Privacy (0.68)
- Technology: