Any 3D Scene is Worth 1K Tokens: 3D-Grounded Representation for Scene Generation at Scale
Abstract
Direct diffusion modeling in a 3D latent space improves spatial consistency and efficiency for scene-level 3D generation compared to 2D-based methods.
Scene-level 3D generation has long been dominated by 2D multi-view or video diffusion models. Typically, these 2D-based approaches perform scene generation in a 2D latent space, which introduces two fundamental issues: (i) representing 3D scenes via 2D views leads to significant representation redundancy, and (ii) latent space rooted in 2D inherently limits the spatial consistency of the generated scenes. In this paper, we propose to perform 3D scene generation directly within a high-dimensional implicit 3D latent space derived from powerful 2D foundation models (e.g., DINOv2, SigLIP2, DA3). Then we tame diffusion transformers to perform diffusion modeling directly in the 3D latent space, enabling 3D-grounded scene generation. In our experiments, we not only demonstrate the advantages of our method over 2D-based approaches in terms of efficiency and spatial consistency, but also comprehensively analyze the impact of 3D latent representation on 3D scene generation performance. Furthermore, we validate the favorable scalability of our method with respect to representation and model capacity.
Community
๐ 3DRAE has been accepted to NeurIPS 2026!
We introduce a compact 3D-grounded latent representation for efficient and spatially consistent scene generation. A scene can be represented with only 256--1K latent tokens, generated in about 2 seconds, and decoded along arbitrary camera trajectories.
- Paper: https://arxiv.org/abs/2604.11331
- Project: https://wswdx.github.io/3DRAE
- Code: https://github.com/WU-CVGL/3DRAE
- Models: https://huggingface.co/tracywei/3drae
- Dataset: https://huggingface.co/datasets/tracywei/scannet_3drae
- Demo: https://huggingface.co/spaces/tracywei/3DRAE-Demo
- Collection: https://huggingface.co/collections/tracywei/3drae-3d-grounded-scene-generation
Feedback and questions are very welcome!
This paper has just been accepted to NeurIPS 2026. In this paper, we introduce 3DRAE, a representation autoencoder that transforms a flexible number of images into a compact, fixed-length, and view-decoupled 3D latent representation using powerful 2D foundation models. Building on this semantically rich 3D representation, 3DDiT performs diffusion directly in the 3D latent space, enabling efficient and spatially consistent scene generation. A scene can be represented with only 256โ1K latent tokens and generated in about 2 seconds. Once generated, it can be decoded along arbitrary camera trajectories without repeated diffusion sampling. We further demonstrate favorable scaling with both representation and model capacity. Code, models, data, and demos are publicly available.
Get this paper in your agent:
hf papers read 2604.11331 Don't have the latest CLI?
curl -LsSf https://hf.co/cli/install.sh | bash