arXiv:2412.12865cs.CL2024-12AAAI被引 8

用偏好学习提升微调效果,让模型更优地利用低质量数据

Preference-Oriented Supervised Fine-Tuning: Favoring Target Model Over Aligned Large Language Models

  • 以目标模型优于对齐大模型为偏好指导微调
  • 在多种数据集和基模型上稳定超越传统微调方法
  • 适合数据质量差但需高效微调的场景

对齐使预训练大语言模型具备遵循指令的能力,是其实现实际应用的关键。传统监督微调(SFT)通常采用交叉熵损失进行因果语言建模,依赖大量高质量指令-响应对。然而,由于数据构建与维护成本高、人力密集,现有SFT数据集的质量难以保证。为此,本文提出一种新型偏好导向的监督微调方法——PoFT。其核心思想是在相同SFT数据上,强制目标模型预测概率高于已对齐的大模型,从而将对齐模型对数据质量的评估(即预测概率)融入训练过程。大量实验验证了该方法的有效性:PoFT在不同训练数据集和基模型上均实现稳定且一致的性能提升。此外,可与现有SFT数据过滤方法结合使用,并进一步通过偏好优化(如DPO)持续改进。

原文摘要 · Abstract (English)

Alignment, endowing a pre-trained Large language model (LLM) with the ability to follow instructions, is crucial for its real-world applications. Conventional supervised fine-tuning (SFT) methods formalize it as causal language modeling typically with a cross-entropy objective, requiring a large amount of high-quality instruction-response pairs. However, the quality of widely used SFT datasets can not be guaranteed due to the high cost and intensive labor for the creation and maintenance in practice. To overcome the limitations associated with the quality of SFT datasets, we introduce a novel \textbf{p}reference-\textbf{o}riented supervised \textbf{f}ine-\textbf{t}uning approach, namely PoFT. The intuition is to boost SFT by imposing a particular preference: \textit{favoring the target model over aligned LLMs on the same SFT data.} This preference encourages the target model to predict a higher likelihood than that predicted by the aligned LLMs, incorporating assessment information on data quality (i.e., predicted likelihood by the aligned LLMs) into the training process. Extensive experiments are conducted, and the results validate the effectiveness of the proposed method. PoFT achieves stable and consistent improvements over the SFT baselines across different training datasets and base models. Moreover, we prove that PoFT can be integrated with existing SFT data filtering methods to achieve better performance, and further improved by following preference optimization procedures, such as DPO.

微调偏好学习大模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。