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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Hopper Hop | DeepMind Control Suite (DMC) | Steps Required (k)153.5 | 12 | |
| Reinforcement Learning | DeepMind Control Cartpole Balance Sparse | Steps to 75% Return1.58e+5 | 11 | |
| Cup Catch | DeepMind Control Suite (DMC) | Sample Efficiency (Steps)1.05e+5 | 10 | |
| Walker Run | DeepMind Control Suite (DMC) | Steps (k)494.9 | 10 | |
| Reinforcement Learning | DeepMind Control Reacher Hard | Steps to 75% Return (k)707 | 8 | |
| Acrobot Swingup | DeepMind Control Suite (DMC) | Steps to 80% Return5.26e+5 | 6 | |
| Cartpole Balance Sparse | DeepMind Control Suite (DMC) | Steps to 80% Proficiency1.58e+5 | 6 | |
| Cartpole Swingup | DeepMind Control Suite (DMC) | Steps (k)3.33e+5 | 6 | |
| Continuous Control | DMC Vision | Acrobot Swingup Score474.2 | 6 | |
| Finger Spin | DeepMind Control Suite (DMC) | Steps to 80% Return (k)102.3 | 6 |