Search papers, labs, and topics across Lattice.
This paper introduces Discrete-WAM, a novel framework for autonomous driving that utilizes aligned discrete tokens to represent future visual states and ego actions, enabling compositional causal reasoning across various driving scenarios. By integrating world modeling, action policy, and hierarchical decision-making into a unified discrete diffusion framework, Discrete-WAM enhances the ability to reason about counterfactual futures. Experimental results demonstrate that Discrete-WAM not only achieves competitive performance on large-scale benchmarks but also facilitates controllable generation and improved decision-making reliability.
Discrete-WAM enables compositional causal reasoning in autonomous driving, outperforming traditional methods that struggle with complex state-action dynamics.
Autonomous driving requires reasoning about how ego actions shape the evolution of the surrounding world. However, most end-to-end methods rely on direct state-to-action mappings, capturing correlations without explicitly modeling action-conditioned dynamics. Conversely, continuous-latent world models often lack compositional structure for causal reasoning across counterfactual futures. We introduce Discrete-WAM, a unified latent vision-action world policy that represents future visual states and ego actions as aligned discrete tokens, enabling compositional causal reasoning across alternative futures. Built upon this unified discrete alignment, Discrete-WAM establishes a shared discrete diffusion framework with unified generative tasks, jointly formulating world modeling, world-action policy, and hierarchical decision-enabled policy, supporting compositional generalization across diverse driving scenarios. Experiments on large-scale autonomous-driving benchmarks show that Discrete-WAM achieves competitive performance while supporting controllable generation and counterfactual reasoning, offering a principled path toward more reliable decision-making.
