Four Discriminator Cycle-Consistent Adversarial Network for Improving Railway Defective Fastener Inspection

Abstract

This article aims to improve the performance of deep learning-based defective fastener inspection method. Due to the defective fasteners are insufficient and far less than the defect-free ones in real railway, it is difficult to train a robust fastener inspection model on such imbalanced dataset. In view of this problem, a novel image generation method called four-discriminator cycle-consistent adversarial network (FD-Cycle-GAN) is proposed to generate the defect fastener images using a large number of defect-free ones. Extensive experiments are conducted on the real fastener images and generated images. Experimental results demonstrate that the defect fastener images generated by our proposed method have better quality and richer diversity than those generated by other state-of-the-art methods. In addition, compared with the CNN-only baseline, the performance of the fastener inspection model trained on the expanded dataset containing the defect fastener images generated by FD-Cycle-GAN is improved significantly. The detection accuracy and relative IMP reach 93.25% and 21.59% respectively.