Adversarial Algorithm Unrolling Network for Interpretable Mechanical Anomaly Detection

被引:37
作者
An, Botao [1 ]
Wang, Shibin [1 ]
Qin, Fuhua [1 ]
Zhao, Zhibin [1 ]
Yan, Ruqiang [1 ]
Chen, Xuefeng [1 ]
机构
[1] Xi An Jiao Tong Univ, Sch Mech Engn, Xian 710049, Peoples R China
基金
中国国家自然科学基金;
关键词
Anomaly detection; Encoding; Feature extraction; Decoding; Vibrations; Codes; Training; Adversarial training; algorithm unrolling; anomaly detection; interpretable neural network; representation learning;
D O I
10.1109/TNNLS.2023.3250664
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
In mechanical anomaly detection, algorithms with higher accuracy, such as those based on artificial neural networks, are frequently constructed as black boxes, resulting in opaque interpretability in architecture and low credibility in results. This article proposes an adversarial algorithm unrolling network (AAU-Net) for interpretable mechanical anomaly detection. AAU-Net is a generative adversarial network (GAN). Its generator, composed of an encoder and a decoder, is mainly produced by algorithm unrolling of a sparse coding model, which is specially designed for feature encoding and decoding of vibration signals. Thus, AAU-Net has a mechanism-driven and interpretable network architecture. In other words, it is ad hoc interpretable. Moreover, a multiscale feature visualization approach for AAU-Net is introduced to verify that meaningful features are encoded by AAU-Net, helping users to trust the detection results. The feature visualization approach enables the results of AAU-Net to be interpretable, i.e., post hoc interpretable. To verify AAU-Net's capability of feature encoding and anomaly detection, we designed and performed simulations and experiments. The results show that AAU-Net can learn signal features that match the dynamic mechanism of the mechanical system. Considering the excellent feature learning ability, unsurprisingly, AAU-Net achieves the best overall anomaly detection performance compared with other algorithms.
引用
收藏
页码:6007 / 6020
页数:14
相关论文
共 53 条
[42]   Ridge-Aware Weighted Sparse Time-Frequency Representation [J].
Tong, Chaowei ;
Wang, Shibin ;
Selesnick, Ivan W. ;
Yan, Ruqiang ;
Chen, Xuefeng .
IEEE TRANSACTIONS ON SIGNAL PROCESSING, 2021, 69 :136-149
[43]   Nonconvex Sparse Regularization and Convex Optimization for Bearing Fault Diagnosis [J].
Wang, Shibin ;
Selesnick, Ivan ;
Cai, Gaigai ;
Feng, Yining ;
Sui, Xin ;
Chen, Xuefeng .
IEEE TRANSACTIONS ON INDUSTRIAL ELECTRONICS, 2018, 65 (09) :7332-7342
[44]   Fault-Attention Generative Probabilistic Adversarial Autoencoder for Machine Anomaly Detection [J].
Wu, Jingyao ;
Zhao, Zhibin ;
Sun, Chuang ;
Yan, Ruqiang ;
Chen, Xuefeng .
IEEE TRANSACTIONS ON INDUSTRIAL INFORMATICS, 2020, 16 (12) :7479-7488
[45]   Scalable Hypergrid k-NN-Based Online Anomaly Detection in Wireless Sensor Networks [J].
Xie, Miao ;
Hu, Jiankun ;
Han, Song ;
Chen, Hsiao-Hwa .
IEEE TRANSACTIONS ON PARALLEL AND DISTRIBUTED SYSTEMS, 2013, 24 (08) :1661-1670
[46]   Semisupervised Training of Deep Generative Models for High-Dimensional Anomaly Detection [J].
Xie, Qin ;
Zhang, Peng ;
Yu, Boseon ;
Choi, Jaesik .
IEEE TRANSACTIONS ON NEURAL NETWORKS AND LEARNING SYSTEMS, 2022, 33 (06) :2444-2453
[47]  
Xu B., 2015, Empirical evaluation of rectified activations in convolutional network, V1505, P853
[48]   Physics-Constraint Variational Neural Network for Wear State Assessment of External Gear Pump [J].
Xu, Wengang ;
Zhou, Zheng ;
Li, Tianfu ;
Sun, Chuang ;
Chen, Xuefeng ;
Yan, Ruqiang .
IEEE TRANSACTIONS ON NEURAL NETWORKS AND LEARNING SYSTEMS, 2024, 35 (05) :5996-6006
[49]   ISTA-Net: Interpretable Optimization-Inspired Deep Network for Image Compressive Sensing [J].
Zhang, Jian ;
Ghanem, Bernard .
2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, :1828-1837
[50]   Visual interpretability for deep learning: a survey [J].
Zhang, Quan-shi ;
Zhu, Song-chun .
FRONTIERS OF INFORMATION TECHNOLOGY & ELECTRONIC ENGINEERING, 2018, 19 (01) :27-39