MHS-DETR: A Railway Frog Defect Detection Model Integrating Multi-Scale Edge Perception and Adaptive Structural Optimization

Abstract As a critical component of the railway track system, the structural integrity of railway frogs is vital to the safe operation of trains. However, detecting surface defects on frogs remains challenging due to their small size, blurred edges, and complex backgrounds, which often lead to low detection accuracy in existing algorithms. To address these issues, this paper proposes a novel object detection model named MHS-DETR, which integrates multi-scale edge perception with adaptive optimization strategies. Built upon the RT-DETR framework, MHS-DETR introduces three systematic improvements: First, a lightweight Multi-scale Edge Information Extraction Network (MEIENet) is developed to significantly enhance the perception of small defect contours by integrating edge enhancement and dual-domain selection mechanisms. Second, a Hierarchical Attention Fusion Block (HAFB) is designed to improve the interaction and fusion of heterogeneous features through a dynamically weighted combination of local and global attention mechanisms, thereby enhancing the representation of multi-scale targets. Third, an adaptive loss function, SlideVarifocalLoss (SVL), is proposed to alleviate the imbalance between positive and negative samples by dynamically adjusting the loss weights of samples with ambiguous boundaries. To validate the effectiveness of the proposed model, a multi-scenario frog defect dataset named ForkDefect-2089—containing 2089 annotated images—was constructed. Experimental results show that MHS-DETR achieves an mAP50 of 70.3%, outperforming the baseline model by 4.2%, while reducing the number of parameters and computational cost by 15.6% and 7.4%, respectively. Moreover, it achieves a 2.3% performance improvement on the public NEU-DET dataset. Overall, the proposed MHS-DETR demonstrates superior robustness and generalization capabilities compared to mainstream detectors, offering strong technical support and practical value for achieving high-reliability, intelligent railway maintenance.