Towards Transparent RAG: Fostering Evidence Traceability in LLM Generation via Reinforcement Learning
On the map: Retrieval-augmented generation → Agentic RAG, agents and memory
Abstract (arXiv)
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.
Key facts. Cited by 4 papers in the map (data through 14 Oct 2026). Placed in Agentic RAG, agents and memory, one of 15 directions in Retrieval-augmented generation. Builds on 37 of its 58 references that are in the map. First posted on arXiv 19 May 2025.
Builds on
37 of 58 references.
Closest ancestors · 1
Same direction, newest first
Other references · 36
- 2017Proximal Policy Optimization Algorithmscited by 15,215
- 2024The Llama 3 Herd of Modelscited by 14,715
- 2020Retrieval-Augmented Generation for Knowledge-Intensive NLP Taskscited by 9,297
- 2023Judging LLM-as-a-Judge with MT-Bench and Chatbot Arenacited by 8,626
- 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,944
- 2024Qwen2.5 Technical Reportcited by 4,942
- 2018HotpotQA: A Dataset for Diverse, Explainable Multi-hop Question Answeringcited by 3,754
- 2023Lost in the Middle: How Language Models Use Long Contextscited by 3,455
- 2025DAPO: An Open-Source LLM Reinforcement Learning System at Scalecited by 2,810
- 2020REALM: Retrieval-Augmented Language Model Pre-Trainingcited by 2,141
- 2021WebGPT: Browser-assisted question-answering with human feedbackcited by 1,745
- 2025Search-R1: Training LLMs to Reason and Leverage Search Engines with Reinforcement Learningcited by 1,444
24 more
- 2020Constructing A Multi-hop QA Dataset for Comprehensive Evaluation of Reasoning Stepscited by 1,282
- 2022Measuring and Narrowing the Compositionality Gap in Language Modelscited by 1,075
- 2022Interleaving Retrieval with Chain-of-Thought Reasoning for Knowledge-Intensive Multi-Step Questionscited by 923
- 2021MuSiQue: Multihop Questions via Single-hop Question Compositioncited by 916
- 2021Retrieval Augmentation Reduces Hallucination in Conversationcited by 692
- 2023Enabling Large Language Models to Generate Text with Citationscited by 595
- 2020KILT: a Benchmark for Knowledge Intensive Language Taskscited by 573
- 2023Making Retrieval-Augmented Language Models Robust to Irrelevant Contextcited by 369
- 2023Benchmarking Large Language Models in Retrieval-Augmented Generationcited by 312
- 2023ARES: An Automated Evaluation Framework for Retrieval-Augmented Generation Systemscited by 265
- 2024Agent Hospital: A Simulacrum of Hospital with Evolvable Medical Agentscited by 218
- 2025ZeroSearch: Incentivize the Search Capability of LLMs without Searchingcited by 208
- 2020How Context Affects Language Models' Factual Predictionscited by 181
- 2024Rethinking Interpretability in the Era of Large Language Modelscited by 111
- 2024LongCite: Enabling LLMs to Generate Fine-grained Citations in Long-context QAcited by 81
- 2024CBR-RAG: Case-Based Reasoning for Retrieval Augmented Generation in LLMs for Legal Question Answeringcited by 72
- 2024Improving Retrieval for RAG based Question Answering Models on Financial Documentscited by 49
- 2025In-depth Analysis of Graph-based RAG in a Unified Frameworkcited by 46
- 2024IM-RAG: Multi-Round Retrieval-Augmented Generation Through Learning Inner Monologuescited by 40
- 2024Citation-Enhanced Generation for LLM-based Chatbotscited by 25
- 2025Knowing You Don't Know: Learning When to Continue Search in Multi-round RAG through Self-Practicingcited by 25
- 2024Improving Attributed Text Generation of Large Language Models via Preference Learningcited by 23
- 2025PIKE-RAG: sPecIalized KnowledgE and Rationale Augmented Generationcited by 18
- 2025The Viability of Crowdsourcing for RAG Evaluationcited by 2
Built on by
Citing papers per year: 2025: 2; 2026: 2.
4 papers.
- 2025QAgent: A modular Search Agent with Interactive Query Understandingcited by 8
- 2026GRACE: Reinforcement Learning for Grounded Response and Abstention under Contextual Evidencecited by 7
- 2025From Sufficiency to Reflection: Reinforcement-Guided Thinking Quality in Retrieval-Augmented Reasoning for LLMscited by 5
- 2026PRISMA: Reinforcement Learning Guided Two-Stage Policy Optimization in Multi-Agent Architecture for Open-Domain Multi-Hop Question Answeringcited by 1
- 2026PRISMA: Reinforcement Learning Guided Two-Stage Policy Optimization in Multi-Agent Architecture for Open-Domain Multi-Hop Question Answering9 Jan 2026
- 2026GRACE: Reinforcement Learning for Grounded Response and Abstention under Contextual Evidence8 Jan 2026
- 2025QAgent: A modular Search Agent with Interactive Query Understanding9 Oct 2025
- 2025From Sufficiency to Reflection: Reinforcement-Guided Thinking Quality in Retrieval-Augmented Reasoning for LLMs30 Jul 2025
Similar, not cited
Close in wording, no citation either way
- 2025Learning to Extract Rational Evidence via Reinforcement Learning for Retrieval-Augmented Generation
- 2025FAIR-RAG: Faithful Adaptive Iterative Refinement for Retrieval-Augmented Generation
- 2026CUE-R: Beyond the Final Answer in Retrieval-Augmented Generation
- 2026Distill Where the Student Goes: Teacher-Regularized RL for English-Evidence Cross-Lingual RAG
- 2025UR2: Unify RAG and Reasoning through Reinforcement Learning
- 2026OThink-SRR1: Search, Refine and Reasoning with Reinforced Learning for Large Language Models
- 2026Parameters vs. Context: TRACE Fine-Tuning for Robust Retrieval-Augmented Generation
- 2025Rethinking All Evidence: Enhancing Trustworthy Retrieval-Augmented Generation via Conflict-Driven Summarization