CRCLMay 22, 2024

TrojanRAG: Retrieval-Augmented Generation Can Be Backdoor Driver in Large Language Models

arXiv:2405.13401v472 citationsh-index: 11
Originality Incremental advance
AI Analysis

This addresses security vulnerabilities in LLMs for users and developers, though it is incremental as it builds on existing backdoor attack methods.

The paper tackles the problem of backdoor attacks in large language models (LLMs) by proposing TrojanRAG, a method that manipulates Retrieval-Augmented Generation (RAG) systems to introduce security threats, achieving versatility in attacks while maintaining normal retrieval capabilities.

Large language models (LLMs) have raised concerns about potential security threats despite performing significantly in Natural Language Processing (NLP). Backdoor attacks initially verified that LLM is doing substantial harm at all stages, but the cost and robustness have been criticized. Attacking LLMs is inherently risky in security review, while prohibitively expensive. Besides, the continuous iteration of LLMs will degrade the robustness of backdoors. In this paper, we propose TrojanRAG, which employs a joint backdoor attack in the Retrieval-Augmented Generation, thereby manipulating LLMs in universal attack scenarios. Specifically, the adversary constructs elaborate target contexts and trigger sets. Multiple pairs of backdoor shortcuts are orthogonally optimized by contrastive learning, thus constraining the triggering conditions to a parameter subspace to improve the matching. To improve the recall of the RAG for the target contexts, we introduce a knowledge graph to construct structured data to achieve hard matching at a fine-grained level. Moreover, we normalize the backdoor scenarios in LLMs to analyze the real harm caused by backdoors from both attackers' and users' perspectives and further verify whether the context is a favorable tool for jailbreaking models. Extensive experimental results on truthfulness, language understanding, and harmfulness show that TrojanRAG exhibits versatility threats while maintaining retrieval capabilities on normal queries.

Code Implementations1 repo
Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes