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 条
[61]   hPSD: A Hybrid PU-Learning-Based Spammer Detection Model for Product Reviews [J].
Wu, Zhiang ;
Cao, Jie ;
Wang, Yaqiong ;
Wang, Youquan ;
Zhang, Lu ;
Wu, Junjie .
IEEE TRANSACTIONS ON CYBERNETICS, 2020, 50 (04) :1595-1606
[62]  
Xie G., 2022, 11 INT C LEARN REPR
[63]   Wood broken defect detection with laser profilometer based on Bi-LSTM network [J].
Xu, Zhezhuang ;
Lin, Ye ;
Chen, Dan ;
Yuan, Meng ;
Zhu, Yuhang ;
Ai, Zhijie ;
Yuan, Yazhou .
EXPERT SYSTEMS WITH APPLICATIONS, 2024, 242
[64]   Attribute Restoration Framework for Anomaly Detection [J].
Ye, Fei ;
Huang, Chaoqin ;
Cao, Jinkun ;
Li, Maosen ;
Zhang, Ya ;
Lu, Cewu .
IEEE TRANSACTIONS ON MULTIMEDIA, 2022, 24 :116-127
[65]  
Yu J., 2021, arXiv, DOI DOI 10.48550/ARXIV.2111.07677
[66]  
Zagoruyko S, 2016, BMVC
[67]   DRAEM - A discriminatively trained reconstruction embedding for surface anomaly detection [J].
Zavrtanik, Vitjan ;
Kristan, Matej ;
Skocaj, Danijel .
2021 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2021), 2021, :8310-8319
[68]  
Zhang Lingrui, 2023, P IEEECVF C COMPUTER, P4344
[69]   Dist-PU: Positive-Unlabeled Learning from a Label Distribution Perspective [J].
Zhao, Yunrui ;
Xu, Qianqian ;
Jiang, Yangbangyan ;
Wen, Peisong ;
Huang, Qingming .
2022 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2022), 2022, :14441-14450