arXiv:2502.18679cs.CL2025-02ICML被引 1

不用人类偏好数据或奖励模型,用判别式方法提升大模型对齐效果。

Discriminative Finetuning of Generative Large Language Models without Reward Models and Human Preference Data

  • 用判别式思路替代生成式训练,同时优化正负样本
  • 在多个基准上超越SFT,接近甚至超过SFT+PO效果
  • 适合想省去人工标注和奖励模型的LLM调优场景

监督微调(SFT)是通过输入输出对数据对预训练大语言模型进行对齐的关键步骤。然而,尽管是监督学习,SFT仍受其生成式目标的固有局限。现有主流策略是在SFT后增加偏好优化(PO)阶段,依赖人工标注的偏好数据或强奖励模型来引导学习。本文提出判别式微调(DFT),借鉴传统监督学习中成功的判别式方法,改进SFT。DFT采用判别式范式,在给定输入下显式建模答案在所有可能输出中的判别似然,提高正确答案概率,同时抑制潜在错误答案,目标是数据预测而非逐标记生成。贡献包括:(i) 构建基于判别似然的微调概率框架;(ii) 设计高效算法优化该似然;(iii) 大量实验验证有效性,DFT性能优于SFT,且可媲美甚至超过SFT→PO。代码见https://github.com/Optimization-AI/DFT。

原文摘要 · Abstract (English)

Supervised fine-tuning (SFT) has become a crucial step for aligning pretrained large language models (LLMs) using supervised datasets of input-output pairs. However, despite being supervised, SFT is inherently limited by its generative training objective. To address its limitations, the existing common strategy is to follow SFT with a separate phase of preference optimization (PO), which relies on either human-labeled preference data or a strong reward model to guide the learning process. In this paper, we address the limitations of SFT by exploring one of the most successful techniques in conventional supervised learning: discriminative learning. We introduce Discriminative Fine-Tuning (DFT), an improved variant of SFT, which mitigates the burden of collecting human-labeled preference data or training strong reward models. Unlike SFT that employs a generative approach and overlooks negative data, DFT adopts a discriminative paradigm that increases the probability of positive answers while suppressing potentially negative ones, aiming for data prediction instead of token prediction. Our contributions include: (i) a discriminative probabilistic framework for fine-tuning LLMs by explicitly modeling the discriminative likelihood of an answer among all possible outputs given an input; (ii) efficient algorithms to optimize this discriminative likelihood; and (iii) extensive experiments demonstrating DFT's effectiveness, achieving performance better than SFT and comparable to if not better than SFT$\rightarrow$PO. The code can be found at https://github.com/Optimization-AI/DFT.

大模型微调判别式学习无需偏好数据

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