Decoupled Alignment For Robust Plug-and-play Adaptation

Luo Haozheng, Yu Jiahao, Zhang Wenxin, Li Jialong, Hu Jerry Yao-chieh, Xing Xinyu, Liu Han. Arxiv 2024

[Paper]    
Agentic Distillation Efficiency And Optimization Fine Tuning Pretraining Methods RAG Reinforcement Learning Responsible AI Training Techniques

We introduce a low-resource safety enhancement method for aligning large language models (LLMs) without the need for supervised fine-tuning (SFT) or reinforcement learning from human feedback (RLHF). Our main idea is to exploit knowledge distillation to extract the alignment information from existing well-aligned LLMs and integrate it into unaligned LLMs in a plug-and-play fashion. 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.41%, reaching as high as 51.39%, in 17 unaligned pre-trained LLMs, without compromising performance.

Similar Work