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.

Adaptive Cluster Head Selection and Spectrum Allocation for D2D-Enabled Collaborative Learning

Efficient Collaborative Learning over Unreliable D2D Network: Adaptive Cluster Head Selection and Resource Allocation

Joint Social, Energy and Transfer Rate Cluster Head Selection Strategy for D2D Multicast Communication

Hierarchical federated learning based on wireless D2D networks

Hierarchical Federated Learning with Adaptive Clustering on Non-IID Data

Machine Learning-Based Resource Optimization for D2D Communication Underlaying Networks

Energy-Efficient Connectivity-Aware Learning Over Time-Varying D2D Networks

Over-the-air Learning Rate Optimization for Federated Learning

Decentralized Edge Learning via Unreliable Device-to-Device Communications

Interference-aware channel allocation for device-to-device communication underlaying cellular networks

Modeling and Performance Analysis of Clustered Device-to-Device Networks

Enhanced Hybrid Hierarchical Federated Edge Learning Over Heterogeneous Networks

D2D-Enabled Data Sharing for Distributed Machine Learning at Wireless Network Edge

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

Dynamic Resource Allocation for Device‐to‐Device Communication Underlaying Cellular Networks

Joint Mode Selection and Resource Allocation for Device-to-Device Communications

Hybrid Centralized-Distributed Resource Allocation Based on Deep Reinforcement Learning for Cooperative D2D Communications

Joint spectrum and power allocation scheme based on value decomposition networks in D2D communication networks

Joint Deep Reinforcement Learning and Unsupervised Learning for Channel Selection and Power Control in D2D Networks

Social-Aware Distributed Joint Mode Selection And Link Allocation For Mobile D2d Communications

Distributed Learning for Optimal Spectrum Access in Dense Device-to-Device Ad-Hoc Networks