THE OPTIMIZATION OF RAILWAY FASTENER DEFECT DETECTION VIA ACTIVATION FUNCTION ADAPTATIONS
Abstract
Manual control of rail defect detection is slow and costly. Deep learning methods can detect some of these defects to a certain extent. However, existing systems produce too many false positives due to environmental factors, resulting in labor and cost losses. One of the most important components in railway systems is the fastener, and their failure can lead to severe accidents. In this study, we developed a deep learning-based method that is designed to remain robust against foreign objects and environmental conditions when detecting railway fasteners. By employing various activation functions and expanding the training dataset through data augmentation techniques, our method significantly reduces false alarms. The best-performing activation function in our tests achieved an F1-score of 0.99 and a mean average precision (mAP) of 100%. Testing on a dataset provided by TCDD Railway Research & Technology Centre (DATEM) confirms the efficacy of our approach, demonstrating a notable decrease in unnecessary work and associated costs.
Keywords
YOLOv4, Railway component, Deep learning, Activation function, Fastener defect
Supporting Institution
Eskisehir Technical University Scientific Research Projects Commission
Project Number
21GAP081
Thanks
This research received funding from the Eskisehir Technical University Scientific Research Projects Commission, under grant number 21GAP081. The authors express their gratitude to TCDD Railway Research & Technology Centre (DATEM) for supplying the dataset.
References
- [1] Dinhobl G, Petz M, Kümmritz S, Hutterer H. On Measurement of Railway Noise: Usability of Acoustic Camera. In: Degrande G, Lombaert G, Anderson D, de Vos P, Gautier P-E, Iida M, Nelson JT, Nielsen JCO, Thompson DJ, Tielkes T, Towers DA, editors. Noise and Vibration Mitigation for Rail Transportation Systems. Cham: Springer International Publishing, 2021, pp. 234–241.
- [2] Assali P, Viguier F, Pollet N. Contribution of Terrestrial Laser Scanning for monitoring and inspection of railway infrastructure. In Proceedings of the World Congress on Railway Research; 2013 Nov 25-28; Sydney, Australia. 2013.
- [3] Szandała T. Review and comparison of commonly used activation functions for deep neural networks. In: Bio-inspired Neurocomputing 903, Bhoi AK, Mallick PK, Liu CM, Balas VE, editors. Studies in Computational Intelligence, vol. 903. Singapore: Springer Singapore, 2021, pp. 203–224.
- [4] Lin Y-W, Hsieh C-C, Huang W-H, Hsieh S-L, Hung W-H. Railway track fasteners fault detection using deep learning. In: 2019 IEEE Eurasia Conference on IOT, Communication and Engineering, 2019, pp. 187–190.
- [5] Qi H, Xu T, Wang G, Cheng Y, Chen C. MYOLOv3-Tiny: A new convolutional neural network architecture for real-time detection of track fasteners. Computers in Industry; 2020; 123: 103303.
- [6] Güçlü E, Aydin İ, Şahbaz K., Akin E, Karaköse M. Detection of defects in railway fasteners using YOLOv4 and fuzzy logic (article in Turkish with an abstract in English). Railway Engineering 2021; 14: 249-262.
- [7] Liao X, Lv S, Li D, Luo Y, Zhu Z, Jiang C. YOLOv4-MN3 for PCB surface defect detection. Applied Sciences 2021; 11(24):1-17.
- [8] Şener A, Ergen B, Toğaçar M. Fault detection from images of railroad lines using the deep learning model built with the tensorflow library. Turkish Journal of Science & Technology 2022; 17(1): 47-53.
- [9] Ozdemir R, Koc M. On the enhancement of semi-supervised deep learning-based railway defect detection using pseudo-labels. Expert Systems With Applications 2024; 251: 124105.
- [10] Sevi̇ M, Aydın İ, Karaköse M. Classification of railway fasteners by deep learning methods (article in Turkish with an abstract in English). European Journal of Science and Technology 2022; 35: 268-274.