<p>Rail track fastener detection faces edge feature degradation under complex conditions. However, existing research pays little attention to optimizing detection performance in edge feature degradation scenarios. This study proposes the YOLO-GEH algorithm, which uses an improved YOLO11 framework to robustly detect track fasteners. To solve the information loss caused by edge feature degradation, we designed the gradient-enhanced hierarchical edge control (GEH) module in the backbone network. Through a three-stage edge control mechanism of generation-preservation-fusion, it directionally strengthens and spreads edge features across layers, effectively relieving the gradient disappearance of shallow features in degradation scenarios. To boost feature discrimination under shape-deformation interference, we introduced the convolutional block attention module (CBAM). It strengthens key-area responses via adaptive weight allocation. Experimental results show that our YOLO-GEH model significantly improves detection accuracy. Specifically, the average accuracy increases by 2.1%. In the three defect detection tasks of deflection, occlusion, and shape deformation, the accuracy improves by 1.1%, 4.3%, and 8.1%, respectively. This proves our method is effective in accurately detecting rail track fasteners with edge feature degradation. The development and deployment of this model are underpinned by high-performance computing (HPC) paradigms, which are crucial for managing the computationally intensive training on large-scale datasets and enabling real-time, parallel processing for railway inspection systems.</p>

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

YOLO-GEH: rail track fastener detection with gradient hierarchical edge enhancement and attention fusion

  • Tangbo Bai,
  • Xiaolan Wang,
  • Yufei Wang,
  • Wangyi Li,
  • Houliang Xiang

摘要

Rail track fastener detection faces edge feature degradation under complex conditions. However, existing research pays little attention to optimizing detection performance in edge feature degradation scenarios. This study proposes the YOLO-GEH algorithm, which uses an improved YOLO11 framework to robustly detect track fasteners. To solve the information loss caused by edge feature degradation, we designed the gradient-enhanced hierarchical edge control (GEH) module in the backbone network. Through a three-stage edge control mechanism of generation-preservation-fusion, it directionally strengthens and spreads edge features across layers, effectively relieving the gradient disappearance of shallow features in degradation scenarios. To boost feature discrimination under shape-deformation interference, we introduced the convolutional block attention module (CBAM). It strengthens key-area responses via adaptive weight allocation. Experimental results show that our YOLO-GEH model significantly improves detection accuracy. Specifically, the average accuracy increases by 2.1%. In the three defect detection tasks of deflection, occlusion, and shape deformation, the accuracy improves by 1.1%, 4.3%, and 8.1%, respectively. This proves our method is effective in accurately detecting rail track fasteners with edge feature degradation. The development and deployment of this model are underpinned by high-performance computing (HPC) paradigms, which are crucial for managing the computationally intensive training on large-scale datasets and enabling real-time, parallel processing for railway inspection systems.