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] On-Device Federated Learning via Second-Order Optimization with Over-the-Air Computation
    Hua, Sheng
    Yang, Kai
    Shi, Yuanming
    2019 IEEE 90TH VEHICULAR TECHNOLOGY CONFERENCE (VTC2019-FALL), 2019,
  • [3] Decentralized Over-the-Air Federated Learning by Second-Order Optimization Method
    Yang, Peng
    Jiang, Yuning
    Wen, Dingzhu
    Wang, Ting
    Jones, Colin N.
    Shi, Yuanming
    IEEE TRANSACTIONS ON WIRELESS COMMUNICATIONS, 2024, 23 (06) : 5632 - 5647
  • [4] Over-the-Air Federated Learning via Weighted Aggregation
    Azimi-Abarghouyi, Seyed Mohammad
    Tassiulas, Leandros
    IEEE TRANSACTIONS ON WIRELESS COMMUNICATIONS, 2024, 23 (12) : 18240 - 18253
  • [5] Scalable Hierarchical Over-the-Air Federated Learning
    Azimi-Abarghouyi, Seyed Mohammad
    Fodor, Viktoria
    IEEE TRANSACTIONS ON WIRELESS COMMUNICATIONS, 2024, 23 (08) : 8480 - 8496
  • [6] 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
  • [7] 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
  • [8] 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,
  • [9] 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
  • [10] Asynchronous Federated Learning via Over-the-air Computation
    Zheng, Zijian
    Deng, Yansha
    Liu, Xiaonan
    Nallanathan, Arumugam
    IEEE CONFERENCE ON GLOBAL COMMUNICATIONS, GLOBECOM, 2023, : 1345 - 1350