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

被引:0
|
作者
Ghalkha, Abdulmomen [1 ]
Ben Issaid, Chaouki [1 ]
Bennis, Mehdi [1 ]
机构
[1] Univ Oulu, Ctr Wireless Commun, Oulu 90014, Finland
关键词
Distributed optimization; federated learning; second-order methods; over-the-air aggregation;
D O I
10.1109/LWC.2024.3521027
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
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 letter, 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.
引用
收藏
页码:716 / 720
页数:5
相关论文
共 50 条
  • [1] Over-the-Air Federated Learning via Second-Order Optimization
    Yang, Peng
    Jiang, Yuning
    Wang, Ting
    Zhou, Yong
    Shi, Yuanming
    Jones, Colin N.
    IEEE TRANSACTIONS ON WIRELESS COMMUNICATIONS, 2022, 21 (12) : 10560 - 10575
  • [2] Over-the-Air Federated Learning via Weighted Aggregation
    Azimi-Abarghouyi, Seyed Mohammad
    Tassiulas, Leandros
    IEEE TRANSACTIONS ON WIRELESS COMMUNICATIONS, 2024, 23 (12) : 18240 - 18253
  • [3] Communication-Efficient Federated Learning: A Second Order Newton-Type Method With Analog Over-the-Air Aggregation
    Krouka, Mounssif
    Elgabli, Anis
    Ben Issaid, Chaouki
    Bennis, Mehdi
    IEEE TRANSACTIONS ON GREEN COMMUNICATIONS AND NETWORKING, 2022, 6 (03): : 1862 - 1874
  • [4] Scalable Hierarchical Over-the-Air Federated Learning
    Azimi-Abarghouyi, Seyed Mohammad
    Fodor, Viktoria
    IEEE TRANSACTIONS ON WIRELESS COMMUNICATIONS, 2024, 23 (08) : 8480 - 8496
  • [5] Federated Learning via Over-the-Air Computation
    Yang, Kai
    Jiang, Tao
    Shi, Yuanming
    Ding, Zhi
    IEEE TRANSACTIONS ON WIRELESS COMMUNICATIONS, 2020, 19 (03) : 2022 - 2035
  • [6] Beamforming and Device Selection Design in Federated Learning With Over-the-Air Aggregation
    Kalarde, Faeze Moradi
    Dong, Min
    Liang, Ben
    Ahmed, Yahia A. Eldemerdash
    Cheng, Ho Ting
    IEEE OPEN JOURNAL OF THE COMMUNICATIONS SOCIETY, 2024, 5 : 1710 - 1723
  • [7] Resource-Efficient Federated Learning with Hierarchical Aggregation in Edge Computing
    Wang, Zhiyuan
    Xu, Hongli
    Liu, Jianchun
    Huang, He
    Qiao, Chunming
    Zhao, Yangming
    IEEE CONFERENCE ON COMPUTER COMMUNICATIONS (IEEE INFOCOM 2021), 2021,
  • [8] Coded Over-the-Air Computation for Model Aggregation in Federated Learning
    Zhang, Naifu
    Tao, Meixia
    Wang, Jia
    Shao, Shuo
    IEEE COMMUNICATIONS LETTERS, 2023, 27 (01) : 160 - 164
  • [9] ROBUST FEDERATED LEARNING VIA OVER-THE-AIR COMPUTATION
    Sifaou, Houssem
    Li, Geoffrey Ye
    2022 IEEE 32ND INTERNATIONAL WORKSHOP ON MACHINE LEARNING FOR SIGNAL PROCESSING (MLSP), 2022,
  • [10] Federated Learning With Over-the-Air Aggregation Over Time-Varying Channels
    Tegin, Busra
    Duman, Tolga M.
    IEEE TRANSACTIONS ON WIRELESS COMMUNICATIONS, 2023, 22 (08) : 5671 - 5684