Adaptive optimization federated learning enabled digital twins in industrial IoT

Abstract

The Industrial Internet of Things (IIoT) plays a pivotal role in steering enterprises towards comprehensive digital transformation and fostering intelligent production, which serves as a critical pillar of Industry 4.0. Digital twin (DT) emerges as a highly promising technology, enabling the digital transformation of the IIoT by seamlessly bridging physical systems with digital spaces. However, the overall service quality of the IIoT is severely impacted by the resource-limited devices and the massive, heterogeneous and sensitive data in the IIoT. As an innovative distributed machine learning paradigm, federated learning (FL) inherently possesses advantages in handling private and heterogeneous data. In this paper, we propose a novel framework integrating FL with DT-enabled IIoT, termed FDEI, which combines the merits of both to improve service quality while maintaining trustworthiness. To enhance the modeling efficiency, we develop FedOA, an adaptive optimization FL method that dynamically adjusts the local update coefficient and model compression rate in resource-limited IIoT scenarios, to construct the FDEI model. Specifically, leveraging the interdependence between the two variables, we conduct a theoretical analysis of the model convergence rate and derive the associated convergence bounds. Building upon the theoretical analysis, we further propose a joint adaptive adjustment strategy by optimizing the two variables across various clients to minimize runtime differences and accelerate the convergence rate. Numerical results demonstrate that our proposed approach achieves an approximate 68% improvement in convergence speed and a reduction of approximately 66% in traffic consumption compared to the benchmarks (e.g., FedAvg, AFL, and CSFL).