GSWAM: A Unified Gaussian World Action Model for Robotic Manipulation

Chengzhi Zhao1,* Hongbo Duan2,* Yueyang Weng1 Xiaopeng Zhang1 Yongjin Mu1 Bin Qian2 Yanjie Li1,†

1Harbin Institute of Technology, Shenzhen   2Tsinghua University, Shenzhen

* Equal contribution   † Corresponding author

TL;DR: GSWAM jointly models 3D scene dynamics and robot actions in a shared 3D-anchored space for geometrically consistent world modeling and spatially grounded action reasoning.

Abstract

World action models (WAMs) provide a promising learning paradigm for robotic manipulation by jointly modeling future observations and robot actions. However, most existing methods operate in 2D image spaces, making it difficult to explicitly capture spatial scene dynamics, maintain cross-view geometric consistency, and effectively align visual observations with robot actions. To address this, we propose GSWAM, a world action model based on 3D Gaussian Splatting (3DGS), which jointly models scene dynamics and robot actions in a unified 3D space. We first compress 3DGS scenes into geometrically consistent latent representations using a Gaussian variational autoencoder (GSVAE). Furthermore, we align scenes and actions within a shared 3D-anchored space, enabling GSWAM to explicitly model the joint distribution between 3D scene dynamics and robot actions. Based on this representation, we design a 3D-anchor-based spatial diffusion transformer to model spatial interactions between scenes and actions. To accommodate their different generative characteristics, we further employ decoupled flow timesteps for the two modalities, allowing the model to jointly learn scene-action generation and action-conditioned future scene prediction. Experiments on both simulated and real-world robotic manipulation tasks demonstrate that GSWAM outperforms representative existing methods in both action generation and future scene prediction.

Motivation

2D WAMs lack the ability to model spatial physical dynamics and suffer from cross-view geometric inconsistency, while GSWAM jointly models 3D scenes and actions in a 3D-anchored space, enabling geometrically consistent world modeling and spatially grounded action reasoning.

Method

Overview of GSWAM. Given multi-view observations, we construct a structured 3DGS scene and compress it into geometrically consistent 3D-aware latents using GSVAE. Scene tokens and action tokens are then aligned in a unified 3D-anchored space, where a spatial diffusion transformer models their joint distribution with decoupled flow timesteps. The model jointly predicts future robot actions and future 3DGS scene latents, which are decoded into future 3D scenes for rendering and evaluation.

Experiments

Real-world

LIBERO

put the bowl on the plate
put the bowl on the stove
put the white mug on the left plate and put the yellow and white mug on the right plate
turn on the stove and put the moka pot on it
pick up the chocolate pudding and place it in the basket
pick up the orange juice and place it in the basket
pick up the black bowl between the plate and the ramekin and place it on the plate
pick up the black bowl next to the cookie box and place it on the plate

RLBench

close the red jar
push the maroon button then push the green button
put the item in the middle drawer
sweep dirt to the tall dustpan

World Model

libero
realworld-1
realworld-2