arXiv:2409.14836cs.CLcs.AI2024-09被引 6

通过旋转权重防止偏好模型过拟合,提升生成多样性。

Orthogonal Finetuning for Direct Preference Optimization

  • 用权重旋转更新替代传统优化,保持神经元间角度不变。
  • 仅用0.0086%参数就实现比DPO高10分的评测表现。
  • 适合追求高效、多样且对齐精准的模型微调场景。

DPO是一种有效的偏好优化算法,但其微调模型容易在不受欢迎样本上过拟合,表现为生成内容过长且缺乏多样性。现有正则化方法虽试图缓解此问题,却以牺牲对齐性能为代价。本文从权重更新角度提出创新正则化思路:通过权重旋转偏好优化(RoPO)方法,仅对权重进行旋转和幅值拉伸,保持超球能量不变,从而保留神经元间角度所编码的知识。实验表明,该方法在仅使用0.0086%可训练参数的情况下,模型在人类偏好对齐上表现完美,且生成多样性平均提升6点;在MT-Bench上性能最高提升10分,在AlpacaEval 2上最高提升2.8分。

原文摘要 · Abstract (English)

DPO is an effective preference optimization algorithm. However, the DPO-tuned models tend to overfit on the dispreferred samples, manifested as overly long generations lacking diversity. While recent regularization approaches have endeavored to alleviate this issue by modifying the objective function, they achieved that at the cost of alignment performance degradation. In this paper, we innovatively incorporate regularization from the perspective of weight updating to curb alignment overfitting. Through the pilot experiment, we discovered that there exists a positive correlation between overfitting and the hyperspherical energy fluctuation. Hence, we introduce orthogonal finetuning for DPO via a weight-Rotated Preference Optimization (RoPO) method, which merely conducts rotational and magnitude-stretching updates on the weight parameters to maintain the hyperspherical energy invariant, thereby preserving the knowledge encoded in the angle between neurons. Extensive experiments demonstrate that our model aligns perfectly with human preferences while retaining the original expressive capacity using only 0.0086% of the trainable parameters, suggesting an effective regularization against overfitting. Specifically, RoPO outperforms DPO by up to 10 points on MT-Bench and by up to 2.8 points on AlpacaEval 2, while enhancing the generation diversity by an average of 6 points.

偏好优化权重更新生成多样性微调

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