arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.23116cs.LGcs.GTstat.ML

具有广义双线性偏好的可证明高效正则化在线RLHF

Provably Efficient Regularized Online RLHF with Generalized Bilinear Preferences

KAIST AI(韩国科学技术院人工智能研究所);KAIST EE(韩国科学技术院电子工程系);POSTECH CSE/AI(POSTECH 计算科学与人工智能系)

Junghyun Lee, Minju Hong, Kwang-Sung Jun, Chulhee Yun, Se-Young Yun

AI总结:

研究在线RLHF中正则化最佳响应最大遗憾最小化问题,通过广义双线性偏好模型证明强凸性可导出多对数遗憾,表明快速遗憾不限于KL散度。

摘要:

我们考虑在一般偏好和bandit反馈下在线RLHF中的正则化最佳响应最大遗憾最小化问题。虽然各种正则化器被用于增强对齐的鲁棒性,但已知的多对数遗憾保证仍然高度特定于KL。为了研究这种快速速率是否扩展到KL之外,我们采用广义双线性偏好模型(GBPM)——通过一个秩为$2r$的斜对称矩阵捕获$d$维逐项特征上的非传递偏好——以隔离一般正则化的影响。关键地,在GBPM下,我们证明任何贪婪策略的对偶间隙受限于平方估计误差,该误差仅利用强凸性和斜对称性导出。在特征覆盖假设下,我们通过贪婪采样建立了$\tilde{\mathcal{O}}(\eta d^4 C_{\min}^{-1} (\log T)^2 \wedge d^2 C_{\min}^{-1/2} \sqrt{T})$的通用多对数遗憾,并通过探索后提交(Explore-Then-Commit)建立了$\tilde{\mathcal{O}}(C_{\min}^{-2} \sqrt{\eta r T} \wedge r^{1/3} C_{\min}^{-4/3} T^{2/3})$的维度改进遗憾(对于条件良好的臂集),其中$\eta^{-1}$是正则化系数,$T$是时间范围,$C_{\min}$是依赖于臂集的量。这表明“快速”遗憾并非KL特有,而是通用强凸几何的基本结果。

英文摘要:

We consider the problem of regularized best-response max-regret minimization in online RLHF under general preferences and bandit feedback. While various regularizers are utilized to robustify alignment, known polylogarithmic regret guarantees remain heavily specific to KL. To investigate whether such fast rates extend beyond KL, we adopt the Generalized Bilinear Preference Model (GBPM) -- capturing intransitive preferences over $d$-dimensional item-wise features via a rank-$2r$ skew-symmetric matrix -- to isolate the impact of generic regularization. Crucially, under GBPM, we prove that the dual gap of any greedy policy is bounded by the squared estimation error, derived using \emph{only} strong convexity and skew-symmetry. Under a feature coverage assumption, we establish a \emph{generic} polylogarithmic regret of $\tilde{\mathcal{O}}(ηd^4 C_{\min}^{-1} (\log T)^2 \wedge d^2 C_{\min}^{-1/2} \sqrt{T})$ with Greedy Sampling, and a dimension-wise improved regret (for well-conditioned arm-sets) of $\tilde{\mathcal{O}}(C_{\min}^{-2} \sqrt{ηr T} \wedge r^{1/3} C_{\min}^{-4/3} T^{2/3})$ with Explore-Then-Commit, where $η^{-1}$ is the regularization coefficient, $T$ is the time horizon, and $C_{\min}$ is an arm-set dependent quantity. This demonstrates that ``fast'' regrets are not KL-specific, but rather a fundamental consequence of generic strongly convex geometry.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑