Abstract:Artificial intelligence (AI) is expected as a revolutionary technology to be widely used in Internet-of-Things (IoT) networks for computationally intensive tasks. However, the traditional centralized training framework imposes large latency, network burdens and high risk of privacy disclosure. As a promising distributed solution, federated learning involves the collaborative model training among edge devices, with the orchestration of a server to carry the capacities of low-latency and privacy preservation for AI-driven networks. To further improve the communication efficiency, over-the-air computation (AirComp) is capable of computing while transmitting data by exploiting the superposition property of wireless channels to harness the interference. However, gradient aggregation suffers from channel distortion induced by channel fading and noise, which may degrade the training performance. Moreover, it is beneficial to schedule the informative edge devices in federated learning under limited energy resources. In this work, we propose a dynamic device scheduling scheme for AirComp enabled federated learning systems. In this scheme, a proper number of qualified edge devices with channel inversion based power control are scheduled to participate the model training, where local updates diversity, channel condition and energy consumption are exploited jointly. Inspired by the Lyapunov drift-plus-penalty method, we formulate the optimization problem to attain the device selection strategy. Simulation results validate that the proposed scheme can achieve a close-to-optimal test accuracy with fast convergence rate, and present good performance of robustness under different channel conditions.

Edge Learning for Large-Scale Internet of Things With Task-Oriented Efficient Communication

Edge Learning for Large-Scale Internet of Things With Task-Oriented Efficient Communication

Hierarchical Federated Edge Learning with Adaptive Clustering in Internet of Things

Edge-Aided Computing and Transmission Scheduling for LTE-U-Enabled IoT

Deep Reinforcement Learning for Task Offloading in Edge Computing Assisted Power IoT

An Introduction to Communication Efficient Edge Machine Learning

Resource Allocation with Edge Computing in IoT Networks Via Machine Learning

A Meta Reinforcement Learning-Based Task Offloading Strategy for IoT Devices in an Edge Cloud Computing Environment

A Density-Based Offloading Strategy for IoT Devices in Edge Computing Systems

Task Offloading Based on LSTM Prediction and Deep Reinforcement Learning for Efficient Edge Computing in IoT

Importance-Aware Data Pre-Processing and Device Scheduling for Multi-Channel Edge Learning

Dependency-Aware and Latency-Optimal Computation Offloading for Multi-User Edge Computing Networks

Joint Task Partitioning and Parallel Scheduling in Device-Assisted Mobile Edge Networks

Distributed Learning Based Joint Communication and Computation Strategy of IoT Devices in Smart Cities

Energy Efficiency Priority IoT Task Collaborative Migration Strategy

Communication-Efficient Device Scheduling via Over-the-Air Computation for Federated Learning

Learning IoT in Edge: Deep Learning for the Internet of Things with Edge Computing

Deep reinforcement learning-based online task offloading in mobile edge computing networks

Incentive-Driven Task Offloading and Collaborative Computing in Device-Assisted MEC Networks

Edge Collaborative Online Task Offloading Method Based on Reinforcement Learning