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.

Hierarchical Over-the-Air FedGradNorm

FedGradNorm: Personalized Federated Gradient-Normalized Multi-Task Learning

Hierarchical Over-the-Air Federated Edge Learning

Over-the-Air Federated Edge Learning with Hierarchical Clustering

HFedMTL: Hierarchical Federated Multi-Task Learning

Hierarchical Federated Learning with Adaptive Clustering on Non-IID Data

Joint User Association and Resource Allocation for Wireless Hierarchical Federated Learning with IID and Non-IID Data

Enhanced Hybrid Hierarchical Federated Edge Learning Over Heterogeneous Networks

Resource Allocation for Multi-Task Federated Learning Algorithm over Wireless Communication Networks

Hierarchical Over-the-Air Federated Learning with Awareness of Interference and Data Heterogeneity

Scalable Hierarchical Over-the-Air Federated Learning

Hierarchical Federated Learning with Multi-Timescale Gradient Correction

Over-the-Air Federated Multi-Task Learning Over MIMO Multiple Access Channels

Client Orchestration and Cost-Efficient Joint Optimization for NOMA-Enabled Hierarchical Federated Learning

Cost-Aware Hierarchical Federated Learning via Over-the-Air Computing

Personalized Decentralized Multi-Task Learning Over Dynamic Communication Graphs

Communication-efficient hierarchical federated learning for IoT heterogeneous systems with imbalanced data

Federated Multi-Task Learning under a Mixture of Distributions

Adaptive Hierarchical Federated Learning Over Wireless Networks

FedAuxHMTL: Federated Auxiliary Hard-Parameter Sharing Multi-Task Learning for Network Edge Traffic Classification

Federated Multi-Task Learning with Non-Stationary and Heterogeneous Data in Wireless Networks