Decoupled Alignment for Robust Plug-and-Play Adaptation
Haozheng Luo · Jiahao Yu · Wenxin Zhang · Jialong Li · Chenghao Qiu · Yimin Wang · Eric Hanchen Jiang · Jerry Yao-Chieh Hu · Yan Chen · Binghui Wang · Xinyu Xing · Han Liu
We introduce a training-free safety enhancement method for aligning large language models (LLMs) without the need of supervised fine-tuning or reinforcement learning from human feedback.
摘要
We introduce a training-free safety enhancement method for aligning large language models (LLMs) without the need for supervised fine-tuning or reinforcement learning from human feedback. Our main idea is to provide a robust plug-and-play approach to prevent shadow alignment when models are adapted to downstream tasks. Specifically, we leverage knowledge distillation to extract alignment signals from well-aligned LLMs and inject them into shadow-aligned models via model fusion, enabling plug-and-play alignment correction. In our methodology, we employ delta debugging to identify the critical components of knowledge necessary for effective distillation. On the harmful question dataset, our method significantly enhances the average defense success rate by approximately 14.42%, reaching as high as 51.39% across 17 influenced LLMs, without compromising performance.