ShanghaiTech University Knowledge Management System
Enhanced Hybrid Hierarchical Federated Edge Learning Over Heterogeneous Networks | |
2023 | |
发表期刊 | IEEE TRANSACTIONS ON VEHICULAR TECHNOLOGY (IF:6.1[JCR-2023],6.5[5-Year]) |
ISSN | 0018-9545 |
EISSN | 1939-9359 |
卷号 | PP期号:99页码:1-15 |
发表状态 | 已发表 |
DOI | 10.1109/TVT.2023.3287355 |
摘要 | In this work, a Hybrid Hierarchical Federated Edge Learning (HHFEL) architecture that consists of a device layer, an edge layer, and a cloud layer over heterogeneous networks, is investigated for large-scale model training. In such systems, learning efficiency is severely degraded by limited communication resources and device heterogeneity in terms of local data distribution and computation capability, especially for synchronous FL mechanisms where the training of each round should wait for the slowest device. To tackle this issue, asynchronous FL is proposed, which allows the devices with powerful computation and communication capabilities exchanging information with the server more frequently. However, this asynchronous FL framework faces a new challenge of low accuracy caused by the imbalanced local model updating. To overcome the shortage of both synchronous and asynchronous FLs, we propose an enhanced online semi-asynchronous FL mechanism between the edge-device layers, where each device trains its local model with the newly generated data and each edge server aggregates a number of local models based on their arrival order in each round. Particularly, devices with faster training speeds would fully utilize the idle time by training their local models repetitively. Meanwhile, synchronous FL with an edge elastic update strategy is adopted to the cloud-edge layers for personalized information exchange. Considering the continuous data generation feature, we formulate the objective problem as an online Markov Decision Process (MDP) to realize efficient communication-and-computing HHFEL via joint device selection and resource allocation. Due to the non-convex and combinatorial problem structure, we develop a hybrid Deep Q-Network (DQN) and Deep Deterministic Policy Gradient (DDPG) approach with low computational complexity to adapt the device selection and resource allocation strategies. Numerical results show the effectiveness of the proposed mechanism compared with existing benchmarks. IEEE ; In this work, a Hybrid Hierarchical Federated Edge Learning (HHFEL) architecture that consists of a device layer, an edge layer, and a cloud layer over heterogeneous networks, is investigated for large-scale model training. In such systems, learning efficiency is severely degraded by limited communication resources and device heterogeneity in terms of local data distribution and computation capability, especially for synchronous FL mechanisms where the training of each round should wait for the slowest device. To tackle this issue, asynchronous FL is proposed, which allows the devices with powerful computation and communication capabilities exchanging information with the server more frequently. However, this asynchronous FL framework faces a new challenge of low accuracy caused by the imbalanced local model updating. To overcome the shortage of both synchronous and asynchronous FLs, we propose an enhanced online semi-asynchronous FL mechanism between the edge-device layers, where each device trains its local model with the newly generated data and each edge server aggregates a number of local models based on their arrival order in each round. Particularly, devices with faster training speeds would fully utilize the idle time by training their local models repetitively. Meanwhile, synchronous FL with an edge elastic update strategy is adopted to the cloud-edge layers for personalized information exchange. Considering the continuous data generation feature, we formulate the objective problem as an online Markov Decision Process (MDP) to realize efficient communication-and-computing HHFEL via joint device selection and resource allocation. Due to the non-convex and combinatorial problem structure, we develop a hybrid Deep Q-Network (DQN) and Deep Deterministic Policy Gradient (DDPG) approach with low computational complexity to adapt the device selection and resource allocation strategies. Numerical results show the effectiveness of the proposed mechanism compared with existing benchmarks. IEEE |
关键词 | Computer architecture Edge computing Hierarchical systems Information management Learning systems Markov processes Network architecture Network layers Resource allocation Cloud layers Computational modelling Device layers Device resources Device selection Edge computing Federated edge learning Local model Resource management Semi-asynchronoi Computer architecture Edge computing Hierarchical systems Information management Learning systems Markov processes Network architecture Network layers Resource allocation Cloud layers Computational modelling Device layers Device resources Device selection Edge computing Federated edge learning Local model Resource management Semi-asynchronoi |
URL | 查看原文 |
收录类别 | SCI ; EI |
语种 | 英语 |
出版者 | Institute of Electrical and Electronics Engineers Inc. |
EI入藏号 | 20232614302214 |
EI主题词 | Heterogeneous networks ; Heterogeneous networks |
EI分类号 | 722.4 Digital Computers and Systems ; 723 Computer Software, Data Handling and Applications ; 912.2 Management ; 922.1 Probability Theory ; 961 Systems Science ; 722.4 Digital Computers and Systems ; 723 Computer Software, Data Handling and Applications ; 912.2 Management ; 922.1 Probability Theory ; 961 Systems Science |
原始文献类型 | Article in Press |
来源库 | IEEE |
引用统计 | 正在获取...
|
文献类型 | 期刊论文 |
条目标识符 | https://kms.shanghaitech.edu.cn/handle/2MSLDSTB/316476 |
专题 | 信息科学与技术学院 信息科学与技术学院_PI研究组_文鼎柱组 |
作者单位 | 1.School of Electronic Information, Wuhan University, Wuhan, China 2.Network Intelligence Center, School of Information Science and Technology, ShanghaiTech University, Shanghai, China 3.College of Information Science and Electronic Engineering, Zhejiang University, Hangzhou, China |
推荐引用方式 GB/T 7714 | Qimei Chen,Zehua You,Dingzhu Wen,et al. Enhanced Hybrid Hierarchical Federated Edge Learning Over Heterogeneous Networks[J]. IEEE TRANSACTIONS ON VEHICULAR TECHNOLOGY,2023,PP(99):1-15. |
APA | Qimei Chen,Zehua You,Dingzhu Wen,&Zhaoyang Zhang.(2023).Enhanced Hybrid Hierarchical Federated Edge Learning Over Heterogeneous Networks.IEEE TRANSACTIONS ON VEHICULAR TECHNOLOGY,PP(99),1-15. |
MLA | Qimei Chen,et al."Enhanced Hybrid Hierarchical Federated Edge Learning Over Heterogeneous Networks".IEEE TRANSACTIONS ON VEHICULAR TECHNOLOGY PP.99(2023):1-15. |
条目包含的文件 | ||||||
文件名称/大小 | 文献类型 | 版本类型 | 开放类型 | 使用许可 |
修改评论
除非特别说明,本系统中所有内容都受版权保护,并保留所有权利。