Decoupled Alignment for Robust Plug-and-Play Adaptation

arXiv:2406.01514 · cs.CL, cs.AI, cs.CR · Submitted 2024-06-03 · Read on arXiv

cs.CL, cs.AI, cs.CR

Submitted: 2024-06-03

Updated: 2026-08-26

Comments: Conference on Language Modeling (COLM) 2026

Code: https://github.com/NWULIST/DAPA

License: http://creativecommons.org/licenses/by/4.0/

The gist: 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.

Terminology

Abstract

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. Our code is available at https://github.com/NWULIST/DAPA.

Sources

Related papers