用分布差异筛选数据,让大模型更高效对齐人类偏好
DEFT: Distribution-guided Efficient Fine-Tuning for Human Alignment
- 基于输出分布差异计算奖励,筛选高质量微调数据
- 仅用少量数据即可提升对齐效果,训练时间大幅减少
- 适合追求高效对齐且不愿牺牲模型泛化能力的研究者
基于人类反馈的强化学习(RLHF)使用PPO等算法对齐大语言模型(LLM)与人类价值观,但成本高且不稳定。现有替代方法虽结合监督微调(SFT)和对比学习实现直接微调,但仍需大量数据学习偏好,可能削弱模型泛化能力。本文提出分布引导的高效微调框架DEFT,通过计算模型输出分布与偏好数据差异分布的差分分布奖励,从原始数据中筛选出小而高质量子集,并将其融入现有对齐方法以引导模型输出分布。实验表明,经DEFT增强的方法在对齐能力和泛化性能上均优于原方法,且训练时间显著降低。
原文摘要 · Abstract (English)
Reinforcement Learning from Human Feedback (RLHF), using algorithms like Proximal Policy Optimization (PPO), aligns Large Language Models (LLMs) with human values but is costly and unstable. Alternatives have been proposed to replace PPO or integrate Supervised Fine-Tuning (SFT) and contrastive learning for direct fine-tuning and value alignment. However, these methods still require voluminous data to learn preferences and may weaken the generalization ability of LLMs. To further enhance alignment efficiency and performance while mitigating the loss of generalization ability, this paper introduces Distribution-guided Efficient Fine-Tuning (DEFT), an efficient alignment framework incorporating data filtering and distributional guidance by calculating the differential distribution reward based on the output distribution of language model and the discrepancy distribution of preference data. A small yet high-quality subset is filtered from the raw data using a differential distribution reward, which is then incorporated into existing alignment methods to guide the model's output distribution. Experimental results demonstrate that the methods enhanced by DEFT outperform the original methods in both alignment capability and generalization ability, with significantly reduced training time.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。