Efficient visual anomaly detection model with adaptive wavelet transform

被引:0
作者
Du, Xiao [1 ]
Li, Bing [1 ,2 ]
Liu, Tongkun [1 ]
Jin, Liuyi [1 ]
Ding, Yi [3 ]
Zhao, Zhuo [1 ,4 ]
机构
[1] Xi An Jiao Tong Univ, State Key Lab Mfg Syst Engn, 99 Yanxiang Rd, Xian 710054, Shaanxi, Peoples R China
[2] Xi An Jiao Tong Univ, Int Joint Res Lab Micro Nano Mfg & Measurement Tec, 99 Yanxiang Rd, Xian 710054, Shaanxi, Peoples R China
[3] Wuxi Huansheng Precis Alloy Mat Co Ltd, 2 Xiangsheng Rd, Wuxi 214000, Jiangsu, Peoples R China
[4] Xi An Jiao Tong Univ, Coll Stomatol, Clin Res Ctr Shaanxi Prov Dent & Maxillofacial Dis, Xian, Shaanxi, Peoples R China
基金
中国国家自然科学基金; 中国博士后科学基金;
关键词
Anomaly detection; Wavelet transform; Synthetic anomalies;
D O I
10.1016/j.optlaseng.2024.108457
中图分类号
O43 [光学];
学科分类号
070207 ; 0803 ;
摘要
Existing unsupervised surface anomaly detection models have shown promising results. However, research on lightweight and fast anomaly detection models is relatively scarce, thereby impeding their practical application in industrial scenarios where rapid detection is required. Following the approach of reconstruction-based detection methods, we introduce EawT, an efficient anomaly detection model. EawT incorporates a novel pseudo-anomaly synthesis method and a depthwise separable convolutional structure to enhance network lightweighting. Additionally, considering that some studies indicate that Deep Neural Networks (DNNs) may exhibit preferences for certain frequency components during the learning process, potentially affecting the robustness of learned features, we propose an Adaptive Wavelet Transform module designed to modulate frequency components with transfer difficulties in latent space, thereby allowing for dynamic adjustment of feature information to adapt to various requirements. Furthermore, we propose wavelet loss to assist in the reconstruction of original images. Our method achieves comparable detection accuracy to existing reconstruction- based models on the VisA and BTAD datasets, with a frame rate of 228 FPS on a 3090Ti GPU and a lower parameter count, demonstrating the effectiveness of the proposed approach.
引用
收藏
页数:10
相关论文
共 45 条
[1]   Image coding using wavelet transform [J].
Antonini, Marc ;
Barlaud, Michel ;
Mathieu, Pierre ;
Daubechies, Ingrid .
IEEE TRANSACTIONS ON IMAGE PROCESSING, 1992, 1 (02) :205-220
[2]   Autoencoders for unsupervised anomaly segmentation in brain MR images: A comparative study [J].
Baur, Christoph ;
Denner, Stefan ;
Wiestler, Benedikt ;
Navab, Nassir ;
Albarqouni, Shadi .
MEDICAL IMAGE ANALYSIS, 2021, 69
[3]   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
[4]   Mixed supervision for surface-defect detection: From weakly to fully supervised learning [J].
Bozic, Jakob ;
Tabernik, Domen ;
Skocaj, Danijel .
COMPUTERS IN INDUSTRY, 2021, 129
[5]  
Cohen N, 2021, Arxiv, DOI arXiv:2005.02357
[6]   Improved anomaly detection by training an autoencoder with skip connections on images corrupted with Stain-shaped noise [J].
Collin, Anne-Sophie ;
De Vleeschouwer, Christophe .
2020 25TH INTERNATIONAL CONFERENCE ON PATTERN RECOGNITION (ICPR), 2021, :7915-7922
[7]  
Defard Thomas, 2021, Pattern Recognition. ICPR International Workshops and Challenges. Proceedings. Lecture Notes in Computer Science (LNCS 12664), P475, DOI 10.1007/978-3-030-68799-1_35
[8]   Anomaly Detection via Reverse Distillation from One-Class Embedding [J].
Deng, Hanqiu ;
Li, Xingyu .
2022 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2022, :9727-9736
[9]  
Howard AG, 2017, Arxiv, DOI arXiv:1704.04861
[10]   Template-guided Hierarchical Feature Restoration for Anomaly Detection [J].
Guo, Hewei ;
Ren, Liping ;
Fu, Jingjing ;
Wang, Yuwang ;
Zhang, Zhizheng ;
Lan, Cuiling ;
Wang, Haoqian ;
Hou, Xinwen .
2023 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION, ICCV, 2023, :6424-6435