Dual-Teacher Network with SSIM Based Reverse Distillation for Anomaly Detection

被引:0
作者
Li, Weihao [1 ]
Huang, Rongjin [1 ]
Wang, Zhanquan [1 ]
机构
[1] East China Univ Sci & Technol, Coll Informat Sci & Engn, Shanghai, Peoples R China
来源
PATTERN RECOGNITION AND COMPUTER VISION, PT XIII, PRCV 2024 | 2025年 / 15043卷
关键词
Anomaly detection; Teacher-student network; Dual-teacher network; SUPPORT;
D O I
10.1007/978-981-97-8493-6_19
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
There are many important applications for anomaly detection, such as industrial image anomaly detection, medical image anomaly detection etc. The teacher-student network (T-S) offers an effective method for identifying anomalies. Previous studies mainly utilize normal data or pseudo-anomalous for training, relying on cosine similarity for differentiation and usually involving only one guiding teacher. This approach limits the learning capacity of the student model, and solely concentrating on pixel-level differences might may hinder the model's grasp of overall anomaly structure and context, resulting in inadequate abnormal behavior recognition. To solve the problems, Dual-Teacher Network with SSIM based on Reverse Distillation (DSRD) is proposed, which combines the functionalities of two pre-trained teacher networks, namely T1 and T2, along with a denoising student encoder-decoder (S), a trainable one-class bottleneck embedding module (OCBE), and a segmentation network, all integrated within a unified framework. This comprehensive approach aims to leverage the strengths of each component to enhance outlier point detection and anomaly recognition capabilities. In the experiments on industrial inspection benchmark datasets, our method achieved state-of-the-art performance. The image-level AUC reached 99.41%, the pixel-level AUC reached 98.80%, the pixel-level average precision reached 79.34%, and the AUPRO reached 94.91%.
引用
收藏
页码:266 / 279
页数:14
相关论文
共 50 条
[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]  
Bergmann P, 2019, Arxiv, DOI [arXiv:1807.02011, DOI 10.48550/ARXIV.1807.02011]
[3]   Uninformed Students: Student-Teacher Anomaly Detection with Discriminative Latent Embeddings [J].
Bergmann, Paul ;
Fauser, Michael ;
Sattlegger, David ;
Steger, Carsten .
2020 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2020, :4182-4191
[4]   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
[5]   DeepLab: Semantic Image Segmentation with Deep Convolutional Nets, Atrous Convolution, and Fully Connected CRFs [J].
Chen, Liang-Chieh ;
Papandreou, George ;
Kokkinos, Iasonas ;
Murphy, Kevin ;
Yuille, Alan L. .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2018, 40 (04) :834-848
[6]  
Cimpoi M., 2013, Describing Textures in the Wild
[7]  
Cohen N, 2021, Arxiv, DOI arXiv:2005.02357
[8]  
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
[9]   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
[10]  
Deng J, 2009, PROC CVPR IEEE, P248, DOI 10.1109/CVPRW.2009.5206848