Adaptive Batch-Wise Sample Scheduling for Direct Preference Optimization

Direct Preference Optimization (DPO) has emerged as an effective approach for aligning large language models (LLMs) with human preferences. However, its performance is highly dependent on the quality of the underlying human preference data. To address this bottleneck, prior work has explored various data selection strategies, but these methods often overlook the impact of the evolving states of the language model during the optimization process. In this paper, we introduce a novel problem: Sample Scheduling for DPO, which aims to dynamically and adaptively schedule training samples based on the model's evolving batch-wise states throughout preference optimization. To solve this problem, we propose SamS, an efficient and effective algorithm that adaptively selects samples in each training batch based on the LLM's learning feedback to maximize the potential generalization performance. Notably, without modifying the core DPO algorithm, simply integrating SamS significantly improves performance across tasks, with minimal additional computational overhead. This work points to a promising new direction for improving LLM alignment through batch-wise sample selection, with potential generalization to RLHF and broader supervised learning paradigms.

Authors

• Zixuan Huang School of Computer Science and Engineering, Beihang University, China
• Yikun Ban
• Lean Fu ByteDance Inc.
• Xiaojie Li
• Zhongxiang Dai
• Jianxin Li Beihang University
• Deqing Wang SKLSDE, School of Computer Science, Beihang University

Keywords

No keywords are indexed for this paper.