提出鲁棒策略迭代方法,提升离线强化学习在分布外状态下的稳定性。
Robust Regularized Policy Iteration under Transition Uncertainty
- 将转移不确定性建模为不确定集,优化最坏情况下的策略
- 在D4RL上平均表现优于主流基线,尤其在分布偏移场景更稳健
- 通过低Q值对齐高认知不确定性,避免执行不可靠动作
离线强化学习可实现数据高效且安全的策略学习,但其性能常因分布偏移而下降。学习到的策略可能访问分布外的状态-动作对,导致价值估计和动态模型不可靠。为统一处理策略诱导的外推与转移不确定性,本文将离线RL建模为鲁棒策略优化问题,将转移核视为不确定集内的决策变量,并针对最坏情况动态优化策略。提出鲁棒正则化策略迭代(RRPI),用可计算的KL正则化代理替代难以求解的极大极小双层目标,基于鲁棒正则化贝尔曼算子推导出高效的策略迭代算法。理论证明该算子是γ-收缩的,且迭代更新代理可单调提升原鲁棒目标并收敛。在D4RL基准上的实验表明,RRPI在多数环境中超越近期基线(包括百分位方法),平均性能优异,同时保持竞争力;此外,其通过将低Q值与高认知不确定性对齐,有效防止策略执行不可靠的分布外动作。
原文摘要 · Abstract (English)
Offline reinforcement learning (RL) enables data-efficient and safe policy learning without online exploration, but its performance often degrades under distribution shift. The learned policy may visit out-of-distribution state-action pairs where value estimates and learned dynamics are unreliable. To address policy-induced extrapolation and transition uncertainty in a unified framework, we formulate offline RL as robust policy optimization, treating the transition kernel as a decision variable within an uncertainty set and optimizing the policy against the worst-case dynamics. We propose Robust Regularized Policy Iteration (RRPI), which replaces the intractable max-min bilevel objective with a tractable KL-regularized surrogate and derives an efficient policy iteration procedure based on a robust regularized Bellman operator. We provide theoretical guarantees by showing that the proposed operator is a $γ$-contraction and that iteratively updating the surrogate yields monotonic improvement of the original robust objective with convergence. Experiments on D4RL benchmarks demonstrate that RRPI achieves strong average performance, outperforming recent baselines including percentile-based methods on the majority of environments while remaining competitive on the rest. Moreover, RRPI exhibits robust performance by aligning lower $Q$-values with high epistemic uncertainty, which prevents the policy from executing unreliable out-of-distribution actions.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。