Dreaming Smoothly and Sample Efficiently with Gradient Penalized Latent Dynamics
TLDR
GPLD regularizes DreamerV3's latent dynamics with a Jacobian penalty to improve sample efficiency in continuous control tasks.
Reasoning
The paper presents a simple yet effective regularization technique that improves sample efficiency, especially on complex locomotion tasks. However, the evaluation is limited to simulated proprioceptive tasks, and the method's generality to other domains (e.g., visual or discrete) is not demonstrated.
Read-first score
Read-first score 60.2, weighted from topical fit, citation, graph, method, reproducibility, and recency signals. Original total remains 38.
Field roles
Rank sensitivity
Stability: volatile; rank range: 471.
Keyword Scores
Deep Analysis
Innovations
- Gradient-penalized latent dynamics regularizer (GPLD) for DreamerV3
- Row-wise Jacobian penalty on the posterior latent distribution to encourage locally smooth transition learning
- Interpretation of the penalty as the continuous-latent analog of finite-difference smoothing in discrete embedded-state MDPs
- Efficient estimation of the penalty using Hutchinson-style stochastic probes
Methodology
The authors propose GPLD, a regularizer that adds a row-wise Jacobian penalty to the posterior latent distribution of DreamerV3's latent dynamics. The penalty is estimated efficiently via Hutchinson stochastic probes. The method is evaluated on DeepMind Control proprioceptive tasks and quadruped locomotion tasks, comparing sample efficiency and learning consistency against the DreamerV3 baseline.
Key Results
GPLD improves aggregate sample efficiency across DeepMind Control tasks, with particularly strong gains on higher-complexity locomotion environments. On quadruped tasks, GPLD reaches high-return behavior earlier and exhibits more consistent late-stage learning over longer horizons.
Limitations
- The method is demonstrated only on smooth continuous control environments; its effectiveness on discrete or non-smooth environments is not evaluated.
- The computational overhead of the Hutchinson-style stochastic probes is not quantified in the abstract.