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 条
[1]   GANomaly: Semi-supervised Anomaly Detection via Adversarial Training [J].
Akcay, Samet ;
Atapour-Abarghouei, Amir ;
Breckon, Toby P. .
COMPUTER VISION - ACCV 2018, PT III, 2019, 11363 :622-637
[2]   Interpretable Neural Network via Algorithm Unrolling for Mechanical Fault Diagnosis [J].
An, Botao ;
Wang, Shibin ;
Zhao, Zhibin ;
Qin, Fuhua ;
Yan, Ruqiang ;
Chen, Xuefeng .
IEEE TRANSACTIONS ON INSTRUMENTATION AND MEASUREMENT, 2022, 71
[3]  
Ba J, 2014, ACS SYM SER
[4]   Learned Super Resolution Ultrasound for Improved Breast Lesion Characterization [J].
Bar-Shira, Or ;
Grubstein, Ahuva ;
Rapson, Yael ;
Suhami, Dror ;
Atar, Eli ;
Peri-Hanania, Keren ;
Rosen, Ronnie ;
Eldar, Yonina C. .
MEDICAL IMAGE COMPUTING AND COMPUTER ASSISTED INTERVENTION - MICCAI 2021, PT VII, 2021, 12907 :109-118
[5]  
Beck A, 2017, MOS-SIAM SER OPTIMIZ, P1, DOI 10.1137/1.9781611974997
[6]   Robust Anomaly Detection in Images Using Adversarial Autoencoders [J].
Beggel, Laura ;
Pfeiffer, Michael ;
Bischl, Bernd .
MACHINE LEARNING AND KNOWLEDGE DISCOVERY IN DATABASES, ECML PKDD 2019, PT I, 2020, 11906 :206-222
[7]   Deep Unrolled Recovery in Sparse Biological Imaging: Achieving fast, accurate results [J].
Ben Sahel, Yair ;
Bryan, John P. ;
Cleary, Brian ;
Farhi, Samouil L. ;
Eldar, Yonina C. .
IEEE SIGNAL PROCESSING MAGAZINE, 2022, 39 (02) :45-57
[8]   Representation Learning: A Review and New Perspectives [J].
Bengio, Yoshua ;
Courville, Aaron ;
Vincent, Pascal .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2013, 35 (08) :1798-1828
[9]  
Chen X., 2018, PROC MIDL C BOOK, P1
[10]   Generative Adversarial Networks An overview [J].
Creswell, Antonia ;
White, Tom ;
Dumoulin, Vincent ;
Arulkumaran, Kai ;
Sengupta, Biswa ;
Bharath, Anil A. .
IEEE SIGNAL PROCESSING MAGAZINE, 2018, 35 (01) :53-65