Data Augmentation Method Based on Partial Noise Diffusion Strategy for One-Class Defect Detection Task

被引:0
作者
Chen, Weiwen [1 ]
Zhang, Yong [1 ,2 ]
Ke, Wenlong [1 ]
机构
[1] Huzhou Univ, Sch Informat Engn, Huzhou, Peoples R China
[2] Liaoning Normal Univ, Sch Comp & Informat Technol, Dalian, Peoples R China
来源
ALGORITHMS AND ARCHITECTURES FOR PARALLEL PROCESSING, ICA3PP 2023, PT VI | 2024年 / 14492卷
基金
中国国家自然科学基金;
关键词
Defect detection; Denoising diffusion probability model; Data augmentation; Deep learning; Image generation;
D O I
10.1007/978-981-97-0811-6_25
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
One-class defect detection has proven to be an effective technique. However, the performance of complex models is often limited by existing data augmentation methods. To address this issue, this paper proposes a novel data augmentation method based on a denoising diffusion probability model. This approach generates high-quality image samples using partial noise diffusion, eliminating the need for extensive training on large-scale datasets. Experimental results demonstrate that the proposed method outperforms current methods in one-class defect detection tasks. The proposed method offers a new perspective on data augmentation and demonstrates its potential to tackle challenging computer vision problems.
引用
收藏
页码:418 / 433
页数:16
相关论文
共 28 条
  • [1] Ahuja NA, 2019, Arxiv, DOI arXiv:1909.11786
  • [2] GANomaly: Semi-supervised Anomaly Detection via Adversarial Training
    Akcay, Samet
    Atapour-Abarghouei, Amir
    Breckon, Toby P.
    [J]. COMPUTER VISION - ACCV 2018, PT III, 2019, 11363 : 622 - 637
  • [3] Almahairi A, 2018, PR MACH LEARN RES, V80
  • [4] 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
  • [5] Bochkovskiy A, 2020, Arxiv, DOI arXiv:2004.10934
  • [6] 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
  • [7] Dhariwal P, 2021, ADV NEUR IN, V34
  • [8] 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
  • [9] Deep Residual Learning for Image Recognition
    He, Kaiming
    Zhang, Xiangyu
    Ren, Shaoqing
    Sun, Jian
    [J]. 2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, : 770 - 778
  • [10] Ho J., 2020, Advances in Neural Information Processing Systems, V33, P6840