Dual-Path Reconstruction Guided Segmentation Network for Unsupervised Anomaly Detection and Localization

被引:1
|
作者
Xiao, Junwei [1 ]
Deng, Lei [2 ]
Chen, Zhixiang [3 ]
Li, Xiu [1 ]
Chen, Baohua [4 ]
Yin, Hanxi [1 ]
机构
[1] Tsinghua Univ, Shenzhen Int Grad Sch, Shenzhen, Peoples R China
[2] Beijing Informat Sci & Technol Univ, Sch Instrument Sci & Optoelect Engn, Beijing, Peoples R China
[3] Univ Sheffield, Dept Comp Sci, Sheffield, S Yorkshire, England
[4] Tsinghua Univ, Dept Automat, Beijing, Peoples R China
关键词
Unsupervised; Anomaly detection; Pixel-level localization; Dual-path reconstruction; MVTecAD;
D O I
10.1109/IJCNN54540.2023.10191880
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Visual anomaly detection methods with localization are critically important for industrial manufacturing quality control. Because of the rarity of anomalies and the irregular variation of anomaly patterns, unsupervised methods have been widely explored. For the anomaly detection and localization tasks, the image reconstruction-based approaches have shown competitive performance. However, reconstruction results of those methods are coarse and visually blurred, which leads to a high rate of false detection and false pixel-level localization. To address those issues, we propose a framework called Dual-Path Reconstruction Guided Segmentation Network (DRGS-Net), which determines the abnormal regions by segmenting the anomaly image with its reconstructed result as the reference template. DRGS-Net mainly consists of a novel dual-path reconstruction sub-network and a specifically designed anomaly segmentation sub-network. They are jointly trained end-to-end, with the former fusing texture repair and image reconstruction information to obtain fine-grained reconstruction results and the latter learning a decision boundary between normal and anomalous regions based on reconstruction results. On the standard benchmark dataset MVTecAD and an additional dataset DAGM, DRGS-Net shows competitive performance in image-level detection and achieves outstanding improvement in pixel-level localization. Further experiments with the few samples setting demonstrate that DRGS-Net retains strong performance with only few anomaly-free training images.
引用
收藏
页数:8
相关论文
共 50 条
  • [41] Unsupervised anomaly segmentation via deep feature reconstruction
    Shi, Yong
    Yang, Jie
    Qi, Zhiquan
    NEUROCOMPUTING, 2021, 424 : 9 - 22
  • [42] Unsupervised Anomaly Detection of Industrial Images Based on Dual Generator Reconstruction Networks
    Gu, Cong
    Ren, Siyv
    Duan, Qiqiang
    INFORMATION TECHNOLOGY AND CONTROL, 2024, 53 (02): : 331 - 341
  • [43] Unsupervised Brain Anomaly Detection and Segmentation with Transformers
    Pinaya, Walter Hugo Lopez
    Tudosiu, Petru-Daniel
    Gray, Robert
    Rees, Geraint
    Nachev, Parashkev
    Ourselin, Sebastien
    Cardoso, M. Jorge
    MEDICAL IMAGING WITH DEEP LEARNING, VOL 143, 2021, 143 : 596 - 617
  • [44] Unsupervised Anomaly Detection and Segmentation on Dirty Datasets
    Guo, Jiahao
    Yu, Xiaohuo
    Wang, Lu
    FUTURE INTERNET, 2022, 14 (03):
  • [45] Dual-path multi-scale context dense aggregation network for retinal vessel segmentation
    Zhou, Wei
    Bai, Weiqi
    Ji, Jianhang
    Yi, Yugen
    Zhang, Ningyi
    Cui, Wei
    COMPUTERS IN BIOLOGY AND MEDICINE, 2023, 164
  • [46] Brain tumor segmentation based on the dual-path network of multi-modal MRI images
    Fang, Lingling
    Wang, Xin
    PATTERN RECOGNITION, 2022, 124
  • [47] Dual-Path Adversarial Learning for Fully Convolutional Network (FCN)-Based Medical Image Segmentation
    Bi, Lei
    Feng, Dagan
    Kim, Jinman
    VISUAL COMPUTER, 2018, 34 (6-8): : 1043 - 1052
  • [48] Attention-based dual-path feature fusion network for automatic skin lesion segmentation
    Zhenxiang He
    Xiaoxia Li
    Yuling Chen
    Nianzu Lv
    Yong Cai
    BioData Mining, 16
  • [49] Attention-based dual-path feature fusion network for automatic skin lesion segmentation
    He, Zhenxiang
    Li, Xiaoxia
    Chen, Yuling
    Lv, Nianzu
    Cai, Yong
    BIODATA MINING, 2023, 16 (01)
  • [50] Automatic segmentation of breast cancer histological images based on dual-path feature extraction network
    Lu, Xi
    Zhu, Xuedong
    MATHEMATICAL BIOSCIENCES AND ENGINEERING, 2022, 19 (11) : 11137 - 11153