arXiv:2606.21297cs.LGcs.AI2026-06

解决低维观测下强化学习的损失不均衡问题,提升样本效率。

NASDAQ: Normalized Observation Space Dynamics-Augmented Q-Learning

  • 通过归一化观测空间平衡各维度重建损失,避免重要信息被忽略。
  • 在多种任务上性能超越或媲美先进方法,训练时间大幅缩短。
  • 适用于物理状态与图像等多类型输入,统一处理低维与高维观测。

将基于观测动态预测的表示学习引入无模型强化学习,可提升样本效率和性能,且改动小、计算开销低。但在低维观测的复杂任务中仍表现不佳。本文发现其根本原因在于各观测维度的重建损失不平衡:取值范围大的维度主导损失,导致小范围维度被忽略。为此,提出一种专为在线强化学习设计的归一化方法,对低维观测进行归一化,平衡损失与梯度。该方法使动态预测在归一化观测空间中进行,实现对低维与高维输入(如物理状态和图像)的统一处理。在此基础上,提出归一化观测空间动态增强Q学习(NASDAQ),通过耦合价值学习与两个辅助任务——短期价值预测和下一归一化观测预测——学习状态-动作表示。大量实验表明,NASDAQ在多种任务上表现优于或媲美顶尖模型基和自预测强化学习方法,同时显著减少训练耗时。

原文摘要 · Abstract (English)

Augmenting model-free reinforcement learning (RL) with representations learned through observation dynamics prediction (observation-predictive RL) can improve sample efficiency and performance, with minor modifications and limited additional computation. However, this approach still struggles in challenging tasks with low-dimensional observations. In this paper, we identify a key factor behind this problem: unbalanced reconstruction losses across observation dimensions, where dimensions with larger value ranges dominate the loss. This encourages the agent to neglect dimensions with relatively small ranges, leading to degraded performance. To address this issue, we propose a novel normalization method tailored to online RL, which normalizes low-dimensional observations and balances the resulting losses and gradients. Beyond balancing reconstruction losses, observation normalization enables dynamics prediction to be performed in a normalized observation space, thereby providing a unified treatment of low- and high-dimensional inputs (e.g., physical states and images). Building on this idea, we further introduce Normalized Observation Space Dynamics-Augmented Q-learning (NASDAQ), a framework for observation-predictive RL applicable across diverse domains. NASDAQ learns state-action representations by coupling value learning with two auxiliary tasks: short-term value prediction and next normalized observation prediction. Extensive experiments demonstrate that NASDAQ achieves competitive or superior performance compared with state-of-the-art model-based and self-predictive RL methods, while requiring significantly less training wall-time.

强化学习动态预测观测归一化样本效率

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