English

Weights-Rotated Preference Optimization for Large Language Models

Computation and Language 2025-08-26 v1 Artificial Intelligence

Abstract

Despite the efficacy of Direct Preference Optimization (DPO) in aligning Large Language Models (LLMs), reward hacking remains a pivotal challenge. This issue emerges when LLMs excessively reduce the probability of rejected completions to achieve high rewards, without genuinely meeting their intended goals. As a result, this leads to overly lengthy generation lacking diversity, as well as catastrophic forgetting of knowledge. We investigate the underlying reason behind this issue, which is representation redundancy caused by neuron collapse in the parameter space. Hence, we propose a novel Weights-Rotated Preference Optimization (RoPO) algorithm, which implicitly constrains the output layer logits with the KL divergence inherited from DPO and explicitly constrains the intermediate hidden states by fine-tuning on a multi-granularity orthogonal matrix. This design prevents the policy model from deviating too far from the reference model, thereby retaining the knowledge and expressive capabilities acquired during pre-training and SFT stages. Our RoPO achieves up to a 3.27-point improvement on AlpacaEval 2, and surpasses the best baseline by 6.2 to 7.5 points on MT-Bench with merely 0.015% of the trainable parameters, demonstrating its effectiveness in alleviating the reward hacking problem of DPO.

Keywords

Cite

@article{arxiv.2508.17637,
  title  = {Weights-Rotated Preference Optimization for Large Language Models},
  author = {Chenxu Yang and Ruipeng Jia and Mingyu Zheng and Naibin Gu and Zheng Lin and Siyuan Chen and Weichong Yin and Hua Wu and Weiping Wang},
  journal= {arXiv preprint arXiv:2508.17637},
  year   = {2025}
}

Comments

EMNLP 2025

R2 v1 2026-07-01T05:03:56.712Z