IDEAS home Printed from https://ideas.repec.org/a/gam/jeners/v15y2022i10p3592-d815324.html
   My bibliography  Save this article

Self-Supervised Railway Surface Defect Detection with Defect Removal Variational Autoencoders

Author

Listed:
  • Yongzhi Min

    (School of Automation and Electrical Engineering, Lanzhou Jiaotong University, Lanzhou 730070, China)

  • Yaxing Li

    (School of Automation and Electrical Engineering, Lanzhou Jiaotong University, Lanzhou 730070, China)

Abstract

In railway surface defect detection applications, supervised deep learning methods suffer from the problems of insufficient defect samples and an imbalance between positive and negative samples. To overcome these problems, we propose a lightweight two-stage architecture including the railway cropping network (RC-Net) and defects removal variational autoencoder (DR-VAE), which requires only normal samples for training to achieve defect detection. First, we design a simple and effective RC-Net to extract railway surfaces accurately from railway inspection images. Second, the DR-VAE is proposed for background reconstruction of railway surface images to detect defects by self-supervised learning. Specifically, during the training process, DR-VAE contains a defect random mask module (D-RM) to generate self-supervised signals and uses a structural similarity index measure (SSIM) as pixel loss. In addition, the decoder of DR-VAE also acts as a discriminator to implement introspective adversarial training. In the inference stage, we reduce the random error of reconstruction by introducing a distribution capacity attenuation factor, and finally use the residuals of the original and reconstructed images to achieve segmentation of the defects. The experiments, including core parameter exploration and comparison with other models, indicate that the model can achieve a high detection accuracy.

Suggested Citation

  • Yongzhi Min & Yaxing Li, 2022. "Self-Supervised Railway Surface Defect Detection with Defect Removal Variational Autoencoders," Energies, MDPI, vol. 15(10), pages 1-15, May.
  • Handle: RePEc:gam:jeners:v:15:y:2022:i:10:p:3592-:d:815324
    as

    Download full text from publisher

    File URL: https://www.mdpi.com/1996-1073/15/10/3592/pdf
    Download Restriction: no

    File URL: https://www.mdpi.com/1996-1073/15/10/3592/
    Download Restriction: no
    ---><---

    Citations

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


    Cited by:

    1. Xiuhua Wang & Kun Yang & Yongzhi Min & Yongliang Wang, 2022. "Localization Method and Finite Element Modelling of the Mid-Point Anchor of High-Speed Railway Distributed in Long Straight Line with Large Slope," Energies, MDPI, vol. 15(16), pages 1-16, August.

    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:gam:jeners:v:15:y:2022:i:10:p:3592-:d:815324. 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.

    We have no bibliographic references for this item. You can help adding them by using 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: MDPI Indexing Manager (email available below). General contact details of provider: https://www.mdpi.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.