Abstract:Combination of the industrial Internet of Things (IIoT) and federated learning (FL) is deemed as a promising solution to realizing Industry 4.0 and beyond. In this paper, we focus on a hierarchical collaborative FL architecture over the IIoT systems, where the three-layer architectural design is conceived for supporting the training process. To effectively balance among the learning speed, energy consumption, and packet error rate for edge aggregation with regard to the participating IIoT devices, a weighted learning utility function is developed from the perspective of the fusing multiple performance metrics. An optimization problem is formulated to maximize the weighted learning utility by jointly optimizing the edge association as well as the allocations of resource block (RB), computation capacity, and transmit power of each IIoT device, under the practical constraints of the FL training process. The resulting problem is a non-convex and mixed integer optimization problem, and consequently it is difficult to solve. By resorting to the block coordinate descent method, we propose an overall alternating optimization algorithm to solve this problem in an iterative way. Specifically, in each iteration, for given transmit power and computation capacity, the sub-problem of joint RB assignment and edge association is transformed to a three-uniform weighted hypergraph model, which is solved by the local search-based three-dimensional hypergraph matching algorithm. Second, given RB assignment, edge association, and computation capacity, we employ the successive convex approximation method to tackle the sub-problem for optimizing the transmit power by turning it into a convex approximation problem. After the proposed alternating optimization algorithm converges to a tolerance threshold, a locally optimal solution of the original problem can be found. Numerical results reveal that our proposed joint optimization scheme can increase the system-wide learning utility and achieve significant performance gains over the four benchmark schemes.

Optimizing Federated Learning With Deep Reinforcement Learning for Digital Twin Empowered Industrial IoT

Adaptive Federated Learning and Digital Twin for Industrial Internet of Things

Energy-Efficient Federated Learning Framework for Digital Twin-Enabled Industrial Internet of Things

Communication-Efficient Personalized Federated Learning for Digital Twin in Heterogeneous Industrial IoT.

Resources-efficient Adaptive Federated Learning for Digital Twin-Enabled IIoT

Communication-Efficient Federated Learning for Digital Twin Edge Networks in Industrial IoT

Resource Efficient Asynchronous Federated Learning for Digital Twin Empowered IoT Network

Semi-Federated Learning Based Digital Twin in Large-Scale Heterogeneous Industrial Internet of Things.

Optimizing Federated Learning in Distributed Industrial IoT: A Multi-Agent Approach.

Digital Twin Empowered Industrial IoT Based on Credibility-weighted Swarm Learning

Efficient Ring-topology Decentralized Federated Learning with Deep Generative Models for Industrial Artificial Intelligent

Deep Reinforcement Learning Assisted Federated Learning Algorithm for Data Management of IIoT

Communication-Efficient Federated Learning for Digital Twin Systems of Industrial Internet of Things

Federated Deep Reinforcement Learning for Task Offloading in Digital Twin Edge Networks

Deep Reinforcement Learning for Energy Efficiency Maximization in SWIPT-Based Over-the-Air Federated Learning

Resource Efficient Federated Learning and DAG Blockchain with Sharding in Digital Twin Driven Industrial IoT

Digital Twin-Assisted Federated Learning with Blockchain in Multi-tier Computing Systems

Low-Latency Federated Learning With DNN Partition in Distributed Industrial IoT Networks

Federated Learning for Industrial Internet of Things in Future Industries

Federated Learning over the Industrial Internet of Things: A Joint Optimization of Edge Association and Resource Allocation

Virtual Network Embedding for Task Offloading in IIoT: A DRL-Assisted Federated Learning Scheme