Make-A-Shape: a Ten-Million-scale 3D Shape Model

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Hui, Ka-Hei, Sanghi, Aditya, Rampini, Arianna, Malekshan, Kamal Rahimi, Liu, Zhengzhe, Shayani, Hooman, Fu, Chi-Wing
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866916387356672000
author Hui, Ka-Hei
Sanghi, Aditya
Rampini, Arianna
Malekshan, Kamal Rahimi
Liu, Zhengzhe
Shayani, Hooman
Fu, Chi-Wing
author_facet Hui, Ka-Hei
Sanghi, Aditya
Rampini, Arianna
Malekshan, Kamal Rahimi
Liu, Zhengzhe
Shayani, Hooman
Fu, Chi-Wing
contents Significant progress has been made in training large generative models for natural language and images. Yet, the advancement of 3D generative models is hindered by their substantial resource demands for training, along with inefficient, non-compact, and less expressive representations. This paper introduces Make-A-Shape, a new 3D generative model designed for efficient training on a vast scale, capable of utilizing 10 millions publicly-available shapes. Technical-wise, we first innovate a wavelet-tree representation to compactly encode shapes by formulating the subband coefficient filtering scheme to efficiently exploit coefficient relations. We then make the representation generatable by a diffusion model by devising the subband coefficients packing scheme to layout the representation in a low-resolution grid. Further, we derive the subband adaptive training strategy to train our model to effectively learn to generate coarse and detail wavelet coefficients. Last, we extend our framework to be controlled by additional input conditions to enable it to generate shapes from assorted modalities, e.g., single/multi-view images, point clouds, and low-resolution voxels. In our extensive set of experiments, we demonstrate various applications, such as unconditional generation, shape completion, and conditional generation on a wide range of modalities. Our approach not only surpasses the state of the art in delivering high-quality results but also efficiently generates shapes within a few seconds, often achieving this in just 2 seconds for most conditions. Our source code is available at https://github.com/AutodeskAILab/Make-a-Shape.
format Preprint
id arxiv_https___arxiv_org_abs_2401_11067
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Make-A-Shape: a Ten-Million-scale 3D Shape Model
Hui, Ka-Hei
Sanghi, Aditya
Rampini, Arianna
Malekshan, Kamal Rahimi
Liu, Zhengzhe
Shayani, Hooman
Fu, Chi-Wing
Computer Vision and Pattern Recognition
Graphics
Significant progress has been made in training large generative models for natural language and images. Yet, the advancement of 3D generative models is hindered by their substantial resource demands for training, along with inefficient, non-compact, and less expressive representations. This paper introduces Make-A-Shape, a new 3D generative model designed for efficient training on a vast scale, capable of utilizing 10 millions publicly-available shapes. Technical-wise, we first innovate a wavelet-tree representation to compactly encode shapes by formulating the subband coefficient filtering scheme to efficiently exploit coefficient relations. We then make the representation generatable by a diffusion model by devising the subband coefficients packing scheme to layout the representation in a low-resolution grid. Further, we derive the subband adaptive training strategy to train our model to effectively learn to generate coarse and detail wavelet coefficients. Last, we extend our framework to be controlled by additional input conditions to enable it to generate shapes from assorted modalities, e.g., single/multi-view images, point clouds, and low-resolution voxels. In our extensive set of experiments, we demonstrate various applications, such as unconditional generation, shape completion, and conditional generation on a wide range of modalities. Our approach not only surpasses the state of the art in delivering high-quality results but also efficiently generates shapes within a few seconds, often achieving this in just 2 seconds for most conditions. Our source code is available at https://github.com/AutodeskAILab/Make-a-Shape.
title Make-A-Shape: a Ten-Million-scale 3D Shape Model
topic Computer Vision and Pattern Recognition
Graphics
url https://arxiv.org/abs/2401.11067