RAG-Enhanced Vision–Language Framework and Dataset for Railway Signal Cognition and Safety Reasoning
Railway scene understanding is critical for ensuring train operational safety and advancing intelligent railway systems. Existing railway vision methods mainly focus on perception and classification, while lacking regulation-guided semantic reasoning capabilities in complex environments. To address these limitations, this paper proposes a retrieval-augmented generation (RAG)-enhanced vision–language framework for railway signal cognition and safety reasoning. The proposed method integrates railway signal perception, regulatory knowledge retrieval, and multi-modal reasoning to improve factual consistency, reasoning reliability, and operational interpretability. In addition, a dedicated railway signal dataset comprising 500 standardized railway scene images with structured QA annotations is constructed to support regulation-oriented multi-modal recognition evaluation. Experimental results show that the proposed framework improves reasoning accuracy from 28.40% to 67.20% with an average end-to-end inference latency of 11.31 s per sample, and the inference speed can be further improved by adjusting experimental configurations to trade off between efficiency and accuracy, demonstrating the potential of RAG-enhanced architectures as a foundational step toward reliable multi-modal cognition in intelligent railway systems.
Authors
- Chao Fan (ORCID: https://orcid.org/0000-0002-4137-8224)
- Weiwei Li (ORCID: https://orcid.org/0000-0002-7329-4236)
- Sen Zhang (ORCID: https://orcid.org/0000-0002-0377-6985)
- J T Li
- Xize Guo
- Wenjun Wu
- Shiyi Xiong
- Chu Huang
- Qunbo Wang
Institutions
- Northwestern Polytechnical University (CN)
- Beijing Jiaotong University (CN)
- Beihang University (CN)
Publication Details
- Journal
- Computers
- Published
- 2026-06-29
- DOI
- https://doi.org/10.3390/computers15070416
- Primary Topic
- Railway Systems and Energy Efficiency
- Type
- article
- Field-Weighted Citation Impact
- 0.00
Funders
- Natural Science Foundation of Beijing Municipality
- Aviation Science Fund
- Shanghai Aerospace Science and Technology Innovation Foundation