arXiv:2504.09975cs.GRcs.CV2025-04International Conf…被引 33

用八叉树结构加速3D形状生成,性能媲美扩散模型。

OctGPT: Octree-based Multiscale Autoregressive Models for 3D Shape Generation

  • 用八叉树序列化3D结构,结合二值化编码生成多尺度二进制序列。
  • 在4张4090显卡上仅用数天即可训练1024³分辨率模型,速度提升69倍。
  • 支持文本、草图、图像等多模态生成,适合高精度3D内容创作。

自回归模型在多个领域取得显著成功,但在3D形状生成方面表现远逊于扩散模型。本文提出OctGPT,一种基于八叉树的多尺度自回归3D生成模型,显著提升先前方法的效率与性能,可媲美或超越最先进的扩散模型。该方法采用序列化八叉树表示,高效捕捉3D形状的层级与空间结构:粗粒度几何由八叉树编码,细粒度细节通过向量量化变分自编码器(VQVAE)生成二值令牌,将3D形状转化为紧凑的多尺度二进制序列以供自回归预测。为应对长序列计算挑战,引入基于八叉树的Transformer,结合3D旋转位置编码、尺度特定嵌入及令牌并行生成策略。该设计使训练时间减少13倍,生成时间减少69倍,可在仅四张NVIDIA 4090 GPU上数日内完成1024³高分辨率3D形状的训练。OctGPT在文本、草图、图像条件生成以及多物体场景合成等任务中表现出卓越泛化能力。大量实验表明,其收敛速度更快、生成质量更高,为高质量、可扩展的3D内容生成提供新范式。代码与训练模型已开源于https://github.com/octree-nn/octgpt。

原文摘要 · Abstract (English)

Autoregressive models have achieved remarkable success across various domains, yet their performance in 3D shape generation lags significantly behind that of diffusion models. In this paper, we introduce OctGPT, a novel multiscale autoregressive model for 3D shape generation that dramatically improves the efficiency and performance of prior 3D autoregressive approaches, while rivaling or surpassing state-of-the-art diffusion models. Our method employs a serialized octree representation to efficiently capture the hierarchical and spatial structures of 3D shapes. Coarse geometry is encoded via octree structures, while fine-grained details are represented by binary tokens generated using a vector quantized variational autoencoder (VQVAE), transforming 3D shapes into compact multiscale binary sequences suitable for autoregressive prediction. To address the computational challenges of handling long sequences, we incorporate octree-based transformers enhanced with 3D rotary positional encodings, scale-specific embeddings, and token-parallel generation schemes. These innovations reduce training time by 13 folds and generation time by 69 folds, enabling the efficient training of high-resolution 3D shapes, e.g.,$1024^3$, on just four NVIDIA 4090 GPUs only within days. OctGPT showcases exceptional versatility across various tasks, including text-, sketch-, and image-conditioned generation, as well as scene-level synthesis involving multiple objects. Extensive experiments demonstrate that OctGPT accelerates convergence and improves generation quality over prior autoregressive methods, offering a new paradigm for high-quality, scalable 3D content creation. Our code and trained models are available at https://github.com/octree-nn/octgpt.

3D生成八叉树自回归高效建模

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。