Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Mind Dreamer: Untethering Imagination via Active Causal Intervention on Latent Manifolds

About

Model-Based Reinforcement Learning yields sample efficiency via latent imagination, yet remains constrained by Historical Tethering: imagination is typically initialized from observed states. This creates a learning asymmetry, where the world model's manifold discovery outpaces the policy's sparse-reward optimization. We propose Mind Dreamer (MD), a framework that instantiates Active Causal Intervention to transcend Markovian continuity. MD reformulates discovery as the minimization of a global Relay Expected Free Energy. Instead of initializing from historical data, it draws initial states from an adversarial generator $s_0 \sim p_{gen}(\cdot)$, creating non-continuous latent jumps to epistemic blind spots that are physically plausible yet cognitively challenging. We derive Relay Value Function and Relay Uncertainty Function to resolve the credit assignment paradox across these spatial ruptures. Treating synthesized anchors as interventional intermediary states, these potentials propagate pragmatic and epistemic value through Bellman-style backups. Notably, we prove that uncertainty propagation across discontinuities necessitates a quadratic discount $\gamma^2$, establishing a formal epistemic horizon. Theoretically, MD approximates a variance-minimizing importance sampler that expands the manifold's spectral gap, reducing the hitting time to critical bottleneck states. Empirically, MD achieves a 1.67$\times$ average speedup over DreamerV3 on DeepMind Control Suite, reaching 8.8$\times$ in sparse-reward tasks.

Shaojun Xu, Xiaoling Zhou, Yihan Lin, Yapeng Meng, Xinglong Ji, Luping Shi, Rong Zhao• 2026

Related benchmarks

TaskDatasetResultRank
Hopper HopDeepMind Control Suite (DMC)
Steps Required (k)153.5
12
Reinforcement LearningDeepMind Control Cartpole Balance Sparse
Steps to 75% Return1.58e+5
11
Cup CatchDeepMind Control Suite (DMC)
Sample Efficiency (Steps)1.05e+5
10
Walker RunDeepMind Control Suite (DMC)
Steps (k)494.9
10
Reinforcement LearningDeepMind Control Reacher Hard
Steps to 75% Return (k)707
8
Acrobot SwingupDeepMind Control Suite (DMC)
Steps to 80% Return5.26e+5
6
Cartpole Balance SparseDeepMind Control Suite (DMC)
Steps to 80% Proficiency1.58e+5
6
Cartpole SwingupDeepMind Control Suite (DMC)
Steps (k)3.33e+5
6
Continuous ControlDMC Vision
Acrobot Swingup Score474.2
6
Finger SpinDeepMind Control Suite (DMC)
Steps to 80% Return (k)102.3
6
Showing 10 of 44 rows

Other info

Follow for update