Rail Bolt Defect Detection Method for Rail Transport Systems in Hilly and Mountainous Areas Based on LHFSE-YOLOv11

Objective: To address small bolt defect targets, complex background interference, and limited edge deployment in hilly and mountainous rail transport environments, a detection method balancing accuracy, lightweight design, and real-time performance was proposed. Methods: A track image dataset containing missing bolts, loose bolts, and missing nuts was constructed. Based on YOLOv11m, HFERBC3K2 was developed by replacing the standard bottleneck in C3K2 with a High-Frequency Enhancement Residual Block to strengthen edge, texture, and local structural feature extraction. A Spectral Enhanced Feed-Forward module was introduced into C2PSA to form SEFFNC2PSA, enhancing defect-related frequency components and suppressing background interference through adaptive frequency-domain modulation. The integrated model was named HFSE-YOLOv11. Channel-level structured pruning was then applied, and the model with a pruning ratio of 0.5 was named LHFSE-YOLOv11. Results: On the validation set, HFSE-YOLOv11 achieved 91.7% precision, 93.4% recall, 91.3% mAP@0.5, and 80.2% mAP@0.5:0.95, improving upon YOLOv11m by 3.6, 2.2, 1.2, and 3.1 percentage points, respectively. After pruning, LHFSE-YOLOv11 had 15.9 M parameters, 53.8 GFLOPs, and a 32.5 MB model size, representing reductions of 16.3%, 14.3%, and 11.7%, while mAP@0.5 and mAP@0.5:0.95 decreased by only 0.3 and 0.9 percentage points. On the independent test set, it achieved 91.7% precision, 93.4% recall, 91.0% mAP@0.5, 79.3% mAP@0.5:0.95, and 81.5 FPS, outperforming all compared models in the four detection metrics. Conclusion: LHFSE-YOLOv11 balances accuracy, efficiency, and model size, supporting deployment on vehicle-mounted inspection terminals and resource-constrained edge devices.