Research on Defect Detection on Steel Rails Based on Improved YOLO11n Algorithm

Aiming at the core issues of the traditional YOLO11n model in rail surface defect detection—fine-grained feature loss of small defects, insufficient micro-target recognition accuracy, and the mismatch of existing downsampling/fusion methods for micro-defect feature extraction—this paper proposes an improved YOLO11n algorithm with two-dimensional network structure innovations. First, the Adaptive Downsampling (ADown) module is introduced into the backbone network for the first time, retaining global features via 2D average pooling and extracting local details through channel-split multi-path convolution/max pooling to avoid fine texture loss. Second, the original SOEP-RFPN-MFM neck network is designed, integrating SNI, GSConvE and MFM modules to achieve dynamic weighted fusion of multi-scale features and break the bottleneck of inefficient small-target feature aggregation. Trained and verified on a 4020-image rail dataset covering four defect types (Spalling, Squat, Wheel Burns, Corrugation), the improved algorithm achieves 93.7% detection accuracy, 92.4% recall and 95.6% mAP, realizing incremental improvements of 1.2, 2.6 and 0.8 percentage points, respectively, compared with the original YOLO11n, which is particularly optimized for rail micro-defect detection scenarios. This study provides a new deep learning method for rail transit micro-defect detection and a reference for scenario-specific improvement of lightweight YOLO11n models.