ArXiv

DLAM: Distributional Latent Actions with Temporal Constraints

Authors
Zuojin Tang, Feifan Luo, Haoyun Liu...
Categories
cs.RO, cs.AI, cs.CV
arXiv
https://arxiv.org/abs/2607.27138v1
PDF
https://arxiv.org/pdf/2607.27138v1

Brief

DLAM (Distributional Latent Actions with Temporal Constraints) tackles limited action-labeled robot data by extracting distributional latent-action priors from action-free videos. It represents each transition as a diagonal Gaussian, enforces normalized composition and reversal over equal-gap triplets, and models adjacent-transition dependence with a shared-correlation coefficient. Freezing the encoder and training a flow-matching policy yields more consistent latent dynamics, better reconstruction on held-out videos, and improved transfer to downstream robot benchmarks (MetaWorld MT50, LIBERO, real tasks).

Why it matters

DLAM models each visual transition as a diagonal Gaussian latent action, grounding the mean via reconstruction from a reference frame and constraining mean and per-dimension variance through normalized composition and reversal over equal-gap triplets.

Key details

  • Variance composition uses a lightweight shared-correlation coefficient to model dependence between adjacent transitions; reversal negates the mean while preserving variance.
  • For control, the encoder is frozen and a flow-matching policy is trained to jointly generate mean transition sequences and robot actions; under the same π0 transfer protocol DLAM yields more temporally consistent latents, stronger direct and cumulative reconstruction on held-out videos, and improved policy performance on MetaWorld MT50, LIBERO, and real-world manipulation tasks (arXiv, 2026-07-29).
Source evidence

Abstract

Vision-language-action (VLA) models remain constrained by scarce action-labeled robot data, whereas action-free videos offer abundant observations of physical change. Latent action models can extract such priors, but reconstruction-trained codes may predict future observations without the structure required for joint generation with robot actions. Existing structured methods add temporal constraints but retain deterministic transition points, so residual errors in locally inferred transitions may propagate and compound under recursive composition. We introduce DLAM, a distributional latent-action model that represents each transition as a diagonal Gaussian. Reconstruction conditioned on the reference frame grounds the mean in observed visual change, while normalized composition and reversal over equal-gap triplets constrain both the mean and dimension-wise variance. Variance composition uses a lightweight shared-correlation coefficient to account for dependence between adjacent transitions that share an intermediate frame, whereas reversal negates the mean and preserves the variance. For downstream policy learning, we freeze the encoder and train a flow-matching policy to jointly generate mean transition sequences and robot actions. On held-out transitions, DLAM learns more temporally consistent latent dynamics than existing latent-action baselines and achieves stronger direct and cumulative reconstruction on held-out videos. Under the same controlled $π_0$ transfer protocol, it also improves policy performance on MetaWorld MT50, LIBERO, and real-world manipulation tasks. Controlled ablations show that normalized mean constraints account for most of the reconstruction gain, while learned variance and correlation-aware composition provide complementary improvements in downstream control.