From Part to Whole: 3D Generative World Model with an Adaptive Structural Hierarchy
Organizations: Institute for Math & AI, Wuhan University, Wuhan, China
Abstract
Single-image 3D generation lies at the core of vision-to-graphics models in the real world. However, it remains a fundamental challenge to achieve reliable generalization across diverse semantic categories and highly variable structural complexity under sparse supervision. Existing approaches typically model objects in a monolithic manner or rely on a fixed number of parts, including recent part-aware models such as PartCrafter, which still require a labor-intensive user-specified part count. Such designs easily lead to overfitting, fragmented or missing structural components, and limited compositional generalization when encountering novel object layouts. To this end, this paper rethinks single-image 3D generation as learning an adaptive part-whole hierarchy in the flexible 3D latent space. We present a novel part-to-whole 3D generative world model that autonomously discovers latent structural slots by inferring soft and compositional masks directly from image tokens. Specifically, an adaptive slot-gating mechanism dynamically determines the slot-wise activation probabilities and smoothly consolidates redundant slots within different objects, ensuring that the emergent structure remains compact yet expressive across categories. Each distilled slot is then aligned to a learnable, class-agnostic prototype bank, enabling powerful cross-category shape sharing and denoising through universal geometric prototypes in the real world. Furthermore, a lightweight 3D denoiser is introduced to reconstruct geometry and appearance via unified diffusion objectives. Experiments show consistent gains in cross-category transfer and part-count extrapolation, and ablations confirm complementary benefits of the prototype bank for shape-prior sharing as well as slot-gating for structural adaptation.
Figures & tables
| Method | Segment Mask | Fixed Part Number | Objaverse [ 24 ] | ShapeNet [ 25 ] | ABO [ 26 ] | ||||||
| IoU | F-Score | CD | IoU | F-Score | CD | IoU | F-Score | CD | |||
| Dataset | / | / | 0.080 | / | / | 0.183 | / | / | 0.014 | / | / |
| TripoSG [ 15 ] | / | / | / | 0.594 | 0.311 | / | 0.505 | 0.375 | / | 0.710 | 0.202 |
| HoloPart [ 28 ] | 0.044 | 0.692 | 0.192 | 0.111 | 0.550 | 0.351 | 0.045 | 0.809 | 0.134 | ||
| PartCrafter [ 14 ] | 0.036 | 0.747 | 0.173 | 0.029 | 0.567 | 0.321 | 0.024 | 0.862 | 0.105 | ||
| Ours | 0.032 | 0.781 | 0.159 | 0.026 | 0.573 | 0.317 | 0.022 | 0.891 | 0.089 | ||
| Method | 3D-Front [ 27 ] | 3D-Front (Occluded) [ 27 ] | ||||
|---|---|---|---|---|---|---|
| IoU | F-Score | CD | IoU | F-Score | CD | |
| MIDI [ 21 ] | 0.001 | 0.793 | 0.160 | 0.002 | 0.662 | 0.259 |
| PartCrafter [ 14 ] | 0.003 | 0.815 | 0.149 | 0.004 | 0.780 | 0.151 |
| Ours | 0.002 | 0.852 | 0.128 | 0.003 | 0.785 | 0.124 |
| Setting | Metrics | ||||
|---|---|---|---|---|---|
| ASG | PB | Warm-up | IoU | F-Score | CD |
| ✓ | ✓ | 0.035 | 0.751 | 0.168 | |
| ✓ | ✓ | 0.039 | 0.710 | 0.184 | |
| ✓ | ✓ | 0.048 | 0.602 | 0.265 | |
| ✓ | ✓ | ✓ | 0.032 | 0.781 | 0.159 |
3D Asset Generation
3D asset generation creates and edits objects, characters, materials, and scenes from text, images, video, or geometric constraints.
See all 167 papers