SPARK: Adaptive Parametric Knowledge Participation for Faithful Retrieval-Augmented Generation
cs.CL, cs.AI
Submitted: 2026-06-26
Updated: 2026-10-13
Comments: 28 pages, 9 Figures
Code: https://github.com/OpenBMB/SHIFT
License: http://creativecommons.org/licenses/by/4.0/
The gist: Retrieval-augmented generation (RAG) enhances the faithfulness of large language models (LLMs) by incorporating external knowledge into the generation process.
Terminology
Abstract
Retrieval-augmented generation (RAG) enhances the faithfulness of large language models (LLMs) by incorporating external knowledge into the generation process. However, conflicts between retrieved context and parametric knowledge can cause LLMs to rely on inconsistent internal knowledge, hindering their ability to faithfully leverage retrieved evidence. Existing approaches mitigate this issue by editing parametric knowledge to resolve such conflicts. However, locating and modifying knowledge encoded in model parameters is challenging and may inadvertently compromise the backbone LLM's capabilities. In this paper, we introduce SPARK, a novel framework that formulates parametric knowledge control as a latent participation policy learning problem, enabling LLMs to adaptively regulate the contribution of parametric knowledge during generation. Specifically, SPARK introduces a lightweight participation policy that dynamically modulates the contribution of FFN-mediated parametric knowledge while keeping the backbone LLM frozen. The participation policy is optimized with a faithfulness-oriented reward, enabling fine-grained control over parametric knowledge participation during generation. Extensive experiments on six datasets validate the effectiveness of SPARK, which introduces and optimizes only 0.01% additional trainable parameters while keeping the backbone LLM fully frozen, thereby preserving its general capabilities and achieving more faithful generation than competitive baselines. All datasets and code are available at https://github.com/OpenBMB/SHIFT.
Sources
- Context-DPO: Aligning Language Models for Context-Faithfulness
- Parameters vs. Context: Fine-Grained Control of Knowledge Reliance in Language Models
- SearchQA: A New Q&A Dataset Augmented with Context from a Search Engine
- Transformer Feed-Forward Layers Are Key-Value Memories
- DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learning
- Measuring Massive Multitask Language Understanding
- ParamMute: Suppressing Knowledge-Critical FFNs for Faithful Retrieval-Augmented Generation
- Taming Knowledge Conflicts in Language Models
- Resisting Contextual Interference in RAG via Parametric-Knowledge Reinforcement
- CtrlA: Adaptive Retrieval-Augmented Generation via Inherent Control
- Entity-Based Knowledge Conflicts in Question Answering
- Locating and Editing Factual Associations in GPT
- Beyond Accuracy: Evaluating Self-Consistency of Code Large Language Models with IdentityChain
- What does the Knowledge Neuron Thesis Have to do with Knowledge?
- Training language models to follow instructions with human feedback
- ToolRL: Reward is All Tool Learning Needs
- In-Context Retrieval-Augmented Language Models
- IRCAN: Mitigating Knowledge Conflicts in LLM Generation via Identifying and Reweighting Context-Aware Neurons
- AdaCAD: Adaptively Decoding to Balance Conflicts between Contextual and Parametric Knowledge
- Self-Consistency Improves Chain of Thought Reasoning in Language Models
Related papers
- Exploring Solution Divergence and Its Effect on Large Language Model Problem Solving
- Ishigaki-IDS-Bench: A Benchmark for Generating Information Delivery Specification from BIM Information Requirements
- Subliminal Steering: Stronger Encoding of Hidden Signals
- MedStruct-S: A Benchmark for Key Discovery, Key-Conditioned QA and Semi-Structured Extraction from OCR Clinical Reports
- The End of Transformers? On Challenging Attention and the Rise of Sub-Quadratic Architectures
- Untangling the Mechanisms of Misleading Context in Medical Question Answering