Abstract:In this work, a <italic>Hybrid Hierarchical Federated Edge Learning</italic> (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 <italic>Markov Decision Process</italic> (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 <italic>Deep Q-Network</italic> (DQN) and <italic>Deep Deterministic Policy Gradient</italic> (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.

Importance-Aware Data Selection and Resource Allocation in Federated Edge Learning System.

Resource Allocation for Wireless Federated Edge Learning based on Data Importance

Joint User Selection and Resource Allocation for Fast Federated Edge Learning

Hierarchical Federated Edge Learning with Adaptive Clustering in Internet of Things

Joint Optimization of Resource Allocation and Data Selection for Fast and Cost-Efficient Federated Edge Learning

Semi-Decentralized Federated Edge Learning for Fast Convergence on Non-IID Data

Energy-Efficient Radio Resource Allocation for Federated Edge Learning

Hierarchical Federated Learning with Adaptive Clustering on Non-IID Data

Adaptive Batchsize Selection and Gradient Compression for Wireless Federated Learning

Enhanced Hybrid Hierarchical Federated Edge Learning Over Heterogeneous Networks

Accelerating DNN Training in Wireless Federated Edge Learning Systems

Research on Resource Allocation for Efficient Federated Learning

Resource-constrained Federated Edge Learning with Heterogeneous Data: Formulation and Analysis

In-Edge AI: Intelligentizing Mobile Edge Computing, Caching and Communication by Federated Learning

Gradient and Channel Aware Dynamic Scheduling for Over-the-Air Computation in Federated Edge Learning Systems

Energy-Efficient Resource Management for Federated Edge Learning With CPU-GPU Heterogeneous Computing

Cost-Efficient Federated Learning for Edge Intelligence in Multi-Cell Networks

FEEL: A Federated Edge Learning System for Efficient and Privacy-Preserving Mobile Healthcare

An Overview on Over-the-Air Federated Edge Learning

Adaptive Federated Learning in Resource Constrained Edge Computing Systems