IDEAS home Printed from https://ideas.repec.org/a/spr/joinma/v33y2022i4d10.1007_s10845-020-01710-x.html
   My bibliography  Save this article

Synthetic data augmentation for surface defect detection and classification using deep learning

Author

Listed:
  • Saksham Jain

    (Netaji Subhas University of Technology)

  • Gautam Seth

    (Netaji Subhas University of Technology)

  • Arpit Paruthi

    (Netaji Subhas University of Technology)

  • Umang Soni

    (Netaji Subhas University of Technology)

  • Girish Kumar

    (Delhi Technological University)

Abstract

Deep learning techniques, especially Convolutional Neural Networks (CNN), dominate the benchmarks for most computer vision tasks. These state-of-the-art results are typically obtained through supervised learning, for which large annotated datasets are required. However, acquiring such datasets for manufacturing applications remains a challenging proposition due to the time and costs involved in their collection. To overcome this disadvantage, a novel framework is proposed for data augmentation by creating synthetic images using Generative Adversarial Networks (GANs). The generator synthesizes new surface defect images from random noise which is trained over time to get realistic fakes. These synthetic images can be used further for training of classification algorithms. Three GAN architectures are trained, and the entire data augmentation pipeline is implemented for the Northeastern University (China) Classification (NEU-CLS) dataset for hot-rolled steel strips from NEU Surface Defect Database. The classification accuracy of a simple CNN architecture is measured on synthetic augmented data and further it is compared with similar state-of-the-arts. It is observed that the proposed GANs-based augmentation scheme significantly improves the performance of CNN for classification of surface defects. The classically augmented CNN yields sensitivity and specificity of 90.28% and 98.06% respectively. In contrast, the synthetically augmented CNN yields better results, with sensitivity and specificity of 95.33% and 99.16% respectively. Also, the use of GANs is demonstrated to disentangle the representation space and to add additional domain knowledge through synthetic augmentation that can be difficult to replicate through classic augmentation. The proposed framework demonstrates high generalization capability. It may be applied to other supervised surface inspection tasks, and thus facilitate the development of advanced vision-based inspection instruments for manufacturing applications.

Suggested Citation

  • Saksham Jain & Gautam Seth & Arpit Paruthi & Umang Soni & Girish Kumar, 2022. "Synthetic data augmentation for surface defect detection and classification using deep learning," Journal of Intelligent Manufacturing, Springer, vol. 33(4), pages 1007-1020, April.
  • Handle: RePEc:spr:joinma:v:33:y:2022:i:4:d:10.1007_s10845-020-01710-x
    DOI: 10.1007/s10845-020-01710-x
    as

    Download full text from publisher

    File URL: http://link.springer.com/10.1007/s10845-020-01710-x
    File Function: Abstract
    Download Restriction: Access to the full text of the articles in this series is restricted.

    File URL: https://libkey.io/10.1007/s10845-020-01710-x?utm_source=ideas
    LibKey link: if access is restricted and if your library uses this service, LibKey will redirect you to where you can use your library subscription to access this item
    ---><---

    As the access to this document is restricted, you may want to search for a different version of it.

    References listed on IDEAS

    as
    1. Maciej Grzenda & Andres Bustillo, 2019. "Semi-supervised roughness prediction with partly unlabeled vibration data streams," Journal of Intelligent Manufacturing, Springer, vol. 30(2), pages 933-945, February.
    2. Olatomiwa Badmos & Andreas Kopp & Timo Bernthaler & Gerhard Schneider, 2020. "Image-based defect detection in lithium-ion battery electrode using convolutional neural networks," Journal of Intelligent Manufacturing, Springer, vol. 31(4), pages 885-897, April.
    3. Domen Tabernik & Samo Šela & Jure Skvarč & Danijel Skočaj, 2020. "Segmentation-based deep-learning approach for surface-defect detection," Journal of Intelligent Manufacturing, Springer, vol. 31(3), pages 759-776, March.
    4. Te-Hsiu Sun & Fang-Cheng Tien & Fang-Chih Tien & Ren-Jieh Kuo, 2016. "Automated thermal fuse inspection using machine vision and artificial neural networks," Journal of Intelligent Manufacturing, Springer, vol. 27(3), pages 639-651, June.
    Full references (including those not matched with items on IDEAS)

    Citations

    Citations are extracted by the CitEc Project, subscribe to its RSS feed for this item.
    as


    Cited by:

    1. Li Wei & Mahmud Iwan Solihin & Sarah ‘Atifah Saruchi & Winda Astuti & Lim Wei Hong & Ang Chun Kit, 2024. "Surface Defects Detection of Cylindrical High-Precision Industrial Parts Based on Deep Learning Algorithms: A Review," SN Operations Research Forum, Springer, vol. 5(3), pages 1-71, September.
    2. Songling Huang & Lisha Peng & Hongyu Sun & Shisong Li, 2023. "Deep Learning for Magnetic Flux Leakage Detection and Evaluation of Oil & Gas Pipelines: A Review," Energies, MDPI, vol. 16(3), pages 1-27, January.
    3. Erica Espinosa & Alvaro Figueira, 2023. "On the Quality of Synthetic Generated Tabular Data," Mathematics, MDPI, vol. 11(15), pages 1-18, July.

    Most related items

    These are the items that most often cite the same works as this one and are cited by the same works as this one.
    1. Yuanyuan Wang & Ling Ma & Lihua Jian & Huiqin Jiang, 2023. "Conductive particle detection via efficient encoder–decoder network," Journal of Intelligent Manufacturing, Springer, vol. 34(8), pages 3563-3577, December.
    2. Swarit Anand Singh & K. A. Desai, 2023. "Automated surface defect detection framework using machine vision and convolutional neural networks," Journal of Intelligent Manufacturing, Springer, vol. 34(4), pages 1995-2011, April.
    3. Ruiyang Hao & Bingyu Lu & Ying Cheng & Xiu Li & Biqing Huang, 2021. "A steel surface defect inspection approach towards smart industrial monitoring," Journal of Intelligent Manufacturing, Springer, vol. 32(7), pages 1833-1843, October.
    4. Xinyu Suo & Jian Liu & Licheng Dong & Chen Shengfeng & Lu Enhui & Chen Ning, 2022. "A machine vision-based defect detection system for nuclear-fuel rod groove," Journal of Intelligent Manufacturing, Springer, vol. 33(6), pages 1649-1663, August.
    5. Zeqing Yang & Mingxuan Zhang & Yingshu Chen & Ning Hu & Lingxiao Gao & Libing Liu & Enxu Ping & Jung Il Song, 2024. "Surface defect detection method for air rudder based on positive samples," Journal of Intelligent Manufacturing, Springer, vol. 35(1), pages 95-113, January.
    6. Feiyang Li & Nian Cai & Xueliang Deng & Jiahao Li & Jianfa Lin & Han Wang, 2022. "Serial number inspection for ceramic membranes via an end-to-end photometric-induced convolutional neural network framework," Journal of Intelligent Manufacturing, Springer, vol. 33(5), pages 1373-1392, June.
    7. Nhat-To Huynh & Duong-Dong Ho & Hong-Nguyen Nguyen, 2023. "An Approach for Designing an Optimal CNN Model Based on Auto-Tuning GA with 2D Chromosome for Defect Detection and Classification," Sustainability, MDPI, vol. 15(6), pages 1-14, March.
    8. Chi Ma & Hongquan Gui & Jialan Liu, 2023. "Self learning-empowered thermal error control method of precision machine tools based on digital twin," Journal of Intelligent Manufacturing, Springer, vol. 34(2), pages 695-717, February.
    9. Andres Bustillo & Roberto Reis & Alisson R. Machado & Danil Yu. Pimenov, 2022. "Improving the accuracy of machine-learning models with data from machine test repetitions," Journal of Intelligent Manufacturing, Springer, vol. 33(1), pages 203-221, January.
    10. Abtin Djavadifar & John Brandon Graham-Knight & Marian Kӧrber & Patricia Lasserre & Homayoun Najjaran, 2022. "Automated visual detection of geometrical defects in composite manufacturing processes using deep convolutional neural networks," Journal of Intelligent Manufacturing, Springer, vol. 33(8), pages 2257-2275, December.
    11. Shuai Ma & Kechen Song & Menghui Niu & Hongkun Tian & Yunhui Yan, 2024. "Cross-scale fusion and domain adversarial network for generalizable rail surface defect segmentation on unseen datasets," Journal of Intelligent Manufacturing, Springer, vol. 35(1), pages 367-386, January.
    12. Li Wei & Mahmud Iwan Solihin & Sarah ‘Atifah Saruchi & Winda Astuti & Lim Wei Hong & Ang Chun Kit, 2024. "Surface Defects Detection of Cylindrical High-Precision Industrial Parts Based on Deep Learning Algorithms: A Review," SN Operations Research Forum, Springer, vol. 5(3), pages 1-71, September.
    13. Zichen Bai & Junfeng Jing, 2024. "Mobile-Deeplab: a lightweight pixel segmentation-based method for fabric defect detection," Journal of Intelligent Manufacturing, Springer, vol. 35(7), pages 3315-3330, October.
    14. Andres Bustillo & Danil Yu. Pimenov & Mozammel Mia & Wojciech Kapłonek, 2021. "Machine-learning for automatic prediction of flatness deviation considering the wear of the face mill teeth," Journal of Intelligent Manufacturing, Springer, vol. 32(3), pages 895-912, March.
    15. Kai Liao & Wenjun Wang & Xuesong Mei & Wenwen Tian & Hai Yuan & Mingqiong Wang & Bozhe Wang, 2023. "Shape regulation of tapered microchannels in silica glass ablated by femtosecond laser with theoretical modeling and machine learning," Journal of Intelligent Manufacturing, Springer, vol. 34(7), pages 2907-2924, October.
    16. Omid Davtalab & Ali Kazemian & Xiao Yuan & Behrokh Khoshnevis, 2022. "Automated inspection in robotic additive manufacturing using deep learning for layer deformation detection," Journal of Intelligent Manufacturing, Springer, vol. 33(3), pages 771-784, March.
    17. Keyur D. Joshi & Vedang Chauhan & Brian Surgenor, 2020. "A flexible machine vision system for small part inspection based on a hybrid SVM/ANN approach," Journal of Intelligent Manufacturing, Springer, vol. 31(1), pages 103-125, January.
    18. Danqing Kang & Jianhuang Lai & Junyong Zhu & Yu Han, 2023. "An adaptive feature reconstruction network for the precise segmentation of surface defects on printed circuit boards," Journal of Intelligent Manufacturing, Springer, vol. 34(7), pages 3197-3214, October.
    19. Changqing Wang & Maoxuan Sun & Yuan Cao & Kunyu He & Bei Zhang & Zhonghao Cao & Meng Wang, 2023. "Lightweight Network-Based Surface Defect Detection Method for Steel Plates," Sustainability, MDPI, vol. 15(4), pages 1-12, February.
    20. Chun Fai Lui & Ahmed Maged & Min Xie, 2024. "A novel image feature based self-supervised learning model for effective quality inspection in additive manufacturing," Journal of Intelligent Manufacturing, Springer, vol. 35(7), pages 3543-3558, October.

    Corrections

    All material on this site has been provided by the respective publishers and authors. You can help correct errors and omissions. When requesting a correction, please mention this item's handle: RePEc:spr:joinma:v:33:y:2022:i:4:d:10.1007_s10845-020-01710-x. See general information about how to correct material in RePEc.

    If you have authored this item and are not yet registered with RePEc, we encourage you to do it here. This allows to link your profile to this item. It also allows you to accept potential citations to this item that we are uncertain about.

    If CitEc recognized a bibliographic reference but did not link an item in RePEc to it, you can help with this form .

    If you know of missing items citing this one, you can help us creating those links by adding the relevant references in the same way as above, for each refering item. If you are a registered author of this item, you may also want to check the "citations" tab in your RePEc Author Service profile, as there may be some citations waiting for confirmation.

    For technical questions regarding this item, or to correct its authors, title, abstract, bibliographic or download information, contact: Sonal Shukla or Springer Nature Abstracting and Indexing (email available below). General contact details of provider: http://www.springer.com .

    Please note that corrections may take a couple of weeks to filter through the various RePEc services.

    IDEAS is a RePEc service. RePEc uses bibliographic data supplied by the respective publishers.