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

The rapid development of Artificial Intelligence (AI) and 5G paradigm, opens up new possibilities for emerging applications in Industrial Internet of Things (IIoT). However, the large amount of data, the limited resources of IoT devices, and the increasing concerns of data privacy, are major obstacles to improve the quality of services in IIoT. In this paper, we propose the Digital Twin Edge Networks (DITEN) by incorporating digital twin into edge networks to fill the gap between physical systems and digital spaces. We further leverage the federated learning to construct digital twin models of IoT devices based on their running data. Moreover, to mitigate the communication overhead, we propose an asynchronous model update scheme and formulate the federated learning scheme as an optimization problem. We further decompose the problem and solve the subproblems based on the Deep Neural Network (DNN) model. Numerical results show that our proposed federated learning scheme for DITEN improves the communication efficiency and reduces the transmission energy cost.