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 条
  • [31] Over-the-Air Decentralized Federated Learning
    Shi, Yandong
    Zhou, Yong
    Shi, Yuanming
    2021 IEEE INTERNATIONAL SYMPOSIUM ON INFORMATION THEORY (ISIT), 2021, : 455 - 460
  • [32] Empowering over-the-air personalized federated learning via RIS
    Shi, Wei
    Yao, Jiacheng
    Xu, Jindan
    Xu, Wei
    Xu, Lexi
    Zhao, Chunming
    SCIENCE CHINA-INFORMATION SCIENCES, 2024, 67 (11)
  • [33] Over-the-Air Federated Learning with Retransmissions
    Hellstrom, Henrik
    Fodor, Viktoria
    Fischione, Carlo
    SPAWC 2021: 2021 IEEE 22ND INTERNATIONAL WORKSHOP ON SIGNAL PROCESSING ADVANCES IN WIRELESS COMMUNICATIONS (IEEE SPAWC 2021), 2020, : 291 - 295
  • [34] Over-the-Air Federated Learning and Optimization
    Zhu, Jingyang
    Shi, Yuanming
    Zhou, Yong
    Jiang, Chunxiao
    Chen, Wei
    Letaief, Khaled B.
    IEEE INTERNET OF THINGS JOURNAL, 2024, 11 (10): : 16996 - 17020
  • [35] OVER-THE-AIR PERSONALIZED FEDERATED LEARNING
    Sami, Hasin Us
    Guler, Basak
    2022 IEEE INTERNATIONAL CONFERENCE ON ACOUSTICS, SPEECH AND SIGNAL PROCESSING (ICASSP), 2022, : 8777 - 8781
  • [36] Robust Over-the-Air Federated Learning
    Kim, Hwanjin
    Nam, Hongjae
    Love, David J.
    2024 58TH ANNUAL CONFERENCE ON INFORMATION SCIENCES AND SYSTEMS, CISS, 2024,
  • [37] 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
  • [38] 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
  • [39] 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
  • [40] Uplink Over-the-Air Aggregation for Multi-Model Wireless Federated Learning
    Zhang, Chong
    Dong, Min
    Liang, Ben
    Afanat, Ali
    Ahmed, Yahia
    2024 IEEE 25TH INTERNATIONAL WORKSHOP ON SIGNAL PROCESSING ADVANCES IN WIRELESS COMMUNICATIONS, SPAWC 2024, 2024, : 36 - 40