CRAG: Can 3D Generative Models Help 3D Assembly?
About
Most existing 3D assembly methods treat the problem as pure pose estimation, rearranging observed parts via rigid transformations. In contrast, human assembly naturally couples structural reasoning with holistic shape inference. Inspired by this intuition, we reformulate 3D assembly as a joint problem of assembly and generation. We show that these two processes are mutually reinforcing: assembly provides part-level structural priors for generation, while generation injects holistic shape context that resolves ambiguities in assembly. Unlike prior methods that cannot synthesize missing geometry, we propose CRAG, which simultaneously generates plausible complete shapes and predicts poses for input parts. Extensive experiments demonstrate state-of-the-art performance across in-the-wild objects with diverse geometries, varying part counts, and missing pieces. Project Page: https://ai4ce.github.io/CRAG/
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| 3D Fracture Reassembly | Breaking Bad 2022 (Complete) | Reconstruction Error (RE)8 | 5 | |
| 3D Fracture Reassembly | Breaking Bad 2022 | RE11.43 | 5 | |
| 3D Part Assembly | PartNeXt 2025a (Complete) | RE45.12 | 5 | |
| 3D Part Assembly | PartNeXt 2025a | Reconstruction Error (RE)42.33 | 5 | |
| 3D Shape Completion | PartNeXt missing-part setting 2–20 parts | CD (L1)9.39 | 4 | |
| 3D Shape Generation | PartNeXt (test) | CD-L16.53 | 3 |