arXiv:2601.01803cs.LGcs.AI2026-01

用收益分布的偏度与峰度稳定强化学习策略优化。

Moments Matter:Stabilizing Policy Optimization using Return Distributions

  • 通过分布评论家捕捉状态-动作收益分布,利用高阶矩修正PPO优势函数。
  • 在Walker2D任务中使策略更新后收益分布宽度缩小75%,显著提升稳定性。
  • 适合关注连续控制任务中策略鲁棒性与算法可复现性的研究者。

深度强化学习代理虽然获得相似的回合回报,但行为差异巨大,原因在于环境随机性(如转移、初始条件、奖励噪声)和算法因素(小批量选择、探索噪声)的共同影响。在连续控制任务中,微小参数变化可能导致不稳定的步态,阻碍算法比较与现实迁移。已有研究表明,策略更新经过噪声邻域时会导致不稳定性,而更新后收益分布 $R(θ)$ 的扩散程度是该噪声的有效指标。尽管直接约束 $R(θ)$ 可提升稳定性,但在高维场景下估计 $R(θ)$ 计算成本过高。本文提出一种新方法:利用环境随机性,通过分布评论家建模状态-动作收益分布,并以该分布的偏度与峰度修正PPO的优势函数。通过惩罚极端尾部行为,抑制策略进入易引发不稳定的参数区域。我们假设当更新后评论家值与实际回报对齐不佳时,标准PPO难以维持窄 $R(θ)$;在此情况下,基于矩的修正可显著缩小 $R(θ)$,在Walker2D上提升稳定性达75%,同时保持相近的评估回报。

原文摘要 · Abstract (English)

Deep Reinforcement Learning (RL) agents often learn policies that achieve the same episodic return yet behave very differently, due to a combination of environmental (random transitions, initial conditions, reward noise) and algorithmic (minibatch selection, exploration noise) factors. In continuous control tasks, even small parameter shifts can produce unstable gaits, complicating both algorithm comparison and real-world transfer. Previous work has shown that such instability arises when policy updates traverse noisy neighborhoods and that the spread of post-update return distribution $R(θ)$, obtained by repeatedly sampling minibatches, updating $θ$, and measuring final returns, is a useful indicator of this noise. Although explicitly constraining the policy to maintain a narrow $R(θ)$ can improve stability, directly estimating $R(θ)$ is computationally expensive in high-dimensional settings. We propose an alternative that takes advantage of environmental stochasticity to mitigate update-induced variability. Specifically, we model state-action return distribution through a distributional critic and then bias the advantage function of PPO using higher-order moments (skewness and kurtosis) of this distribution. By penalizing extreme tail behaviors, our method discourages policies from entering parameter regimes prone to instability. We hypothesize that in environments where post-update critic values align poorly with post-update returns, standard PPO struggles to produce a narrow $R(θ)$. In such cases, our moment-based correction narrows $R(θ)$, improving stability by up to 75% in Walker2D, while preserving comparable evaluation returns.

强化学习策略优化分布强化学习稳定性

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