Trustworthy DNN Partition for Blockchain-enabled Digital Twin in Wireless IIoT Networks

Xiumei Deng,Jun Li,Long Shi,Kang Wei,Ming Ding,Yumeng Shao,Wen Chen,Shi Jin
2024-05-28
Abstract:Digital twin (DT) has emerged as a promising solution to enhance manufacturing efficiency in industrial Internet of Things (IIoT) networks. To promote the efficiency and trustworthiness of DT for wireless IIoT networks, we propose a blockchain-enabled DT (B-DT) framework that employs deep neural network (DNN) partitioning technique and reputation-based consensus mechanism, wherein the DTs maintained at the gateway side execute DNN inference tasks using the data collected from their associated IIoT devices. First, we employ DNN partitioning technique to offload the top-layer DNN inference tasks to the access point (AP) side, which alleviates the computation burden at the gateway side and thereby improves the efficiency of DNN inference. Second, we propose a reputation-based consensus mechanism that integrates Proof of Work (PoW) and Proof of Stake (PoS). Specifically, the proposed consensus mechanism evaluates the off-chain reputation of each AP according to its computation resource contributions to the DNN inference tasks, and utilizes the off-chain reputation as a stake to adjust the block generation difficulty. Third, we formulate a stochastic optimization problem of communication resource (i.e., partition point) and computation resource allocation (i.e., computation frequency of APs for top-layer DNN inference and block generation) to minimize system latency under the time-varying channel state and long-term constraints of off-chain reputation, and solve the problem using Lyapunov optimization method. Experimental results show that the proposed dynamic DNN partitioning and resource allocation (DPRA) algorithm outperforms the baselines in terms of reducing the overall latency while guaranteeing the trustworthiness of the B-DT system.
Machine Learning
What problem does this paper attempt to address?
### Problems Addressed by the Paper The paper primarily aims to address several key issues faced by Digital Twin (DT) technology in executing Deep Neural Network (DNN) inference tasks within Industrial Internet of Things (IIoT) networks: 1. **Computational Resource Constraints**: Lightweight IIoT devices struggle to efficiently perform resource-intensive DNN inference tasks due to insufficient computational power. To tackle this, the authors propose a method of layered processing of DNN models, where the top-layer inference tasks of the DNN are offloaded to access points (APs) with sufficient computational capabilities. This approach alleviates the computational burden on the gateway side and enhances DNN inference efficiency. 2. **Trust and Security**: To ensure data consistency and security in distributed systems, the paper designs a reputation-based consensus mechanism that combines Proof of Work (PoW) and Proof of Stake (PoS). By evaluating each AP's contribution of computational resources to DNN inference tasks, the mechanism determines its off-chain reputation and uses this reputation as a stake to adjust the difficulty of on-chain PoW. This ensures system credibility while improving consensus efficiency. 3. **Resource Optimization Allocation**: The paper also proposes a stochastic optimization problem aimed at jointly optimizing communication resources (i.e., DNN layering points) and computational resources (including the computational frequency of APs for top-layer DNN inference and block generation). The goal is to minimize system latency under time-varying wireless channel conditions and long-term off-chain reputation constraints. In summary, the paper aims to address the challenges of computational resource limitations, trust and security assurance, and resource optimization allocation in IIoT networks when executing complex inference tasks by introducing blockchain technology and DNN layered processing methods.