结合领域知识运用机器学习对铁路接触网系统进行检测。
Inspection of railway catenary systems using machine learning with domain knowledge integration.
作者信息
Marciniak Kacper, Majewski Paweł, Reiner Jacek
机构信息
Faculty of Mechanical Engineering, Wrocław University of Science and Technology, ul. Łukasiewicza 5, 50-371, Wrocław, Poland.
Faculty of Information and Communication Technology, Wrocław University of Science and Technology, ul. Janiszewskiego 11/17, 50-372, Wrocław, Poland.
出版信息
Sci Rep. 2025 Aug 11;15(1):29426. doi: 10.1038/s41598-025-15289-x.
Railway catenary system inspection is a critical task where high accuracy and reliability are essential to ensure operational efficiency and safety. This objective is achieved by assessing the technical condition of the infrastructure and maintaining a comprehensive inventory of its components. The application of machine learning methods to this problem is non-trivial, due to various constraints, including the cost of data acquisition. This paper presents innovative solutions leveraging domain knowledge to significantly improve the inference quality of machine learning models using existing training data. Key innovations include a two-stage approach and clustering of selected objects to extract regions of interest (ROI), dynamic confidence score weighting, and ROI masking, aimed at reducing false positives and enhancing precision. Additionally, the system was extended with ensemble learning methods and custom test-time augmentations (TTA). Proposed methods substantially improve metrics such as AP50, precision, recall, and F1-score, particularly in detecting small and hard-to-spot catenary components such as insulators. Notably, the proposed enhancements were optimized to mitigate processing time increases, enabling their application in industrial settings. The results demonstrate the effectiveness of integrating domain knowledge into the process of machine vision inspection, achieving an improvement in the F1-score metric from 61.97 (0.59) to 82.53 (0.38) compared to the baseline single-model approach while maintaining practical runtime constraints for real-world overhead catenary system inspection.
铁路接触网系统检查是一项至关重要的任务,高精度和可靠性对于确保运营效率和安全至关重要。这一目标是通过评估基础设施的技术状况并维护其组件的全面清单来实现的。由于包括数据采集成本在内的各种限制,将机器学习方法应用于这个问题并非易事。本文提出了利用领域知识的创新解决方案,以显著提高使用现有训练数据的机器学习模型的推理质量。关键创新包括两阶段方法和对选定对象进行聚类以提取感兴趣区域(ROI)、动态置信度评分加权和ROI掩码,旨在减少误报并提高精度。此外,该系统还采用了集成学习方法和自定义测试时增强(TTA)进行了扩展。所提出的方法显著改善了诸如AP50、精度、召回率和F1分数等指标,特别是在检测诸如绝缘子等小型且难以发现的接触网组件方面。值得注意的是,所提出的增强措施经过优化以减轻处理时间的增加,从而使其能够应用于工业环境。结果表明,将领域知识集成到机器视觉检查过程中是有效的,与基线单模型方法相比,F1分数指标从61.97(0.59)提高到了82.53(0.38),同时在实际运行时保持了对实际架空接触网系统检查的限制。