arXiv:2508.18884cs.LGcs.AI2025-08

新方法HAEPO让智能体在长任务中更全面探索,同时保持学习稳定。

HAEPO: History-Aggregated Exploratory Policy Optimization

  • 用轨迹累计对数似然构建历史感知探索损失
  • 收敛快、探索广,对齐真实奖励优于或持平现有方法
  • 适合需要长期规划与稳定探索的任务

探索对现代学习至关重要,从小型神经策略的强化学习到大语言模型(LLMs)。现有方法如DPO利用完整序列的似然来捕捉模型决策轨迹,而GRPO将逐标记比率聚合为轨迹级更新。然而两者在长时序任务中常限制探索。本文提出历史聚合探索策略优化(HAEPO),通过将每条轨迹压缩为对数概率之和(累积对数似然),并使用Plackett-Luce Softmax对轨迹加权以匹配回报,促进更广泛探索。引入熵正则化稳定剧烈更新,以及相对于冻结前一策略的软KL惩罚。实验表明,HAEPO收敛迅速、探索充分,与真实奖励高度对齐,在多样任务上表现优于或持平于PPO、GRPO和DPO,提供一个稳定且可解释的框架,显式利用完整轨迹历史并平衡探索与稳定性。

原文摘要 · Abstract (English)

Exploration is essential in modern learning, from reinforcement learning environments with small neural policies to large language models (LLMs). Existing work, such as DPO, leverages full sequence log-likelihoods to capture an entire trajectory of the model's decisions, while methods like GRPO aggregate per-token ratios into a trajectory-level update. However, both often limit exploration on long-horizon tasks. We introduce History-Aggregated Exploratory Policy Optimization (HAEPO), a history-aware exploratory loss to combat these shortcomings. HAEPO compresses each trajectory into the sum of its logarithmic probabilities (a cumulative logarithmic likelihood), and applies a Plackett-Luce softmax across trajectories to obtain normalized weights proportional to their returns, thus encouraging broader exploration. We add entropy regularization to stabilize the aggressive updates to prevent premature collapse and a soft KL penalty relative to a frozen copy of the previous (reference) policy. Empirically, HAEPO converges fast, explores thoroughly, aligns closely with true rewards, and demonstrates robust learning behavior better or at par with PPO, GRPO, and DPO across diverse tasks. Thus, HAEPO provides a stable and interpretable framework by explicitly leveraging full-trajectory history while balancing exploration and stability.

强化学习探索策略轨迹优化

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