将语言模型对齐转化为希尔伯特空间的正交投影,避免梯度消失。
Orthogonalized Policy Optimization:Policy Optimization as Orthogonal Projection in Hilbert Space
- 在L2(pi_k)空间中用正交约束替代概率归一化,实现闭式更新。
- 实验显示其在高置信度区仍保持非零梯度,提升长序列训练效果。
- 适合追求稳定优化与鲁棒泛化的强化学习研究者。
我们提出正交化策略优化(OPO),一种基于希尔伯特函数空间L2(pi_k)优化的大语言模型对齐框架。将策略更新从概率单纯形提升到L2(pi_k)空间,使非线性归一化约束转化为密度涨落场v = pi/pi_k - 1上的线性正交条件⟨v, 1⟩_{pi_k} = 0。根据希尔伯特投影定理,唯一闭式更新为v_star = (omega_alpha - E[omega_alpha]) / mu,其中减去均值作为化学势保障概率守恒。该解释揭示优势z-score归一化实为守恒律投影而非方差缩减启发式。OPO清晰分离采样几何(由援引指数alpha控制)与优化几何(由刚度参数mu决定),此分离在KL基目标下无法实现。同一更新亦可视为欧氏镜像下降步与近平衡统计力学的线性响应律,确立其在比值几何中的结构唯一性。结构上,OPO诱导恒定曲率、非饱和线性梯度动力学及内在卡方信任域。在MATH基准测试中,希尔伯特投影形式有效防止了传统KL约束方法的梯度饱和现象。通过在高置信度区域维持非零梯度,OPO避免了过早平台期,相比截断基线实现了更强的长时序训练奖励与更好的分布外泛化能力。
原文摘要 · Abstract (English)
We propose Orthogonalized Policy Optimization (OPO), a principled framework for large language model alignment derived from optimization in the Hilbert function space L2(pi_k). Lifting policy updates from the probability simplex into L2(pi_k) transforms the nonlinear normalization constraint into a linear orthogonality condition <v, 1>_{pi_k} = 0 on the density fluctuation field v = pi/pi_k - 1. By the Hilbert projection theorem, the unique closed-form update is v_star = (omega_alpha - E[omega_alpha]) / mu, where the subtracted mean acts as a chemical potential enforcing probability conservation. This interpretation reveals advantage z-score normalization as a conservation-law projection rather than a variance-reduction heuristic. OPO cleanly decouples sampling geometry, controlled by the escort exponent alpha, from optimization geometry, governed by the stiffness parameter mu, a separation not attainable under KL-based objectives. The same update can also be derived as a Euclidean mirror-descent step and as the linear-response law of near-equilibrium statistical mechanics, establishing its structural uniqueness within ratio geometry. Structurally, OPO induces constant curvature, non-saturating linear gradient dynamics, and an intrinsic chi-square trust region. Experiments on MATH benchmarks show that the Hilbert projection formulation prevents gradient saturation typical of KL-constrained methods. By sustaining non-vanishing gradients in high-confidence regimes, OPO avoids premature plateaus and achieves stronger long-horizon training rewards and improved out-of-distribution generalization compared to clipping-based baselines.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。