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 条
  • [41] Device Scheduling for Relay-Assisted Over-the-Air Aggregation in Federated Learning
    Zhang, Fan
    Chen, Jining
    Wang, Kunlun
    Chen, Wen
    IEEE TRANSACTIONS ON VEHICULAR TECHNOLOGY, 2024, 73 (05) : 7412 - 7417
  • [42] BYZANTINE-RESILIENT HIERARCHICAL FEDERATED LEARNING WITH CLUSTERED OVER-THE-AIR AGGREGATION
    Nordlund, David
    Liao, Baling
    Chen, Zheng
    2024 IEEE INTERNATIONAL CONFERENCE ON ACOUSTICS, SPEECH, AND SIGNAL PROCESSING WORKSHOPS, ICASSPW 2024, 2024, : 715 - 719
  • [43] Device Scheduling in Over-the-Air Federated Learning Via Matching Pursuit
    Bereyhi, Ali
    Vagollari, Adela
    Asaad, Saba
    Muller, Ralf R.
    Gerstacker, Wolfgang
    Poor, H. Vincent
    IEEE TRANSACTIONS ON SIGNAL PROCESSING, 2023, 71 : 2188 - 2203
  • [44] Federated Learning via Active RIS Assisted Over-the-Air Computation
    Zhang, Deyou
    Xiao, Ming
    Skoglund, Mikael
    Poor, H. Vincent
    2024 IEEE INTERNATIONAL CONFERENCE ON MACHINE LEARNING FOR COMMUNICATION AND NETWORKING, ICMLCN 2024, 2024, : 201 - 207
  • [45] Blind Federated Learning via Over-the-Air q-QAM
    Razavikia, Saeed
    da Silva, José Mairton Barros
    Fischione, Carlo
    IEEE Transactions on Wireless Communications, 2024, 23 (12) : 19570 - 19586
  • [46] Federated Learning with Partial Gradients Over-the-Air
    Wang, Wendi
    Chen, Zihan
    Pappas, Nikolaos
    Yang, Howard H.
    2023 20TH ANNUAL IEEE INTERNATIONAL CONFERENCE ON SENSING, COMMUNICATION, AND NETWORKING, SECON, 2023,
  • [47] Over-the-Air Hierarchical Personalized Federated Learning
    Zhou, Fangtong
    Wang, Zhibin
    Shan, Hangguan
    Wu, Liantao
    Tian, Xiaohua
    Shi, Yuanming
    Zhou, Yong
    IEEE TRANSACTIONS ON VEHICULAR TECHNOLOGY, 2025, 74 (03) : 5006 - 5021
  • [48] Federated Learning Based on Over-the-Air Computation
    Yang, Kai
    Jiang, Tao
    Shi, Yuanming
    Ding, Zhi
    ICC 2019 - 2019 IEEE INTERNATIONAL CONFERENCE ON COMMUNICATIONS (ICC), 2019,
  • [49] Inverse Feasibility in Over-the-Air Federated Learning
    Piotrowski, Tomasz
    Ismayilov, Rafail
    Frey, Matthias
    Cavalcante, Renato L. G.
    IEEE SIGNAL PROCESSING LETTERS, 2024, 31 : 1434 - 1438
  • [50] Hierarchical Over-the-Air Federated Edge Learning
    Aygun, Ozan
    Kazemi, Mohammad
    Gunduz, Deniz
    Duman, Tolga M.
    IEEE INTERNATIONAL CONFERENCE ON COMMUNICATIONS (ICC 2022), 2022, : 3376 - 3381