arXiv:2603.08118cs.LG2026-03中稿 · ICLR被引 5

提出ROMI方法,解决离线强化学习中模型误差导致的策略退化问题。

Model-based Offline RL via Robust Value-Aware Model Learning with Implicitly Differentiable Adaptive Weighting

  • 用值感知的鲁棒模型学习替代梯度更新,控制保守性并稳定训练
  • 在D4RL和NeoRL上显著优于RAMBO,尤其在复杂数据集表现突出
  • 适合需要高稳定性与泛化能力的离线强化学习任务

基于模型的离线强化学习旨在通过动力学模型提升策略探索能力,但模型误差易引发模型滥用,降低算法性能。对抗式模型学习提供理论框架以缓解该问题,其中RAMBO是代表性实用方法,利用模型梯度进行更新。然而我们发现,即使微调超参数,RAMBO仍可能出现严重的Q值低估与梯度爆炸,表明其过于保守且模型更新不稳定。为此,我们提出鲁棒值感知模型学习与隐式可微自适应加权方法(ROMI)。ROMI不采用模型梯度更新,而是要求动力学模型在可调节状态不确定性集内预测未来状态时,其值接近最小Q值,实现可控保守性与稳定更新。为进一步提升多步回放中的分布外泛化能力,引入隐式可微自适应加权机制,实现动态与值双重感知的学习。在D4RL和NeoRL数据集上的实验表明,ROMI显著优于RAMBO,且在多数场景下达到或超过现有先进方法性能。代码已开源:https://github.com/zq2r/ROMI.git。

原文摘要 · Abstract (English)

Model-based offline reinforcement learning (RL) aims to enhance offline RL with a dynamics model that facilitates policy exploration. However, \textit{model exploitation} could occur due to inevitable model errors, degrading algorithm performance. Adversarial model learning offers a theoretical framework to mitigate model exploitation by solving a maximin formulation. Within such a paradigm, RAMBO~\citep{rigter2022rambo} has emerged as a representative and most popular method that provides a practical implementation with model gradient. However, we empirically reveal that severe Q-value underestimation and gradient explosion can occur in RAMBO with only slight hyperparameter tuning, suggesting that it tends to be overly conservative and suffers from unstable model updates. To address these issues, we propose \textbf{RO}bust value-aware \textbf{M}odel learning with \textbf{I}mplicitly differentiable adaptive weighting (ROMI). Instead of updating the dynamics model with model gradient, ROMI introduces a novel robust value-aware model learning approach. This approach requires the dynamics model to predict future states with values close to the minimum Q-value within a scale-adjustable state uncertainty set, enabling controllable conservatism and stable model updates. To further improve out-of-distribution (OOD) generalization during multi-step rollouts, we propose implicitly differentiable adaptive weighting, a bi-level optimization scheme that adaptively achieves dynamics- and value-aware model learning. Empirical results on D4RL and NeoRL datasets show that ROMI significantly outperforms RAMBO and achieves competitive or superior performance compared to other state-of-the-art methods on datasets where RAMBO typically underperforms. Code is available at https://github.com/zq2r/ROMI.git.

离线RL模型学习强化学习稳健优化

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