Deep Data Augmentation for Defect Detection Enhancement: A Diffusion Model Based Approach
DOI: 10.23977/acss.2024.080114 | Downloads: 9 | Views: 100
Author(s)
Yi Peng 1, Yaping Zhang 1
Affiliation(s)
1 School of Information Science and Technology, Yunnan Normal University, Kunming, Yunnan, 650500, China
Corresponding Author
Yaping ZhangABSTRACT
Weld defect detection is a crucial step in industrial production processes. To effectively identify these defects, the X-ray inspection method based on non-destructive testing is commonly employed. Addressing the challenges of limited sample size and class imbalance in X-ray images, this study proposes an enhanced diffusion model algorithm to augment samples, thereby improving the defect detection capability for rare categories. Experimental results prove the enhanced dataset's detection performance surpassing that of the original dataset. The improvement is notable, with a 5.1% enhancement on the WDD dataset. This paper presents a viable data augmentation solution for small-sample weld seam defect detection.
KEYWORDS
Weld defect detection, data augmentation, GAN, diffusion modelCITE THIS PAPER
Yi Peng, Yaping Zhang, Deep Data Augmentation for Defect Detection Enhancement: A Diffusion Model Based Approach. Advances in Computer, Signals and Systems (2024) Vol. 8: 122-126. DOI: http://dx.doi.org/10.23977/acss.2024.080114.
REFERENCES
[1] Nacereddine N, Zelmat M, Belaifa S S, et al. Weld defect detection in industrial radiography based digital image processing [J]. Transactions on Engineering Computing and Technology, 2005, 2: 145-148.
[2] Goodfellow I, Pouget-Abadie J, Mirza M, et al. Generative adversarial networks[J]. Communications of the ACM, 2020, 63(11): 139-144.
[3] Kingma D P, Welling M. Auto-encoding variational bayes [J]. arXiv preprint arXiv:1312.6114, 2013.
[4] Ho J, Jain A, Abbeel P. Denoising diffusion probabilistic models [J]. Advances in neural information processing systems, 2020, 33: 6840-6851.
[5] Zhong Z, Zheng L, Kang G, et al. Random erasing data augmentation[C]//Proceedings of the AAAI conference on artificial intelligence. 2020, 34(07): 13001-13008.
[6] Moosavi-Dezfooli S M, Fawzi A, Frossard P. Deepfool: a simple and accurate method to fool deep neural networks[C]//Proceedings of the IEEE conference on computer vision and pattern recognition. 2016: 2574-2582.
[7] Gatys L A, Ecker A S, Bethge M. A neural algorithm of artistic style [J]. arXiv preprint arXiv:1508.06576, 2015.
[8] Inoue H. Data augmentation by pairing samples for images classification [J]. arXiv preprint arXiv:1801.02929, 2018.
[9] Konno T, Iwazume M. Icing on the cake: An easy and quick post-learnig method you can try after deep learning[J]. arXiv preprint arXiv:1807.06540, 2018.
[10] Moreno-Barea F J, Strazzera F, Jerez J M, et al. Forward noise adjustment scheme for data augmentation[C]// 2018 IEEE symposium series on computational intelligence (SSCI). IEEE, 2018: 728-734.
[11] Bowles C, Chen L, Guerrero R, et al. Gan augmentation: Augmenting training data using generative adversarial networks [J]. arXiv preprint arXiv:1810.10863, 2018.
[12] Frid-Adar M, Diamant I, Klang E, et al. GAN-based synthetic medical image augmentation for increased CNN performance in liver lesion classification[J]. Neurocomputing, 2018, 321: 321-331.
[13] Jiangsha A, Tian L, Bai L, et al. Data augmentation by a CycleGAN-based extra-supervised model for nondestructive testing[J]. Measurement Science and Technology, 2022, 33(4): 045017.
[14] Mery D, Riffo V, Zscherpel U, et al. GDXray: The database of X-ray images for nondestructive testing [J]. Journal of Nondestructive Evaluation, 2015, 34(4): 42.
[15] Nichol A Q, Dhariwal P. Improved denoising diffusion probabilistic models[C]//International Conference on Machine Learning. PMLR, 2021: 8162-8171.
[16] Ronneberger O, Fischer P, Brox T. U-net: Convolutional networks for biomedical image segmentation[C]// Medical Image Computing and Computer-Assisted Intervention–MICCAI 2015: 18th International Conference, Munich, Germany, October 5-9, 2015, Proceedings, Part III 18. Springer International Publishing, 2015: 234-241.
[17] Wang H, Cao P, Wang J, et al. Uctransnet: rethinking the skip connections in u-net from a channel-wise perspective with transformer[C]//Proceedings of the AAAI conference on artificial intelligence. 2022, 36(3): 2441-2449.
[18] Chen L C, Papandreou G, Schroff F, et al. Rethinking atrous convolution for semantic image segmentation[J]. arXiv preprint arXiv:1706.05587, 2017.
[19] Hou Q, Zhou D, Feng J. Coordinate attention for efficient mobile network design[C]//Proceedings of the IEEE/CVF conference on computer vision and pattern recognition. 2021: 13713-13722.
Downloads: | 13688 |
---|---|
Visits: | 260498 |
Sponsors, Associates, and Links
-
Power Systems Computation
-
Internet of Things (IoT) and Engineering Applications
-
Computing, Performance and Communication Systems
-
Journal of Artificial Intelligence Practice
-
Journal of Network Computing and Applications
-
Journal of Web Systems and Applications
-
Journal of Electrotechnology, Electrical Engineering and Management
-
Journal of Wireless Sensors and Sensor Networks
-
Journal of Image Processing Theory and Applications
-
Mobile Computing and Networking
-
Vehicle Power and Propulsion
-
Frontiers in Computer Vision and Pattern Recognition
-
Knowledge Discovery and Data Mining Letters
-
Big Data Analysis and Cloud Computing
-
Electrical Insulation and Dielectrics
-
Crypto and Information Security
-
Journal of Neural Information Processing
-
Collaborative and Social Computing
-
International Journal of Network and Communication Technology
-
File and Storage Technologies
-
Frontiers in Genetic and Evolutionary Computation
-
Optical Network Design and Modeling
-
Journal of Virtual Reality and Artificial Intelligence
-
Natural Language Processing and Speech Recognition
-
Journal of High-Voltage
-
Programming Languages and Operating Systems
-
Visual Communications and Image Processing
-
Journal of Systems Analysis and Integration
-
Knowledge Representation and Automated Reasoning
-
Review of Information Display Techniques
-
Data and Knowledge Engineering
-
Journal of Database Systems
-
Journal of Cluster and Grid Computing
-
Cloud and Service-Oriented Computing
-
Journal of Networking, Architecture and Storage
-
Journal of Software Engineering and Metrics
-
Visualization Techniques
-
Journal of Parallel and Distributed Processing
-
Journal of Modeling, Analysis and Simulation
-
Journal of Privacy, Trust and Security
-
Journal of Cognitive Informatics and Cognitive Computing
-
Lecture Notes on Wireless Networks and Communications
-
International Journal of Computer and Communications Security
-
Journal of Multimedia Techniques
-
Automation and Machine Learning
-
Computational Linguistics Letters
-
Journal of Computer Architecture and Design
-
Journal of Ubiquitous and Future Networks