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 条
  • [21] Over-The-Air Federated Learning Over Scalable Cell-Free Massive MIMO
    Sifaou, Houssem
    Li, Geoffrey Ye
    IEEE TRANSACTIONS ON WIRELESS COMMUNICATIONS, 2024, 23 (05) : 4214 - 4227
  • [22] Over-the-Air Federated Edge Learning With Hierarchical Clustering
    Aygun, Ozan
    Kazemi, Mohammad
    Gunduz, Deniz
    Duman, Tolga M.
    IEEE TRANSACTIONS ON WIRELESS COMMUNICATIONS, 2024, 23 (12) : 17856 - 17871
  • [23] Model Pruning for Efficient Over-the-Air Federated Learning in Tactical Networks
    Khan, Fazal Muhammad Ali
    Abou-Zeid, Hatem
    Hassan, Syed Ali
    2023 IEEE INTERNATIONAL CONFERENCE ON COMMUNICATIONS WORKSHOPS, ICC WORKSHOPS, 2023, : 1806 - 1811
  • [24] Energy and Spectrum Efficient Federated Learning via High-Precision Over-the-Air Computation
    Li, Liang
    Huang, Chenpei
    Shi, Dian
    Wang, Hao
    Zhou, Xiangwei
    Shu, Minglei
    Pan, Miao
    IEEE TRANSACTIONS ON WIRELESS COMMUNICATIONS, 2024, 23 (02) : 1228 - 1242
  • [25] 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
  • [26] 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
  • [27] 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
  • [28] 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,
  • [29] 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
  • [30] Over-the-Air Federated Learning with Enhanced Privacy
    Xue, Xiaochan
    Hasan, Moh Khalid
    Yu, Shucheng
    Kandel, Laxima Niure
    Song, Min
    ICC 2023-IEEE INTERNATIONAL CONFERENCE ON COMMUNICATIONS, 2023, : 4546 - 4551