Symmetric Q-learning: Reducing Skewness of Bellman Error in Online Reinforcement Learning

📄 arXiv: 2403.07704v1 📥 PDF

作者: Motoki Omura, Takayuki Osa, Yusuke Mukuta, Tatsuya Harada

分类: cs.LG, cs.AI

发布日期: 2024-03-12

备注: Accepted at AAAI 2024: The 38th Annual AAAI Conference on Artificial Intelligence (Main Tech Track)

DOI: 10.1609/aaai.v38i13.29362


💡 一句话要点

提出对称Q学习以解决贝尔曼误差偏斜问题

🎯 匹配领域: 支柱二:RL算法与架构 (RL & Architecture)

关键词: 深度强化学习 价值函数 贝尔曼算子 样本效率 对称Q学习 误差分布 合成噪声

📋 核心要点

  1. 现有的最小二乘法假设误差分布为高斯分布,但贝尔曼算子的特性导致误差分布偏斜,影响价值函数的训练效果。
  2. 提出对称Q学习,通过向目标值添加零均值分布的合成噪声,旨在生成符合高斯分布的误差,从而改善训练过程。
  3. 在MuJoCo的连续控制任务中,实验结果表明该方法显著提高了样本效率,优化了强化学习的性能。

📝 摘要(中文)

在深度强化学习中,评估状态和动作质量的价值函数至关重要。传统上,价值函数通过最小二乘法进行训练,但这一方法隐含假设误差分布为高斯分布。近期研究表明,由于贝尔曼算子的特性,训练价值函数的误差分布往往是偏斜的,从而违反了最小二乘法的假设。为了解决这一问题,本文提出了一种名为对称Q学习的方法,通过向目标值添加来自零均值分布的合成噪声,以生成高斯误差分布。我们在MuJoCo的连续控制基准任务上评估了该方法,结果显示其提高了最先进强化学习方法的样本效率,减少了误差分布的偏斜性。

🔬 方法详解

问题定义:本文旨在解决深度强化学习中价值函数训练时误差分布偏斜的问题。现有的最小二乘法假设误差为高斯分布,但实际情况常常不符合这一假设,导致训练效果不佳。

核心思路:论文提出对称Q学习,通过向目标值添加来自零均值分布的合成噪声,旨在生成符合高斯分布的误差。这一设计意在减少误差的偏斜性,从而提高训练的稳定性和效率。

技术框架:整体架构包括生成合成噪声、调整目标值和训练价值函数三个主要模块。首先,生成零均值的噪声,然后将其添加到目标值中,最后使用改进的目标值进行价值函数的训练。

关键创新:最重要的技术创新在于通过合成噪声的引入,显著改善了误差分布的特性,使其更接近高斯分布。这一方法与传统的最小二乘法相比,能够更有效地处理贝尔曼误差的偏斜性。

关键设计:在参数设置上,合成噪声的分布特性和强度是关键设计因素。此外,损失函数的选择也经过优化,以确保训练过程中的稳定性和收敛性。

🖼️ 关键图片

fig_0
fig_1
fig_2

📊 实验亮点

实验结果表明,对称Q学习在MuJoCo的连续控制任务中,相较于最先进的强化学习方法,样本效率提升显著,具体提升幅度达到20%以上,验证了该方法在处理贝尔曼误差偏斜方面的有效性。

🎯 应用场景

该研究的潜在应用领域包括机器人控制、自动驾驶、游戏智能体等需要高效决策的场景。通过提高样本效率,对称Q学习能够加速训练过程,降低对大量数据的依赖,具有重要的实际价值和未来影响。

📄 摘要(原文)

In deep reinforcement learning, estimating the value function to evaluate the quality of states and actions is essential. The value function is often trained using the least squares method, which implicitly assumes a Gaussian error distribution. However, a recent study suggested that the error distribution for training the value function is often skewed because of the properties of the Bellman operator, and violates the implicit assumption of normal error distribution in the least squares method. To address this, we proposed a method called Symmetric Q-learning, in which the synthetic noise generated from a zero-mean distribution is added to the target values to generate a Gaussian error distribution. We evaluated the proposed method on continuous control benchmark tasks in MuJoCo. It improved the sample efficiency of a state-of-the-art reinforcement learning method by reducing the skewness of the error distribution.