arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2406.03324cs.LG

UDQL:弥合离线强化学习中MSE损失与最优值函数之间的差距

UDQL: Bridging The Gap between MSE Loss and The Optimal Value Function in Offline Reinforcement Learning

Shenyang University of Chemical Technology(沈阳化工大学);University of Macau(澳门大学);University of Louisville(路易斯维尔大学)

Yu Zhang, Rui Yu, Zhipeng Yao, Wenyuan Zhang, Jun Wang, Liming Zhang

AI总结:

本文发现MSE会导致离线强化学习中值函数的过估计现象,通过理论分析给出误差上界,并提出Bellman低估算子与扩散策略模型结合的算法,在D4RL任务上优于现有方法。

摘要:

在绝大多数离线强化学习(RL)模型中,均方误差(MSE)通常被用于估计最优值函数的解,并取得了出色的性能。然而,我们发现其原理会导致值函数的过估计现象。在本文中,我们首先从理论上分析了MSE导致的过估计现象,并给出了过估计误差的理论上界。此外,为了解决这一问题,我们提出了一种新颖的Bellman低估算子来抵消过估计现象,随后证明了其收缩特性。最后,我们提出了基于低估算子和扩散策略模型的离线RL算法。在D4RL任务上的大量实验结果表明,我们的方法优于最先进的离线RL算法,这证明了我们的理论分析和低估方法对离线RL任务是有效的。

英文摘要:

The Mean Square Error (MSE) is commonly utilized to estimate the solution of the optimal value function in the vast majority of offline reinforcement learning (RL) models and has achieved outstanding performance. However, we find that its principle can lead to overestimation phenomenon for the value function. In this paper, we first theoretically analyze overestimation phenomenon led by MSE and provide the theoretical upper bound of the overestimated error. Furthermore, to address it, we propose a novel Bellman underestimated operator to counteract overestimation phenomenon and then prove its contraction characteristics. At last, we propose the offline RL algorithm based on underestimated operator and diffusion policy model. Extensive experimental results on D4RL tasks show that our method can outperform state-of-the-art offline RL algorithms, which demonstrates that our theoretical analysis and underestimation way are effective for offline RL tasks.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑