KFS-RAG: Mitigating Database Leakage in RAG Systems with Keyword-Grounded Fact Substitution

ZiLiang Zhang, Yubo Zhu, Wei Tong, Jingyu Hua, Zijian Wang, Yuan Zhang, Sheng Zhong


Abstract
Retrieval-Augmented Generation (RAG) has emerged as a powerful paradigm for combining large language models (LLMs) with external knowledge sources. However, RAG systems remain vulnerable to prompt injection attacks, which may mislead the retriever or generator to expose sensitive database contents. To address this issue, we propose KFS-RAG, a defense that mitigates information leakage by reformulating the retrieved context. Specifically, our method first identifies a small set of influential keywords from the retrieved context via an attention rollout plus a causal perturbation mechanism. These keywords are then used to guide an auxiliary LLM to generate a compact set of keyword-grounded facts from the retrieved passages. Finally, the original context is substituted with these curated facts, ensuring that the generator operates on sanitized evidence rather than the raw retrieved text. Experimental evaluations demonstrate that KFS-RAG significantly reduces the risk of database leakage under injection attacks while maintaining response accuracy and relevance. This work highlights a practical pathway toward building secure and trustworthy RAG systems.
Anthology ID:
2026.findings-emnlp.74
Volume:
Findings of the Association for Computational Linguistics: EMNLP 2026
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
Findings
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
1516–1532
Language:
URL:
https://aclanthology.org/2026.findings-emnlp.74/
DOI:
Bibkey:
Cite (ACL):
ZiLiang Zhang, Yubo Zhu, Wei Tong, Jingyu Hua, Zijian Wang, Yuan Zhang, and Sheng Zhong. 2026. KFS-RAG: Mitigating Database Leakage in RAG Systems with Keyword-Grounded Fact Substitution. In Findings of the Association for Computational Linguistics: EMNLP 2026, pages 1516–1532, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
KFS-RAG: Mitigating Database Leakage in RAG Systems with Keyword-Grounded Fact Substitution (Zhang et al., Findings 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.findings-emnlp.74.pdf