LGFDR: local and global feature denoising reconstruction for unsupervised anomaly detection

被引:3
作者
Chen, Yichi [1 ,2 ]
Chen, Bin [2 ,3 ,4 ]
Xian, Weizhi [4 ]
Wang, Junjie [3 ]
Huang, Yao [5 ]
Chen, Min [1 ,2 ]
机构
[1] Chinese Acad Sci, Chengdu Inst Comp Applicat, Chengdu 610041, Sichuan, Peoples R China
[2] Univ Chinese Acad Sci, Sch Comp Sci & Technol, Beijing 100049, Peoples R China
[3] HIT, Int Inst Art Intelligence, Shenzhen 518000, Guangdong, Peoples R China
[4] Harbin Inst Technol, Chongqing Res Inst, Chongqing 401120, Peoples R China
[5] Shanghai Spaceflight Precis Machinery Inst, Dept Test & Inspect, Shanghai 610101, Peoples R China
关键词
Unsupervised learning; Anomaly detection; Anomaly localization; Local and global anomalies; Feature denoising reconstruction;
D O I
10.1007/s00371-024-03281-x
中图分类号
TP31 [计算机软件];
学科分类号
081202 ; 0835 ;
摘要
Unsupervised anomaly detection is a challenging task in many visual inspection scenarios and has attracted significant attention. Anomalies are typically related to local low-level features or require global semantic information to be detected. However, most of the existing methods fail to strike a balance between local and global features and thus lack versatility and practicality. To address this issue, we propose local and global feature denoising reconstruction (LGFDR). The proposed method can implicitly learn the latent distribution of local and global features for normal images via a dual-tower reconstruction network. Next, a selective reconstruction head (SRH) is designed to adaptively fuse the information from local and global reconstructions. Moreover, adding noise to the features proves a simple and general operation that can further enhance the generalization of reconstruction networks. On the MVTec AD benchmark, LGFDR achieves 98.8% and 65.3% of pixel-level AUROC and AP for anomaly localization and 99.3% of image-level AUROC for anomaly detection, respectively. In addition, a real-world metal plate surface defect detection project is adopted to validate LGFDR. Both the public and the practical experimental results show the effectiveness of our proposed approach. The code will be available at https://github.com/Karma1628/work-1.
引用
收藏
页码:8881 / 8894
页数:14
相关论文
共 34 条
[1]   Skip-GANomaly: Skip Connected and Adversarially Trained Encoder-Decoder Anomaly Detection [J].
Akcay, Samet ;
Atapour-Abarghouei, Amir ;
Breckon, Toby P. .
2019 INTERNATIONAL JOINT CONFERENCE ON NEURAL NETWORKS (IJCNN), 2019,
[2]   MVTec AD - A Comprehensive Real-World Dataset for Unsupervised Anomaly Detection [J].
Bergmann, Paul ;
Fauser, Michael ;
Sattlegger, David ;
Steger, Carsten .
2019 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2019), 2019, :9584-9592
[3]   UTRAD: Anomaly detection and localization with U-Transformer [J].
Chen, Liyang ;
You, Zhiyuan ;
Zhang, Nian ;
Xi, Juntong ;
Le, Xinyi .
NEURAL NETWORKS, 2022, 147 :53-62
[4]  
Cohen N., 2020, ARXIV
[5]  
Defard Thomas, 2020, P INT C PATT REC NOV, P475, DOI [10.1007/978-3-030-68799-1_35, DOI 10.1007/978-3-030-68799-1_35]
[6]  
Dosovitskiy A, 2021, INT C LEARN REPR ICL
[7]  
Gaus Y. F. A., 2019, 2019 International Joint Conference on Neural Networks (IJCNN), P1, DOI [10.1109/IJCNN.2019.8851829, DOI 10.1109/IJCNN.2019.8851829]
[8]   DMCP: Differentiable Markov Channel Pruning for Neural Networks [J].
Guo, Shaopeng ;
Wang, Yujie ;
Li, Quanquan ;
Yan, Junjie .
2020 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2020, :1536-1544
[9]   Deep Residual Learning for Image Recognition [J].
He, Kaiming ;
Zhang, Xiangyu ;
Ren, Shaoqing ;
Sun, Jian .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :770-778
[10]   Coordinate Attention for Efficient Mobile Network Design [J].
Hou, Qibin ;
Zhou, Daquan ;
Feng, Jiashi .
2021 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR 2021, 2021, :13708-13717