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

Li Jiang,Yi Liu,Hui Tian,Lun Tang,Shengli Xie
DOI: https://doi.org/10.1109/jiot.2024.3357827
IF: 10.6
2024-01-01
IEEE Internet of Things Journal
Abstract:The development of industry 4.0 relies on emerging technologies of digital twin, machine learning, blockchain and Internet of Things (IoT) to build autonomous self-configuring systems that maximize manufactory efficiency, precision and accuracy. In this paper, we propose a new distributed and secure digital twin driven IIoT framework that integrates federated learning and Directed Acyclic Graph (DAG) blockchain with sharding. The proposed framework includes three planes: the data plane, the blockchain plane and the digital twin plane. Specifically, the data plane performs federated learning through a set of cluster heads to train models at network edges for twin model construction. The blockchain plane, which supports sharding, utilizes a hierarchical consensus scheme based on DAG blockchain to verify both local model updates and global model updates. The digital twin plane is responsible for constructing and maintaining twin model. Then, an efficient resource scheduling scheme is designed by considering performance of both federated learning and DAG blockchain with sharding. Accordingly, an optimization problem is formulated to maximize long-term utility of the digital twin driven IIoT. To cope with mapping error in the digital twin plane, a multi-agent Proximal Policy Optimization (MAPPO) approach is developed to solve the optimization problem. Numerical results illustrate that comparing with traditional approach, the proposed MAPPO improves utility by about 37 %, and reduces time latency by about 14%. Moreover, it also can well adapt to the mapping error.
What problem does this paper attempt to address?