BadAgent: Inserting and Activating Backdoor Attacks in LLM Agents

Yifei Wang, Dizhan Xue, Shengjie Zhang, Shengsheng Qian
6/5/2024
cs.CLcs.AIcs.CR

Official code: DPamK/BadAgent (36 stars, last commit 2025-02-27).

Abstract

With the prosperity of large language models (LLMs), powerful LLM-based intelligent agents have been developed to provide customized services with a set of user-defined tools. State-of-the-art methods for constructing LLM agents adopt trained LLMs and further fine-tune them on data for the agent task. However, we show that such methods are vulnerable to our proposed backdoor attacks named BadAgent on various agent tasks, where a backdoor can be embedded by fine-tuning on the backdoor data. At test time, the attacker can manipulate the deployed LLM agents to execute harmful operations by showing the trigger in the agent input or environment. To our surprise, our proposed attack methods are extremely robust even after fine-tuning on trustworthy data. Though backdoor attacks have been studied extensively in natural language processing, to the best of our knowledge, we could be the first to study them on LLM agents that are more dangerous due to the permission to use external tools. Our work demonstrates the clear risk of constructing LLM agents based on untrusted LLMs or data. Our code is public at https://github.com/DPamK/BadAgent

View on arXivView PDF

Code Implementations(1)

DPamK/BadAgentOfficial100%
365PythonJan 6, 20241 years ago

Similar Papers

Cite this paper

@article{wang2024badagent,
  title  = {BadAgent: Inserting and Activating Backdoor Attacks in LLM Agents},
  author = {Yifei Wang and Dizhan Xue and Shengjie Zhang and Shengsheng Qian},
  year   = {2024},
  eprint = {2406.03007},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2406.03007v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2406.03007.svg)](https://codeofpaper.com/paper/2406.03007)

Discussion