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 条
  • [11] CFA: Coupled-Hypersphere-Based Feature Adaptation for Target-Oriented Anomaly Localization
    Lee, Sungwook
    Lee, Seunghyun
    Song, Byung Cheol
    [J]. IEEE ACCESS, 2022, 10 : 78446 - 78454
  • [12] CutPaste: Self-Supervised Learning for Anomaly Detection and Localization
    Li, Chun-Liang
    Sohn, Kihyuk
    Yoon, Jinsung
    Pfister, Tomas
    [J]. 2021 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR 2021, 2021, : 9659 - 9669
  • [13] Hone: Mitigating Stragglers in Distributed Stream Processing With Tuple Scheduling
    Li, Wenxin
    Liu, Duowen
    Chen, Kai
    Li, Keqiu
    Qi, Heng
    [J]. IEEE TRANSACTIONS ON PARALLEL AND DISTRIBUTED SYSTEMS, 2021, 32 (08) : 2021 - 2034
  • [14] Li WX, 2020, IEEE INFOCOM SER, P1319, DOI [10.1109/INFOCOM41043.2020.9155511, 10.1109/infocom41043.2020.9155511]
  • [15] Luo Jing, 2014, Journal of Frontiers of Computer Science and Technology, V8, P1041, DOI 10.3778/j.issn.1673-9418.1405007
  • [16] Ma D., 2021, Chin. J. Image Graph, V26, P487
  • [17] U-Net: Convolutional Networks for Biomedical Image Segmentation
    Ronneberger, Olaf
    Fischer, Philipp
    Brox, Thomas
    [J]. MEDICAL IMAGE COMPUTING AND COMPUTER-ASSISTED INTERVENTION, PT III, 2015, 9351 : 234 - 241
  • [18] Towards Total Recall in Industrial Anomaly Detection
    Roth, Karsten
    Pemula, Latha
    Zepeda, Joaquin
    Scholkopf, Bernhard
    Brox, Thomas
    Gehler, Peter
    [J]. 2022 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2022, : 14298 - 14308
  • [19] f-AnoGAN: Fast unsupervised anomaly detection with generative adversarial networks
    Schlegl, Thomas
    Seebock, Philipp
    Waldstein, Sebastian M.
    Langs, Georg
    Schmidt-Erfurth, Ursula
    [J]. MEDICAL IMAGE ANALYSIS, 2019, 54 : 30 - 44
  • [20] Unsupervised Anomaly Detection with Generative Adversarial Networks to Guide Marker Discovery
    Schlegl, Thomas
    Seeboeck, Philipp
    Waldstein, Sebastian M.
    Schmidt-Erfurth, Ursula
    Langs, Georg
    [J]. INFORMATION PROCESSING IN MEDICAL IMAGING (IPMI 2017), 2017, 10265 : 146 - 157