Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

GroupDPO: Memory efficient Group-wise Direct Preference Optimization

About

Preference optimization is widely used to align Large Language Models (LLMs) with preference feedback. However, most existing methods train on a single positive-negative pair per prompt, discarding additional supervision available in preference datasets that typically contain multiple candidate responses. Motivated by this limitation, recent work explores group-wise preference optimization, which jointly contrasts multiple responses for the same prompt, but its empirical behavior and scalability remain underexplored due to the memory overhead of group-coupled objectives. In this work, we introduce a memory-efficient group-wise preference optimization algorithm that preserves gradients while decoupling samples during backpropagation, substantially reducing peak memory usage, which enables scalable training with larger group sizes. Across both offline and online alignment settings, we show that leveraging multiple responses consistently outperforms single-pair training. Furthermore, incorporating a negative log-likelihood (NLL) term on positive responses is critical for both performance gains and training stability.

Jixuan Leng, Si Si, Hsiang-Fu Yu, Vinod Raman, Inderjit S. Dhillon• 2026

Related benchmarks

TaskDatasetResultRank
Instruction FollowingIFEval--
836
Mathematical ReasoningOlympiad
Accuracy0.519
134
Mathematical ReasoningMATH500
Accuracy84.6
86
Instruction FollowingIFBench
Accuracy35.4
72
MathAIME24--
57
ReasoningBig-Bench Hard (BBH)
Accuracy82.3
54
MathematicsMinerva Math
Pass@1 Accuracy44.5
44
MathMATH500
Score90.8
39
ReasoningGPQA Diamond
GPQA Diamond Reasoning Score49.1
36
ReasoningMMLU-Pro
MMLU-Pro Reasoning Score69.4
36
Showing 10 of 19 rows

Other info

Follow for update