用扩散模型修正离线强化学习中的价值估计偏差,提升策略表现。
Diffusion Policy with Behavioral Advantage Correction for Offline Reinforcement Learning

- 引入行为优势修正机制,校正学习策略的Q值估计
- 在D4RL多个任务上超越现有最优算法,性能显著提升
- 适合需要高精度策略优化的离线强化学习场景
在离线强化学习中,行为数据分布与学习策略之间的分布偏移会导致错误的Q值估计,进而误导策略优化方向。为此,本文提出行为优势修正策略评估(BAC-PE)方法,利用行为策略的Q函数对学习策略的Q函数进行修正,从而缓解悲观保守性和高估偏差。理论上分析了BAC-PE的收敛性,并推导出学习Q函数与真实Q函数之间差异的上界。为缓解分布偏移,本文采用扩散模型表示行为策略和学习策略,通过分布匹配实现精准策略正则化。同时,在训练中引入Q值引导,实现有效策略改进。结合BAC-PE与扩散策略建模,提出扩散策略行为优势修正(DPBAC)算法。在多个D4RL任务上的实验表明,相比现有离线方法,DPBAC展现出更强的策略表达能力,有效缓解了Q值估计偏差,性能优于当前最先进算法。
原文摘要 · Abstract (English)
In offline reinforcement learning (RL), the distribution shift between behavioral data and the learned policy can lead to erroneous \emph{Q}-value estimation, thereby misguiding the direction of policy optimization. To address this issue, we develop a behavioral advantage corrected policy evaluation (BAC-PE) approach, which utilizes the \emph{Q}-function of the behavior policy to correct the learned policy's \emph{Q}-function, thus mitigating pessimistic conservatism and overestimation bias. Furthermore, the convergence of BAC-PE is analyzed theoretically, and an upper bound on the difference between the learned \emph{Q}-function and the true \emph{Q}-function is derived. To alleviate distribution shift, this work employs diffusion models to represent both the behavior policy and the learned policy, performing distribution matching for accurate policy regularization. Additionally, \emph{Q}-value guidance is incorporated into the training process to achieve effective policy improvement. By combining BAC-PE with diffusion policy modeling, we propose the diffusion policy with behavioral advantage correction (DPBAC) algorithm. Compared to existing offline methods, DPBAC demonstrates stronger policy representation capabilities and effectively mitigates the bias in \emph{Q}-value estimation. Experimental results on multiple domains of D4RL tasks show that DPBAC achieves superior performance, with notable advantages over state-of-the-art (SOTA) algorithms.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。