高级检索

    基于多尺度WideResNet的铁轨缺陷小样本检测算法

    A Few-Shot Rail Defect Detection Algorithm Based on Multi-Scale WideResNet Model

    • 摘要: 铁轨缺陷检测对铁路安全和降低维护成本非常重要。面对铁路网络扩张和缺陷样本稀缺带来的挑战,尤其是小样本条件下的过拟合问题,提出了一种基于多尺度WideResNet的小样本铁轨表面缺陷检测算法,通过数据增强等图像处理技术扩大有限的训练集,提高模型的泛化能力。利用迁移学习策略,将预训练的WideResNet模型提取多尺度特征应用于铁轨缺陷检测任务,减少对大量标注数据的依赖,加快小样本模型训练快速收敛。设计小样本深度学习模型策略,构建度量学习模块,从有限的标注数据中快速学习并进行有效的泛化。结果表明,该算法在10-shot小样本条件下能够有效地检测铁轨表面缺陷,模型精度mAP达到83.6%,召回率高达93.8%。本研究为铁轨表面缺陷自动化检测提供了新技术,对于提升铁路运输安全和经济性具有重大意义。

       

      Abstract: Rail defect detection is crucial for railway safety and reducing maintenance costs. Faced with the challenges posed by the expansion of the railway network and the scarcity of defect samples, particularly the issue of overfitting under few-shot conditions, this study proposes a few-shot deep learning algorithm for the detection of rail surface defects based on multi-scale WideResNet(Wide Residual Networks), which employs data augmentation techniques to expand the finite training set, thereby improving the model's generalization capability. By leveraging transfer learning strategies, it applies pre-trained deep learning models to rail defect detection tasks with extracting multi-scale features, reducing reliance on large annotated datasets. The study designs few-shot deep learning model strategies and constructing metric learning module that enable rapid learning from limited annotated data and effective generalization. Experimental results demonstrate that the algorithm can effectively detect rail surface defects under 10-shot conditions, achieving a model precision mAP of 83.6% and a recall rate as high as 93.8%. This research provides new technology for the automated detection of rail surface defects, which is of great significance for enhancing the safety and economy of railway transportation.

       

    /

    返回文章
    返回