AUTHOR=Zhendong He , Xiangyang Gao , Zhiyuan Liu , Xiaoyu An , Anping Zheng TITLE=Rail surface defect data enhancement method based on improved ACGAN JOURNAL=Frontiers in Neurorobotics VOLUME=18 YEAR=2024 URL=https://www.frontiersin.org/journals/neurorobotics/articles/10.3389/fnbot.2024.1397369 DOI=10.3389/fnbot.2024.1397369 ISSN=1662-5218 ABSTRACT=

Rail surface defects present a significant safety concern in railway operations. However, the scarcity of data poses challenges for employing deep learning in defect detection. This study proposes an enhanced ACGAN augmentation method to address these issues. Residual blocks mitigate vanishing gradient problems, while a spectral norm regularization-constrained discriminator improves stability and image quality. Substituting the generator’s deconvolution layer with upsampling and convolution operations enhances computational efficiency. A gradient penalty mechanism based on regret values addresses gradient abnormality concerns. Experimental validation demonstrates superior image clarity and classification accuracy compared to ACGAN, with a 17.6% reduction in FID value. MNIST dataset experiments verify the model’s generalization ability. This approach offers practical value for real-world applications.