arXiv:2509.24804cs.LG2025-09NeurIPS被引 2

通过动态调制提升视觉强化学习的采样效率

DyMoDreamer: World Modeling with Dynamic Modulation

  • 用帧间差分掩码提取物体运动特征,分离动态与静态信息
  • 在Atari 100k上达156.6%人类表现,创深度思维控制套件新纪录832
  • 适合追求高效视觉决策的强化学习研究者

深度强化学习的核心瓶颈是样本效率低下,训练高性能智能体通常需要大量环境交互。模型基于强化学习(MBRL)通过构建世界模型模拟环境动态并生成合成经验来缓解该问题。然而,传统世界模型对观测进行整体处理,无法解耦动态物体与静态背景的时序特征,导致计算效率低,尤其在视觉任务中,动态物体对奖励和决策影响显著。为此,我们提出DyMoDreamer,一种新型MBRL算法,引入动态调制机制以增强动态特征提取并丰富时序信息。DyMoDreamer采用新颖的帧间差分掩码生成差异观测,显式编码物体级运动线索与时间动态。动态调制以随机分类分布建模,并融入循环状态空间模型(RSSM),提升模型对奖励相关动态的关注。实验表明,DyMoDreamer在Atari 100k基准上达到156.6%均值人类归一化得分,创下深度思维控制套件新纪录832分,并在Crafter基准上于100万步后获得9.5%性能提升。代码已开源。

原文摘要 · Abstract (English)

A critical bottleneck in deep reinforcement learning (DRL) is sample inefficiency, as training high-performance agents often demands extensive environmental interactions. Model-based reinforcement learning (MBRL) mitigates this by building world models that simulate environmental dynamics and generate synthetic experience, improving sample efficiency. However, conventional world models process observations holistically, failing to decouple dynamic objects and temporal features from static backgrounds. This approach is computationally inefficient, especially for visual tasks where dynamic objects significantly influence rewards and decision-making performance. To address this, we introduce DyMoDreamer, a novel MBRL algorithm that incorporates a dynamic modulation mechanism to improve the extraction of dynamic features and enrich the temporal information. DyMoDreamer employs differential observations derived from a novel inter-frame differencing mask, explicitly encoding object-level motion cues and temporal dynamics. Dynamic modulation is modeled as stochastic categorical distributions and integrated into a recurrent state-space model (RSSM), enhancing the model's focus on reward-relevant dynamics. Experiments demonstrate that DyMoDreamer sets a new state-of-the-art on the Atari $100$k benchmark with a $156.6$\% mean human-normalized score, establishes a new record of $832$ on the DeepMind Visual Control Suite, and gains a $9.5$\% performance improvement after $1$M steps on the Crafter benchmark. Our code is released at https://github.com/Ultraman-Tiga1/DyMoDreamer.

强化学习世界模型动态建模视觉决策

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