Global Composite Compression of Deep Neural Network in Wireless Sensor Networks for Edge Intelligent Fault Diagnosis

Liaoyuan Huan,Baoping Tang,Chunhua Zhao
DOI: https://doi.org/10.1109/jsen.2023.3290153
IF: 4.3
2023-01-01
IEEE Sensors Journal
Abstract:To address the problem of large-scale complex deep neural network (DNN) models that cannot be deployed at the edge of wireless sensor networks (WSNs) with limited storage and computing resources, resulting in a lack of real-time intelligent data-processing capability, global composite compression of DNN in WSNs for edge intelligent fault diagnosis is proposed. First, the criteria importance through intercriteria correlation (CRITIC) weighted Euclidean–Pearson distance similarity (CEPDS) algorithm is proposed to perform coarse and fine-grained composite pruning on the DNN model to remove redundant parameters and kernels. The Wasserstein distance (WD) method and the mini-batch ${K}$ -means clustering algorithm were used to globally quantify the output features and weight parameters of the pruned model, further reducing model storage and improving model inference speed. The experimental results show that the proposed method can compress the DNN model by approximately $20\times $ , maintain a high diagnostic accuracy of approximately 99%, reduce the power consumption of nodes by approximately 10%, and reduce the monitoring system time by 42%, indicating that the method has reached an advanced level in DNN model compression, node power consumption, and data transmission delay.
What problem does this paper attempt to address?