CLDec 17, 2024

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

arXiv:2412.12865v17 citationsh-index: 4AAAI
Originality Incremental advance
AI Analysis

This addresses the challenge of data quality in SFT for LLM alignment, offering an incremental improvement that can be integrated with existing methods.

The paper tackles the problem of low-quality supervised fine-tuning (SFT) datasets for aligning large language models by introducing PoFT, a preference-oriented method that favors the target model over aligned LLMs, resulting in stable and consistent improvements over SFT baselines across different datasets and models.

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.

Code Implementations1 repo
Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes