Decoupled Alignment for Robust Plug-and-Play Adaptation

Haozheng Luo, Jiahao Yu, Wenxin Zhang, Jialong Li, Jerry Yao-Chieh Hu +2 more
6/3/2024
cs.CLcs.AIcs.CR

No official code found. Best implementation: MAGICS-LAB/DAPA (2 stars, Apache-2.0, last commit 2025-06-11).

Abstract

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.

View on arXivView PDF

Code Implementations(1)

MAGICS-LAB/DAPACommunity60%
Apache-2.0

Decoupled Alignment for Robust Plug-and-Play Adaptation

22Shell, PythonMar 6, 20241 years agoApache-2.0
alignmentdelta-debuggingknowledge-distillationllmsmemory-editing+2 more

Similar Papers

Cite this paper

@article{luo2024decoupled,
  title  = {Decoupled Alignment for Robust Plug-and-Play Adaptation},
  author = {Haozheng Luo and Jiahao Yu and Wenxin Zhang and Jialong Li and Jerry Yao-Chieh Hu and Xinyu Xing and Han Liu},
  year   = {2024},
  eprint = {2406.01514},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2406.01514v3}
}

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.01514.svg)](https://codeofpaper.com/paper/2406.01514)

Discussion