BEEAR: Embedding-based Adversarial Removal of Safety Backdoors in Instruction-tuned Language Models
<|reference_start|>BEEAR: Embedding-based Adversarial Removal of Safety Backdoors in Instruction-tuned Language Models: Safety backdoor attacks in large language models (LLMs) enable the stealthy triggering of unsafe behaviors while evading detection during normal interactions. The high dimensionality of potential triggers in the token space and the diverse range of malicious behaviors make this a critical challenge. We present BEEAR, a mitigation approach leveraging the insight that backdoor triggers induce relatively uniform drifts in the model's embedding space. Our bi-level optimization method identifies universal embedding perturbations that elicit unwanted behaviors and adjusts the model parameters to reinforce safe behaviors against these perturbations. Experiments show BEEAR reduces the success rate of RLHF time backdoor attacks from >95% to <1% and from 47% to 0% for instruction-tuning time backdoors targeting malicious code generation, without compromising model utility. Requiring only defender-defined safe and unwanted behaviors, BEEAR represents a step towards practical defenses against safety backdoors in LLMs, providing a foundation for further advancements in AI safety and security.<|reference_end|>
@article{zeng2024beear:,
    title={BEEAR: Embedding-based Adversarial Removal of Safety Backdoors in
  Instruction-tuned Language Models},
    author={Yi Zeng, Weiyu Sun, Tran Ngoc Huynh, Dawn Song, Bo Li, Ruoxi Jia},
    journal={arXiv preprint arXiv:2406.17092},
    year={2024},
    archivePrefix={arXiv},
    eprint={2406.17092},
    primaryClass={cs.CR cs.AI}
}