Railway fastener defect detection using RFD-DETR: A lightweight real-time transformer-based approach

Huixiang Zhou, Yuhao Liu and Jian Wang

PLOS ONE, 2025, vol. 20, issue 11, 1-25

Abstract: Railway fasteners play a crucial role in ensuring track stability and safety; however, manual inspection is both inefficient and prone to errors. Deep learning models, such as YOLO, are widely utilized for defect detection. However, these methods demand comparatively larger model sizes, greater computational power, and more storage capacity, yet their defect feature extraction remains inadequate. To overcome these challenges, this research introduces RFD-DETR, an enhanced detection transformer model specifically optimized for real-time rail fastener defect identification. The model incorporates three distinct modules to facilitate multi-scale feature extraction, enhance model efficiency, and improve defect detection. Firstly, a wavelet transform convolution module (WTConv) is employed, which integrates a wavelet transform to enhance multi-scale feature extraction while reducing model computation. Secondly, a cross-scale feature fusion module (CSPPDC) is utilised, incorporating channel gated attention downsampling (CGAD) to refine defect detection. Finally, a wavelet transform feature upgrading (WFU) module is integrated within the neck module, enhancing feature fusion and contributing to the overall efficacy of the model. Experimental findings based on an expanded rail fastener dataset indicate that RFD-DETR achieves a 98.27% mean average precision (mAP) when evaluated at an IoU threshold of 0.5, outperforming the baseline model. Furthermore, it lowers computational expenses by 18.8% and reduces parameter count by 14.7%.

Date: 2025
References: View complete reference list from CitEc
Citations:

Downloads: (external link)
https://journals.plos.org/plosone/article?id=10.1371/journal.pone.0331513 (text/html)
https://journals.plos.org/plosone/article/file?id= ... 31513&type=printable (application/pdf)

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:plo:pone00:0331513

DOI: 10.1371/journal.pone.0331513

Access Statistics for this article

More articles in PLOS ONE from Public Library of Science
Bibliographic data for series maintained by plosone ().