arXiv:2505.13258cs.CL2025-05被引 4

用强化学习让大模型生成带证据来源的透明回答

Towards Transparent RAG: Fostering Evidence Traceability in LLM Generation via Reinforcement Learning

  • 通过强化学习引导模型输出带证据标注的结构化结果
  • 在多跳问答数据集上准确率提升10%-30%,接近商用模型水平
  • 适合需要可解释性推理的医疗、法律等高风险领域

检索增强生成(RAG)在知识密集型应用中价值显著,但生成结果常缺乏可追溯到原始文档的透明推理路径。为解决此问题,我们提出TRACE框架,通过强化学习提升大语言模型(LLM)的证据可追溯性。该框架通过提示和奖励机制,同时优化证据相关性、格式规范性和生成准确性,实现结构化证据引用。为应对多目标奖励带来的训练不稳定性,引入自适应奖励融合策略与稳定化的KL散度估计器。在Qwen2.5-7B-Instruct和Llama-3.1-8B-Instruct上对三个多跳问答数据集的实验表明,TRACE不仅生成具有透明证据链的结果,且准确率提升10%-30%,性能媲美OpenAI o1、DeepSeek-R1等先进商用模型。进一步分析显示其在未见任务上具备良好泛化能力。代码已公开。

原文摘要 · Abstract (English)

Retrieval-Augmented Generation (RAG) delivers substantial value in knowledge-intensive applications. However, its generated responses often lack transparent reasoning paths that trace back to source evidence from retrieved documents. This opacity not only compromises the interpretability of the output but also limits the model's ability to fully exploit the provided context. To address this, we propose TRACE (Transparent RAG with evidenCE tracing), a framework designed to enhance evidence traceability in Large Language Models (LLMs) through reinforcement learning (RL). TRACE guides LLMs to produce structured outputs with explicit evidence citations by prompting and rewarding evidence relevance and proper formatting, alongside accuracy, to optimize structured traceability. To ensure training stability with multiple reward signals, we further introduce an adaptive strategy for merging rewards and adopt a stabilized KL-divergence estimator. Experiments on three multi-hop QA datasets using Qwen2.5-7B-Instruct and Llama-3.1-8B-Instruct show that TRACE achieves both transparent, evidence-attributed outputs and accuracy improvements of 10-30%. The resulting performance is comparable to advanced commercial LLMs (e.g., OpenAI o1, DeepSeek-R1). Further analyses demonstrate strong generalization capabilities to unseen tasks. Our code is publicly available now.

RAG强化学习可解释性多跳问答

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