SOTAVerified

Enhancing Prompt Injection Attacks to LLMs via Poisoning Alignment

2024-10-18Code Available0· sign in to hype

Zedian Shao, Hongbin Liu, Jaden Mu, Neil Zhenqiang Gong

Code Available — Be the first to reproduce this paper.

Reproduce

Code

Abstract

In a prompt injection attack, an attacker injects a prompt into the original one, aiming to make an LLM follow the injected prompt to perform an attacker-chosen task. Existing attacks primarily focus on how to blend the injected prompt into the original prompt without altering the LLM itself. Our experiments show that these attacks achieve some success, but there is still significant room for improvement. In this work, we show that an attacker can boost the success of prompt injection attacks by poisoning the LLM's alignment process. Specifically, we propose PoisonedAlign, a method to strategically create poisoned alignment samples. When even a small fraction of the alignment data is poisoned using our method, the aligned LLM becomes more vulnerable to prompt injection while maintaining its foundational capabilities. The code is available at https://github.com/Sadcardation/PoisonedAlign

Reproductions