arXiv:2410.20312cs.LGstat.ML2024-10NeurIPS被引 11

用不确定性惩罚机制,防止离线强化学习中错误估值。

Q-Distribution guided Q-learning for offline reinforcement learning: Uncertainty penalized Q-value via consistency model

  • 基于一致性模型学习条件Q值分布,估算动作不确定性。
  • 在分布外动作区域对高不确定性行为进行悲观修正,避免过估计。
  • 适合追求稳定性能的离线强化学习研究者,尤其在复杂任务中表现突出。

「分布偏移」是离线强化学习成功的主要障碍。学习策略可能采取超出行为策略知识范围的动作,称为分布外(OOD)动作,其对应的Q值容易被过高估计,导致学习策略产生偏差。为避免Q值过估计,常见方法是进行悲观调整。本文提出一种基于不确定性惩罚的Q值估计方法——QDQ,通过一致性模型学习高保真、高效的条件Q值分布,从而评估动作的不确定性,并在分布外区域对高不确定性动作进行悲观修正。同时,引入不确定性感知的目标函数,防止估计过度保守。QDQ在理论层面保证了Q值分布学习与不确定性测量的准确性,以及策略性能的稳定性。在D4RL基准测试中表现稳健,多数任务上实现显著提升。

原文摘要 · Abstract (English)

``Distribution shift'' is the main obstacle to the success of offline reinforcement learning. A learning policy may take actions beyond the behavior policy's knowledge, referred to as Out-of-Distribution (OOD) actions. The Q-values for these OOD actions can be easily overestimated. As a result, the learning policy is biased by using incorrect Q-value estimates. One common approach to avoid Q-value overestimation is to make a pessimistic adjustment. Our key idea is to penalize the Q-values of OOD actions associated with high uncertainty. In this work, we propose Q-Distribution Guided Q-Learning (QDQ), which applies a pessimistic adjustment to Q-values in OOD regions based on uncertainty estimation. This uncertainty measure relies on the conditional Q-value distribution, learned through a high-fidelity and efficient consistency model. Additionally, to prevent overly conservative estimates, we introduce an uncertainty-aware optimization objective for updating the Q-value function. The proposed QDQ demonstrates solid theoretical guarantees for the accuracy of Q-value distribution learning and uncertainty measurement, as well as the performance of the learning policy. QDQ consistently shows strong performance on the D4RL benchmark and achieves significant improvements across many tasks.

强化学习离线学习不确定性分布外

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