TY - JOUR
T1 - RAG-Enhanced Vision–Language Framework and Dataset for Railway Signal Cognition and Safety Reasoning
AU - Wang, Qunbo
AU - Xiong, Shiyi
AU - Li, Jiawei
AU - Li, Weiliang
AU - Huang, Chu
AU - Zhang, Sen
AU - Guo, Xize
AU - Fan, Chao
AU - Wu, Wenjun
N1 - Publisher Copyright:
© 2026 by the authors.
PY - 2026/7
Y1 - 2026/7
N2 - 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.
AB - 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.
KW - railway safety
KW - railway signal
KW - retrieval-augmented generation
KW - vision–language model
UR - https://www.scopus.com/pages/publications/105045829733
U2 - 10.3390/computers15070416
DO - 10.3390/computers15070416
M3 - 文章
AN - SCOPUS:105045829733
SN - 2073-431X
VL - 15
JO - Computers
JF - Computers
IS - 7
M1 - 416
ER -