arXiv:2504.04524cs.LGcs.AI2025-04被引 12

TRPA算法通过规则构建偏好对,实现稳定高效的LLM推理优化。

Trust Region Preference Approximation: A simple and stable reinforcement learning algorithm for LLM reasoning

  • 用预设规则生成偏好对,结合新优化算法进行训练
  • 在推理任务上表现媲美PPO,且训练更稳定
  • 适合追求稳定性和抗奖励欺骗的LLM推理研究者

大型语言模型(LLMs)正快速向通用人工智能(AGI)演进,通过大规模强化学习提升人类对齐(HA)与推理能力。尽管基于奖励的优化方法(如PPO、GRPO)在推理任务中表现优异,但易受奖励欺骗影响;而基于偏好的方法(如DPO)虽能提升对齐性,但在推理任务上性能仍不及奖励类方法。为此,本文提出信任区域偏好近似(TRPA)算法,融合规则驱动与偏好优化,利用预设规则构建偏好层级和对应偏好对,采用新型优化算法实现强化学习训练,并具备理论保证的单调改进性。实验表明,TRPA在推理任务上达到与PPO相当的性能,同时展现出更强的稳定性。代码已开源。

原文摘要 · Abstract (English)

Recently, Large Language Models (LLMs) have rapidly evolved, approaching Artificial General Intelligence (AGI) while benefiting from large-scale reinforcement learning to enhance Human Alignment (HA) and Reasoning. Recent reward-based optimization algorithms, such as Proximal Policy Optimization (PPO) and Group Relative Policy Optimization (GRPO) have achieved significant performance on reasoning tasks, whereas preference-based optimization algorithms such as Direct Preference Optimization (DPO) significantly improve the performance of LLMs on human alignment. However, despite the strong performance of reward-based optimization methods in alignment tasks , they remain vulnerable to reward hacking. Furthermore, preference-based algorithms (such as Online DPO) haven't yet matched the performance of reward-based optimization algorithms (like PPO) on reasoning tasks, making their exploration in this specific area still a worthwhile pursuit. Motivated by these challenges, we propose the Trust Region Preference Approximation (TRPA) algorithm, which integrates rule-based optimization with preference-based optimization for reasoning tasks. As a preference-based algorithm, TRPA naturally eliminates the reward hacking issue. TRPA constructs preference levels using predefined rules, forms corresponding preference pairs, and leverages a novel optimization algorithm for RL training with a theoretical monotonic improvement guarantee. Experimental results demonstrate that TRPA not only achieves competitive performance on reasoning tasks but also exhibits robust stability. The code of this paper are released and updating on https://github.com/XueruiSu/Trust-Region-Preference-Approximation.git.

强化学习大模型推理偏好优化

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