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] UAV-Assisted Hierarchical Aggregation for Over-the-Air Federated Learning
    Zhong, Xiangyu
    Yuan, Xiaojun
    Yang, Huiyuan
    Zhong, Chenxi
    2022 IEEE GLOBAL COMMUNICATIONS CONFERENCE (GLOBECOM 2022), 2022, : 807 - 812
  • [22] IRS Assisted Federated Learning: A Broadband Over-the-Air Aggregation Approach
    Zhang, Deyou
    Xiao, Ming
    Pang, Zhibo
    Wang, Lihui
    Poor, H. Vincent
    IEEE TRANSACTIONS ON WIRELESS COMMUNICATIONS, 2024, 23 (05) : 4069 - 4082
  • [23] Power Minimization in Federated Learning with Over-the-air Aggregation and Receiver Beamforming
    Kalarde, Faeze Moradi
    Liang, Ben
    Dong, Min
    Ahmed, Yahia A. Eldemerdash
    Cheng, Ho Ting
    PROCEEDINGS OF THE INT'L ACM CONFERENCE ON MODELING, ANALYSIS AND SIMULATION OF WIRELESS AND MOBILE SYSTEMS, MSWIM 2023, 2023, : 259 - 267
  • [24] Empowering over-the-air personalized federated learning via RIS
    Wei SHI
    Jiacheng YAO
    Jindan XU
    Wei XU
    Lexi XU
    Chunming ZHAO
    Science China(Information Sciences), 2024, 67 (11) : 371 - 372
  • [25] Over-the-Air Federated Graph Learning
    Wang, Zixin
    Zhou, Yong
    Shi, Yuanming
    IEEE TRANSACTIONS ON WIRELESS COMMUNICATIONS, 2024, 23 (12) : 18669 - 18683
  • [26] Over-the-Air Clustered Federated Learning
    Sami, Hasin Us
    Guler, Basak
    IEEE TRANSACTIONS ON WIRELESS COMMUNICATIONS, 2024, 23 (07) : 7877 - 7893
  • [27] Federated Linear Bandit Learning via Over-the-air Computation
    Wang, Jiali
    Jiang, Yuning
    Liu, Xin
    Wang, Ting
    Shi, Yuanming
    IEEE CONFERENCE ON GLOBAL COMMUNICATIONS, GLOBECOM, 2023, : 1363 - 1368
  • [28] Multiple Parallel Federated Learning via Over-the-Air Computation
    Shi, Gaoxin
    Guo, Shuaishuai
    Ye, Jia
    Saeed, Nasir
    Dang, Shuping
    IEEE OPEN JOURNAL OF THE COMMUNICATIONS SOCIETY, 2022, 3 : 1252 - 1264
  • [29] Federated Learning in Wireless Networks via Over-the-Air Computations
    Oksuz, Halil Yigit
    Molinari, Fabio
    Sprekeler, Henning
    Raisch, Joerg
    2023 62ND IEEE CONFERENCE ON DECISION AND CONTROL, CDC, 2023, : 4379 - 4386
  • [30] Federated Learning Over-the-Air by Retransmissions
    Hellstrom, Henrik
    Fodor, Viktoria
    Fischione, Carlo
    IEEE TRANSACTIONS ON WIRELESS COMMUNICATIONS, 2023, 22 (12) : 9143 - 9156