Overcoming Statistical Bias in Action-Controllable World Models

• URL: http://arxiv.org/abs/2608.04653v1
• Date: Wed, 05 Aug 2026 10:10:36 GMT
• Title: Overcoming Statistical Bias in Action-Controllable World Models
• Abstract summary: Action-conditioned world models aim to predict how visual environments evolve under an agent's actions.<n>Future frames are often highly predictable from visual inertia and recurring motion patterns alone.<n>We introduce CoCo, a Counterfactual Consistency framework to enhance action controllability.
• Score: 18.890337378174678
• License: http://creativecommons.org/licenses/by-nc-sa/4.0/
• Abstract: Action-conditioned world models aim to predict how visual environments evolve under an agent's actions. Yet future frames are often highly predictable from visual inertia and recurring motion patterns alone. This creates a shortcut: models can fit the data by exploiting statistical biases without making their visible dynamics meaningfully depend on the action. As a result, different actions may produce similar futures, while motion may persist even under zero action. The key question is how to reduce reliance on statistical shortcuts from dominating action-conditioned prediction. We argue that action control requires more than injecting action features; it requires enforcing consistency under counterfactual changes to actions and observations. Based on this insight, we introduce CoCo, a Counterfactual Consistency framework to enhance action controllability through two complementary constraints. Multi-step counterfactual consistency constrains reference, inverse-action, and zero-action rollouts, while action-spatial counterfactual consistency enforces consistent predictions under mirrored scenes and transformed actions. Together, they reduce reliance on statistical shortcuts from substituting for action-dependent dynamics. We further introduce Action Response Consistency (ARC) and Drift Energy (DE) to assess action controllability, together with Mini-SSMB for same-state, multi-action counterfactual evaluation. On Mini-SSMB, our full model achieved ARC_inv of 0.412 and ARC_ref of 0.483, while reducing DE by 17.07% relative to the baseline. On VP2 visual planning, it achieves the highest average success rate among SOTA models, at 73.1%. Experiments on BAIR and RoboNet further show that these gains preserve video prediction quality and transfer across model settings.

Related papers

• SLIM-0.5B: Learning Action-Grounded Predictive Latents for Robot Manipulation [35.54319311306262]
  We present SLIM, a compact latent interaction policy.<n>It learns representations through self-supervised masked trajectory prediction.<n>SLIM matches or exceeds representative large-scale VLA and world-action-model baselines with fewer parameters.
  arXiv  Detail & Related papers  (2026-08-10T15:58:39Z)
• Causally Debiased Latent Action Model for Embodied Action Conditioned World Models [47.281442841934876]
  Action-conditioned world models (ACWMs) aim to simulate future observations conditioned on embodied actions.<n>Existing LAMs are typically trained with reconstruction-only objectives.<n>We propose CD-LAM, a causally debiased framework for LAM-based ACWMs.
  arXiv  Detail & Related papers  (2026-07-10T08:20:27Z)
• Sensorimotor World Models: Perception for Action via Inverse Dynamics [59.121736411156384]
  We introduce a sensorimotor world model trained end-to-end with inverse dynamics regularization.<n>It prevents representation collapse and induces action-aligned representations.<n>It learns compact, interpretable latent spaces and enables competitive planning performance across simple 2D and 3D control tasks.
  arXiv  Detail & Related papers  (2026-06-18T11:25:16Z)
• SC3-Eval: Evaluating Robot Foundation Models via Self-Consistent Video Generation [73.2142090645987]
  SC3-Eval is a self-consistent video generation recipe that adapts a pre-trained video foundation model into an accurate policy evaluator.<n>We demonstrate SC3-Eval rollouts reproduce the failure modes that policies exhibit in real-world rollouts, supporting fine-grained diagnostic comparison.
  arXiv  Detail & Related papers  (2026-06-17T02:15:46Z)
• Feedback World Model Enables Precise Guidance of Diffusion Policy [36.965417653906535]
  We propose feedback world model, a new paradigm that closes the loop between prediction and observation at inference time.<n>We show that our method substantially improves both prediction accuracy and policy performance under distribution shift.<n>In particular, it reduces world model prediction error by up to 76.4% and improves out-of-distribution (OOD) success rate by 30%.
  arXiv  Detail & Related papers  (2026-05-15T07:52:13Z)
• From Imagined Futures to Executable Actions: Mixture of Latent Actions for Robot Manipulation [88.39072412680633]
  We propose MoLA, a control-oriented interface that transforms imagined future videos into executable representations.<n>We evaluate our approach on simulated benchmarks and real-world robot manipulation tasks.
  arXiv  Detail & Related papers  (2026-05-12T14:15:16Z)
• Is the Future Compatible? Diagnosing Dynamic Consistency in World Action Models [28.165928090888986]
  World Action Models (WAMs) enable decision-making through imagined rollouts by predicting future observations and actions.<n>We identify action-state consistency, the alignment between predicted actions and induced state transitions, as a missing reliability axis for WAMs.<n>We introduce a value-free consensus strategy for test-time selection, which ranks candidate rollouts by agreement among predicted futures.
  arXiv  Detail & Related papers  (2026-05-08T09:44:43Z)
• World Action Verifier: Self-Improving World Models via Forward-Inverse Asymmetry [82.93104394404781]
  General-purpose world models promise scalable policy evaluation, optimization, and planning.<n>We propose World Action Verifier (WAV), a framework that enables world models to identify their own prediction errors and self-improve.
  arXiv  Detail & Related papers  (2026-04-02T12:48:36Z)
• ResWM: Residual-Action World Model for Visual RL [0.06999740786886534]
  Learning predictive world models from raw visual observations is a central challenge in reinforcement learning (RL)<n>We introduce the Residual-Action World Model (ResWM), a new framework that reformulates the control variable from absolute actions to residual actions.<n>ResWM produces more stable and energy-efficient action trajectories, a property critical for robotic systems deployed in real-world environments.
  arXiv  Detail & Related papers  (2026-03-11T11:27:08Z)
• Reshaping Action Error Distributions for Reliable Vision-Language-Action Models [69.38615670891038]
  In robotic manipulation, vision-language-action (VLA) models have emerged as a promising paradigm for learning generalizable and scalable robot policies.<n>We focus on continuous-action VLA models and move beyond conventional MSE-based regression by reshaping action error distributions during training.<n>We evaluate our approaches across standard, few-shot, and noisy settings on multiple representative VLA architectures.
  arXiv  Detail & Related papers  (2026-02-04T05:37:09Z)
• Drift No More? Context Equilibria in Multi-Turn LLM Interactions [58.69551510148673]
  contexts drift is the gradual divergence of a model's outputs from goal-consistent behavior across turns.<n>Unlike single-turn errors, drift unfolds temporally and is poorly captured by static evaluation metrics.<n>We show that multi-turn drift can be understood as a controllable equilibrium phenomenon rather than as inevitable decay.
  arXiv  Detail & Related papers  (2025-10-09T04:48:49Z)

This list is automatically generated from the titles and abstracts of the papers in this site.

This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.