Abnormity detection based on fusion feature distribution learning and image reconstruction

被引:0
|
作者
Zhu, Siyu [1 ]
Zhu, Lei [1 ]
Wang, Wenwu [1 ]
Yue, Huagang [1 ]
机构
[1] Wuhan Univ Sci & Technol, Sch Informat Sci & Engn, Wuhan 430070, Peoples R China
关键词
abnormal detection; rebuild; memory modules; flow model; fusion algorithm;
D O I
10.37188/CJLCD.2023-0304
中图分类号
O7 [晶体学];
学科分类号
0702 ; 070205 ; 0703 ; 080501 ;
摘要
Unsupervised learning is the main research direction in the field of industrial product defect detection at present,and it is mainly divided into two types of methods: reconstruction based and feature based methods. The former constructs content-aware mappings to map abnormal regions into normal regions and detect defects through residual images,focusing on overall performance of the images. The latter uses high-level semantic features to achieve positioning exceptions and pay more attention to image detail presentation. According to the advantages and disadvantages of the two methods,a method is proposed based on the fusion of characteristics and reconstruction, which effectively combines advantages of the two methods to complement their shortcomings and realize unified end-to-end learning and reasoning. A reconstructed model is trained firstly,then a normalized flow model is adopted to fully learn the probability distribution of high probability data of input normal samples, and it is integrated with the reconstructed model to effectively improve the accuracy of defect detection and defect positioning of the reconstructed model. On the widely used MVTec AD data set, the average image level AUROC of the proposed fusion model reaches 98. 7 degrees o, the average pixel-level AUROC reaches 94. 2 degrees o,in particular,an increase of 3. 3 degrees o compared to a single reconstruction model. The convergence model of characteristics and proposed reconstruction network has significantly improved the shortcomings of defect positioning in the reconstruction network part, which makes experimental results more accurate.
引用
收藏
页码:1116 / 1127
页数:12
相关论文
共 19 条
  • [1] Uninformed Students: Student-Teacher Anomaly Detection with Discriminative Latent Embeddings
    Bergmann, Paul
    Fauser, Michael
    Sattlegger, David
    Steger, Carsten
    [J]. 2020 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2020, : 4182 - 4191
  • [2] MVTec AD - A Comprehensive Real-World Dataset for Unsupervised Anomaly Detection
    Bergmann, Paul
    Fauser, Michael
    Sattlegger, David
    Steger, Carsten
    [J]. 2019 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2019), 2019, : 9584 - 9592
  • [3] A Hybrid Autoencoder and Density Estimation Model for Anomaly Detection
    Cao, Van Loi
    Nicolau, Miguel
    McDermott, James
    [J]. PARALLEL PROBLEM SOLVING FROM NATURE - PPSN XIV, 2016, 9921 : 717 - 726
  • [4] Cohen N, 2021, Arxiv, DOI [arXiv:2005.02357, DOI 10.48550/ARXIV.2005.02357]
  • [5] 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
  • [6] Memorizing Normality to Detect Anomaly: Memory-augmented Deep Autoencoder for Unsupervised Anomaly Detection
    Gong, Dong
    Liu, Lingqiao
    Le, Vuong
    Saha, Budhaditya
    Mansour, Moussa Reda
    Venkatesh, Svetha
    van den Hengel, Anton
    [J]. 2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2019), 2019, : 1705 - 1714
  • [7] CFLOW-AD: Real-Time Unsupervised Anomaly Detection with Localization via Conditional Normalizing Flows
    Gudovskiy, Denis
    Ishizaka, Shun
    Kozuka, Kazuki
    [J]. 2022 IEEE WINTER CONFERENCE ON APPLICATIONS OF COMPUTER VISION (WACV 2022), 2022, : 1819 - 1828
  • [8] Divide-and-Assemble: Learning Block-wise Memory for Unsupervised Anomaly Detection
    Hou, Jinlei
    Zhang, Yingying
    Zhong, Qiaoyong
    Xie, Di
    Pu, Shiliang
    Zhou, Hong
    [J]. 2021 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2021), 2021, : 8771 - 8780
  • [9] Self-Supervised Masking for Unsupervised Anomaly Detection and Localization
    Huang, Chaoqin
    Xu, Qinwei
    Wang, Yanfeng
    Wang, Yu
    Zhang, Ya
    [J]. IEEE TRANSACTIONS ON MULTIMEDIA, 2023, 25 : 4426 - 4438
  • [10] Kingma D.P., 2014, arXiv, DOI [DOI 10.48550/ARXIV.1412.6980, 10.48550/arXiv.1412.6980]