Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

PoisonedRAG: Knowledge Corruption Attacks to Retrieval-Augmented Generation of Large Language Models

About

Large language models (LLMs) have achieved remarkable success due to their exceptional generative capabilities. Despite their success, they also have inherent limitations such as a lack of up-to-date knowledge and hallucination. Retrieval-Augmented Generation (RAG) is a state-of-the-art technique to mitigate these limitations. The key idea of RAG is to ground the answer generation of an LLM on external knowledge retrieved from a knowledge database. Existing studies mainly focus on improving the accuracy or efficiency of RAG, leaving its security largely unexplored. We aim to bridge the gap in this work. We find that the knowledge database in a RAG system introduces a new and practical attack surface. Based on this attack surface, we propose PoisonedRAG, the first knowledge corruption attack to RAG, where an attacker could inject a few malicious texts into the knowledge database of a RAG system to induce an LLM to generate an attacker-chosen target answer for an attacker-chosen target question. We formulate knowledge corruption attacks as an optimization problem, whose solution is a set of malicious texts. Depending on the background knowledge (e.g., black-box and white-box settings) of an attacker on a RAG system, we propose two solutions to solve the optimization problem, respectively. Our results show PoisonedRAG could achieve a 90% attack success rate when injecting five malicious texts for each target question into a knowledge database with millions of texts. We also evaluate several defenses and our results show they are insufficient to defend against PoisonedRAG, highlighting the need for new defenses.

Wei Zou, Runpeng Geng, Binghui Wang, Jinyuan Jia• 2024

Related benchmarks

TaskDatasetResultRank
RetrievalEconomic
NDCG@250.242
35
Multi-hop Question AnsweringLoCoMo Multi-Hop (test)
F1 Score25.95
24
Single-hop Question AnsweringLoCoMo Single-Hop (test)
F135.14
24
Temporal Question AnsweringLoCoMo Temporal (test)
F1 Score41.66
24
Open-domain Question AnsweringLoCoMo Open-Domain (test)
F1 Score12.06
24
Targeted AnswerSingle-Agent Evaluation Set
R@5100
12
In-domain corpus poisoning attackMS Marco
ASR61
8
Knowledge Poisoning AttackHotpotQA
ASR64
8
In-domain corpus poisoning attackNQ
ASR58
8
Fact-Level RAG Poisoning AttackBioASQ
RSR@569.6
7
Showing 10 of 19 rows

Other info

Follow for update