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.

Machine Learning for Heterogeneous Ultra-Dense Networks with Graphical Representations

AI in 6G: Energy-Efficient Distributed Machine Learning for Multilayer Heterogeneous Networks

Heterogeneous Ultra-Dense Networks: Part 1

Joint User Association and Power Allocation in Heterogeneous Ultra Dense Network via Semi-Supervised Representation Learning

Heterogeneous Deep Graph Infomax

Generalized Sparse and Low-Rank Optimization for Ultra-Dense Networks

Heterogeneous Ultra Dense Networks: Part 2

Fine-grained graph representation learning for heterogeneous mobile networks with attentive fusion and contrastive learning

Enhanced Hybrid Hierarchical Federated Edge Learning Over Heterogeneous Networks

Hypergraph-Based SCMA Codebook Allocation in User-Centric Ultra-Dense Networks with Machine Learning.

DDHH: A Decentralized Deep Learning Framework for Large-scale Heterogeneous Networks

Hypergraph Theory: Applications in 5G Heterogeneous Ultra-Dense Networks

Resource Allocation for Ultradense Networks with Machine-Learning-Based Interference Graph Construction.

Scalable Power Control/Beamforming in Heterogeneous Wireless Networks with Graph Neural Networks

An Optimization Design of Ultra Dense Networks Balancing Mobility and Densification

Fine-grained Spatio-Temporal Distribution Prediction of Mobile Content Delivery in 5G Ultra-Dense Networks

HybridGNN: Learning Hybrid Representation in Multiplex Heterogeneous Networks

Resource allocation in heterogeneous network with node and edge enhanced graph attention network

Machine Learning for 6G Enhanced Ultra-Reliable and Low-Latency Services

Machine Learning for 6G Wireless Networks: Carrying Forward Enhanced Bandwidth, Massive Access, and Ultrareliable/Low-Latency Service

Learning Load Balancing with GNN in MPTCP-Enabled Heterogeneous Networks