Few-shot anomaly detection using positive unlabeled learning with cycle consistency and co-occurrence features

被引:5
作者
An, Sion [1 ,2 ]
Kim, Jaehong [5 ]
Kim, Soopil [1 ]
Chikontwe, Philip [3 ]
Jung, Jiwook [4 ]
Jeon, Hyejeong [4 ]
Park, Sang Hyun [1 ,2 ]
机构
[1] Daegu Gyeongbuk Inst Sci & Technol DGIST, Dept Robot & Mechatron Engn, Daegu, South Korea
[2] Univ Penn, Dept Biostat Epidemiol & Informat, Philadelphia, PA USA
[3] Harvard Med Sch, Dept Biomed Informat, Boston, MA USA
[4] LG Elect, Artificial Intelligence Lab, Seoul, South Korea
[5] Korea Adv Inst Sci & Technol KAIST, Sch Comp, Daejeon, South Korea
关键词
Anomaly detection; Defect detection; Few-shot learning; Positive unlabeled learning;
D O I
10.1016/j.eswa.2024.124890
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
In industrial manufacturing sites, defect detection is crucial to improve reliability and lower inspection costs. Though prior anomaly detectors have shown success, they rely on large amounts of labeled data. Now, we aim to answer "how to cost-effectively utilize limited data for defect detection?''. . Inspired by positive and unlabeled learning and few-shot learning, we propose a Positive Unlabeled learning based Few-shot Anomaly Detection model (PUFAD) that builds a representative memory bank of patch features using the large unlabeled set and few normal samples. Frequently co-occurring patch features in the unlabeled set, and cycle-consistent features between normal and unlabeled samples are regarded as pseudo normal features used for classifier training, including memory bank updates. Given test samples during inference, abnormal samples are detected by comparing the features in the memory bank via density estimation. Our method achieves state-of-theart performance compared to existing few-shot anomaly detection methods on two benchmarks. This work addresses the real-world challenges of collecting large datasets for visual anomaly detection by proposing a practical solution that requires only a few normal samples and unlabeled data.
引用
收藏
页数:12
相关论文
共 69 条
[11]   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
[12]  
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
[13]  
Dehaene D, 2020, Arxiv, DOI arXiv:2008.05369
[14]   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
[15]  
Dosovitskiy A., 2021, 9 INT C LEARN REPR I
[16]  
du Plessis MC, 2014, ADV NEUR IN, V27
[17]  
du Plessis MC, 2015, PR MACH LEARN RES, V37, P1386
[18]   CFLOW-AD: Real-Time Unsupervised Anomaly Detection with Localization via Conditional Normalizing Flows [J].
Gudovskiy, Denis ;
Ishizaka, Shun ;
Kozuka, Kazuki .
2022 IEEE WINTER CONFERENCE ON APPLICATIONS OF COMPUTER VISION (WACV 2022), 2022, :1819-1828
[19]   Positive-Unlabeled Data Purification in the Wild for Object Detection [J].
Guo, Jianyuan ;
Han, Kai ;
Wu, Han ;
Zhang, Chao ;
Chen, Xinghao ;
Xu, Chunjing ;
Xu, Chang ;
Wang, Yunhe .
2021 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR 2021, 2021, :2652-2661
[20]   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