Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

InjecGuard: Benchmarking and Mitigating Over-defense in Prompt Injection Guardrail Models

About

Prompt injection attacks pose a critical threat to large language models (LLMs), enabling goal hijacking and data leakage. Prompt guard models, though effective in defense, suffer from over-defense -- falsely flagging benign inputs as malicious due to trigger word bias. To address this issue, we introduce NotInject, an evaluation dataset that systematically measures over-defense across various prompt guard models. NotInject contains 339 benign samples enriched with trigger words common in prompt injection attacks, enabling fine-grained evaluation. Our results show that state-of-the-art models suffer from over-defense issues, with accuracy dropping close to random guessing levels (60%). To mitigate this, we propose InjecGuard, a novel prompt guard model that incorporates a new training strategy, Mitigating Over-defense for Free (MOF), which significantly reduces the bias on trigger words. InjecGuard demonstrates state-of-the-art performance on diverse benchmarks including NotInject, surpassing the existing best model by 30.8%, offering a robust and open-source solution for detecting prompt injection attacks. The code and datasets are released at https://github.com/leolee99/InjecGuard.

Hao Li, Xiaogeng Liu• 2024

Related benchmarks

TaskDatasetResultRank
Prompt injection detectionNotInject
FPR0.29
12
False Positive Rate Detection on Benign PromptsWildGuard benign
FPR13.84
10
LLM Agent Defense EvaluationAgentLure (320 samples)
ASR12.8
10
Prompt injection detectionilion-bench
F1 Score1.1
6
Showing 4 of 4 rows

Other info

Follow for update