Accelerated Federated Learning with Dynamic Model Partitioning for H-IoT

被引:0
作者
He, Peng [1 ]
Lan, Chunhui
Cui, Yaping
Wang, Ruyan
Wu, Dapeng
机构
[1] Chongqing Univ Posts & Telecommun, Sch Commun & Informat Engn, Chongqing, Peoples R China
来源
2023 IEEE WIRELESS COMMUNICATIONS AND NETWORKING CONFERENCE, WCNC | 2023年
关键词
Federated learning; split learning; data privacy; Lyapunov optimization;
D O I
10.1109/WCNC55385.2023.10118865
中图分类号
TP3 [计算技术、计算机技术];
学科分类号
0812 ;
摘要
In the Healthcare Internet of Things (H-IoT), Federated Learning (FL) is a promising solution for processing huge amounts of medical data. At present, FL applied in H-IoT still faces many challenges such as low training efficiency and high data privacy risk. In this work, we develop a three-layer FL architecture, which introduces split learning to both prevent the leakage of medical data and improve training efficiency according to the inherent properties of Neural Networks (NN). Moreover, we formulate a long-term optimization problem with the goal of accelerating training speed of models in H-IoT. Then, an online model partitioning algorithm namely Privacy-aware Model Partitioning Algorithm (PMPA) is derived based on Lyapunov optimization theory that enables mobile devices of the FL architecture to efficiently train local models and protect the data privacy. Furthermore, the simulation results show that compared with traditional FL, the local training delay of the proposed algorithm can be reduced by 28.94% and 39.89%, respectively.
引用
收藏
页数:6
相关论文
共 11 条
  • [11] Optimizing Federated Learning in Distributed Industrial IoT: A Multi-Agent Approach
    Zhang, Weiting
    Yang, Dong
    Wu, Wen
    Peng, Haixia
    Zhang, Ning
    Zhang, Hongke
    Shen, Xuemin
    [J]. IEEE JOURNAL ON SELECTED AREAS IN COMMUNICATIONS, 2021, 39 (12) : 3688 - 3703