CRAICLApr 6, 2024

Goal-guided Generative Prompt Injection Attack on Large Language Models

arXiv:2404.07234v433 citationsh-index: 9ICDM
Originality Incremental advance
AI Analysis

This addresses security challenges for LLM users by providing a more systematic attack method, though it is incremental as it builds on existing prompt injection research.

The paper tackles the problem of improving prompt injection attacks on large language models by redefining the attack goal to maximize KL divergence, which is shown to be equivalent to maximizing Mahalanobis distance under Gaussian assumptions, and proposes a query-free black-box attack method (G2PIA) that achieves effective results across seven models and four datasets.

Current large language models (LLMs) provide a strong foundation for large-scale user-oriented natural language tasks. A large number of users can easily inject adversarial text or instructions through the user interface, thus causing LLMs model security challenges. Although there is currently a large amount of research on prompt injection attacks, most of these black-box attacks use heuristic strategies. It is unclear how these heuristic strategies relate to the success rate of attacks and thus effectively improve model robustness. To solve this problem, we redefine the goal of the attack: to maximize the KL divergence between the conditional probabilities of the clean text and the adversarial text. Furthermore, we prove that maximizing the KL divergence is equivalent to maximizing the Mahalanobis distance between the embedded representation $x$ and $x'$ of the clean text and the adversarial text when the conditional probability is a Gaussian distribution and gives a quantitative relationship on $x$ and $x'$. Then we designed a simple and effective goal-guided generative prompt injection strategy (G2PIA) to find an injection text that satisfies specific constraints to achieve the optimal attack effect approximately. It is particularly noteworthy that our attack method is a query-free black-box attack method with low computational cost. Experimental results on seven LLM models and four datasets show the effectiveness of our attack method.

Foundations

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

Your Notes