Skip to content
aiqbench takeAI Safety

aiqbench analysis

This research introduces TRACE, a lightweight framework designed to detect corpus poisoning attacks in Retrieval-Augmented Generation (RAG) systems by tracing high-influence tokens rather than relying on computationally expensive auxiliary classifiers. By identifying recurrent keywords that steer model outputs toward attacker-specified answers, the method offers an efficient defense mechanism for securing retrieval pipelines against malicious document injection.
Source coveringarXiv
What this isA model's reading of the article and related coverage, written for aiqbench. Not the publisher's words.

Tracing Target Answers in Poisoned Retrieval Corpora via Token Influence Attribution

arXiv1 min read

arXiv:2606.25721v2 Announce Type: replace-cross Abstract: Retrieval-Augmented Generation (RAG) systems are vulnerable to corpus poisoning attacks that manipulate model outputs through malicious retrieved documents. Existing detection methods typically rely on auxiliary classifiers or additional LLM-based verification, introducing substantial computational overhead. We present TRACE, a lightweight detection framework that identifies poisoning attacks by tracing answer-related tokens through token influence attribution. TRACE first discovers recurrent high-influence keywords across retrieved documents and then performs a secondary verification to confirm their influence on model predictions. Experiments on three QA benchmarks and six LLMs demonstrate strong detection performance while simultaneously uncovering attacker-specified target answers.

Abstract:Retrieval-Augmented Generation (RAG) systems are vulnerable to corpus poisoning attacks that manipulate model outputs through malicious retrieved documents. Existing detection methods typically rely on auxiliary classifiers or additional LLM-based verification, introducing substantial computational overhead. We present TRACE, a lightweight detection framework that identifies poisoning attacks by tracing answer-related tokens through token influence attribution. TRACE first discovers recurrent high-influence keywords across retrieved documents and then performs a secondary verification to confirm their influence on model predictions. Experiments on three QA benchmarks and six LLMs demonstrate strong detection performance while simultaneously uncovering attacker-specified target answers.

Cryptography and Security (cs.CR); Computation and Language (cs.CL); Information Retrieval (cs.IR)

arXiv:2606.25721 [cs.CR]

(or arXiv:2606.25721v2 [cs.CR] for this version)

https://doi.org/10.48550/arXiv.2606.25721 Focus to learn more arXiv-issued DOI via DataCite

arXivLabs is a framework that allows collaborators to develop and share new arXiv features directly on our website.

Both individuals and organizations that work with arXivLabs have embraced and accepted our values of openness, community, excellence, and user data privacy. arXiv is committed to these values and only works with partners that adhere to them.

Have an idea for a project that will add value for arXiv's community? Learn more about arXivLabs.

Full text under the publisher's open licence · 193 words

Related coverage