动态调整奖励权重,让大模型在线对齐更优均衡。
Learning to Optimize Multi-Objective Alignment Through Dynamic Reward Weighting
- 训练中自适应调整多目标奖励权重,突破固定权重局限。
- 在多个数据集上用更少步数达成帕累托最优解。
- 适合需要平衡多种目标的大模型在线对齐任务。
多目标强化学习通常采用固定权重的线性奖励加权,无法捕捉非凸帕累托前沿,导致次优结果,尤其在大语言模型的在线偏好对齐中问题突出。由于参数化策略生成的随机轨迹在参数与目标间形成高度非线性、非凸映射,单一静态权重难以找到最优权衡。为此,本文提出动态奖励加权,在在线强化学习过程中自适应调整权重。相比依赖固定权重插值的现有方法,本方法持续平衡并优先处理各目标,促进目标空间中帕累托前沿的有效探索。提出两种渐进式增强的方法:基于超体积的权重自适应和基于梯度的权重优化,构成通用工具包,适用于在线多目标对齐。大量实验表明,该方法兼容常见在线强化学习算法,跨多个数据集有效,适用于不同模型家族,始终以更少训练步数获得帕累托主导解,优于固定权重线性标量化基线。
原文摘要 · Abstract (English)
Prior work in multi-objective reinforcement learning typically uses linear reward scalarization with fixed weights, which provably fails to capture non-convex Pareto fronts and thus yields suboptimal results. This limitation becomes especially critical in online preference alignment for large language models. Here, stochastic trajectories generated by parameterized policies create highly non-linear and non-convex mappings from parameters to objectives that no single static weighting scheme can find optimal trade-offs. We address this limitation by introducing dynamic reward weighting, which adaptively adjusts reward weights during the online reinforcement learning process. Unlike existing approaches that rely on fixed-weight interpolation, our dynamic weighting continuously balances and prioritizes objectives in training, facilitating effective exploration of Pareto fronts in objective space. We introduce two approaches of increasing sophistication and generalizability: hypervolume-guided weight adaptation and gradient-based weight optimization, offering a versatile toolkit for online multi-objective alignment. Our extensive experiments demonstrate their compatibility with commonly used online reinforcement learning algorithms, effectiveness across multiple datasets, and applicability to different model families, consistently achieving Pareto dominant solutions with fewer training steps than fixed-weight linear scalarization baselines.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。