arXiv:2501.07837cs.AI2025-01被引 1

用大模型提升高铁司机故障处理能力,更准更透明。

A Driver Advisory System Based on Large Language Model for High-speed Train

  • 用铁路知识微调大模型,提升问答准确率
  • 引入检索增强生成,问答召回率提升4%
  • 模拟真实场景验证,适合铁路运维人员参考

随着中国高铁快速发展,司机在运行中面临日益严峻的技术挑战,如牵引力丢失或传感器故障。目前司机依赖随车机械师处理技术问题,等待过程可能影响运营效率甚至导致事故。为提升故障处理的准确性与可解释性,本文提出基于大语言模型(LLM)的智能司机辅助系统(IDAS-LLM)。首先,使用自建铁路知识问答数据集对LLM进行领域微调,以提高其在铁路相关问题上的回答准确率;随后,集成检索增强生成(RAG)架构,增强生成结果的可解释性。在自建的铁路驾驶知识评估数据集上开展对比实验,结果表明:领域微调后的LLM平均回答准确率提升10%,优于部分主流LLM;引入RAG框架后,问答会话的平均召回率提升约4%。最后,通过真实运行场景的仿真测试,验证了IDAS-LLM在故障处理方面的实际应用潜力。

原文摘要 · Abstract (English)

With the rapid development of China high-speed railway, drivers face increasingly significant technical challenges during operations, such as fault handling. Currently, drivers depend on the onboard mechanic when facing technical issues, for instance, traction loss or sensor faults. This dependency can hinder effective operation, even lead to accidents, while waiting for faults to be addressed. To enhance the accuracy and explainability of actions during fault handling, an Intelligent Driver Advisory System (IDAS) framework based on a large language model (LLM) named IDAS-LLM, is introduced. Initially, domain-fine-tuning of the LLM is performed using a constructed railway knowledge question-and-answer dataset to improve answer accuracy in railway-related questions. Subsequently, integration of the Retrieval-augmented Generation (RAG) architecture is pursued for system design to enhance the explainability of generated responses. Comparative experiments are conducted using the constructed railway driving knowledge assessment dataset. Results indicate that domain-fine-tuned LLMs show an improvement in answer accuracy by an average of 10%, outperforming some current mainstream LLMs. Additionally, the inclusion of the RAG framework increases the average recall rate of question-and-answer sessions by about 4%. Finally, the fault handling capability of IDAS-LLM is demonstrated through simulations of real operational scenarios, proving that the proposed framework has practical application prospects.

大模型高铁智能辅助RAG

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。