用大模型指导小模型对齐人类偏好,提升小模型输出质量。
Advantage-Guided Distillation for Preference Alignment in Small Language Models
- 用对齐的大模型做教师,通过双约束知识蒸馏传递偏好知识。
- 引入优势函数提供更精细的奖励信号,显著提升小模型对齐效果。
- 适合想提升小模型对齐能力的研究者或应用开发者。
对齐技术使大型语言模型生成符合人类偏好的输出,对模型有效性至关重要。然而,这些技术在小型语言模型(SLMs)上的效果常因模型容量有限而减弱。我们提出利用一个已对齐的教师大模型来指导小模型的对齐过程,实现人类偏好知识的迁移。首先探索了双重约束知识蒸馏(DCKD),通过两个KL散度约束从对齐教师向未对齐学生传递知识。为进一步增强学生区分偏好与非偏好响应的能力,提出优势引导蒸馏(ADPA),利用教师的收益函数为学生提供更细致的分布级奖励信号。实验表明,两种方法均显著提升小模型的对齐性能,缩小与大模型的差距;其中ADPA表现更优,且与DCKD结合时效果更佳。代码已公开于https://github.com/SLIT-AI/ADPA。
原文摘要 · Abstract (English)
Alignment techniques enable Large Language Models (LLMs) to generate outputs that align with human preferences and play a crucial role in their effectiveness. However, their impact often diminishes when applied to Small Language Models (SLMs), likely due to the limited capacity of these models. Instead of directly applying existing alignment techniques to SLMs, we propose to utilize a well-aligned teacher LLM to guide the alignment process for these models, thereby facilitating the transfer of the teacher's knowledge of human preferences to the student model. To achieve this, we first explore a straightforward approach, Dual-Constrained Knowledge Distillation (DCKD), that employs knowledge distillation with two KL-divergence constraints from the aligned teacher to the unaligned student. To further enhance the student's ability to distinguish between preferred and dispreferred responses, we then propose Advantage-Guided Distillation for Preference Alignment (ADPA), which leverages an advantage function from the aligned teacher to deliver more nuanced, distribution-level reward signals for the student's alignment. Our experimental results show that these two approaches appreciably improve the alignment of SLMs and narrow the performance gap with larger counterparts. Among them, ADPA demonstrates superior performance and achieves even greater effectiveness when integrated with DCKD. Our code is available at https://github.com/SLIT-AI/ADPA.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。