arXiv:2505.17412cs.CV2025-05NeurIPS被引 101

用稀疏注意力实现千级分辨率3D生成,8卡即可完成以往需32卡的任务

Direct3D-S2: Gigascale 3D Generation Made Easy with Spatial Sparse Attention

论文配图:Direct3D-S2: Gigascale 3D Generation Made Easy with Spatial Sparse Attention
图 1 · 摘自论文原文
  • 提出空间稀疏注意力机制,高效处理稀疏体素数据
  • 前向/反向计算分别提速3.9倍和9.6倍,1024分辨率仅需8张卡
  • 统一稀疏体素格式,提升训练稳定性和效率,适合大规模3D生成

使用体素表示(如有符号距离函数)生成高分辨率3D形状面临巨大的计算与内存挑战。我们提出Direct3D-S2,一种基于稀疏体素的可扩展3D生成框架,在显著降低训练成本的同时实现了更优的输出质量。核心创新是空间稀疏注意力(SSA)机制,大幅提升了扩散变换器(DiT)在稀疏体素数据上的计算效率。SSA使模型能有效处理大规模稀疏体素中的令牌,显著减少计算开销,前向传播加速3.9倍,反向传播加速9.6倍。框架还包含一个变分自编码器(VAE),在输入、潜在和输出阶段保持一致的稀疏体素格式。相比以往采用异构表示的3D VAE,该统一设计极大提升了训练效率与稳定性。模型在公开数据集上训练,实验表明,Direct3D-S2不仅在生成质量与效率上超越现有方法,还能仅用8张GPU实现1024分辨率的训练,而此前体积表示在256分辨率下通常至少需要32张卡,真正让千级规模3D生成变得可行且易用。

原文摘要 · Abstract (English)

Generating high-resolution 3D shapes using volumetric representations such as Signed Distance Functions (SDFs) presents substantial computational and memory challenges. We introduce Direct3D-S2, a scalable 3D generation framework based on sparse volumes that achieves superior output quality with dramatically reduced training costs. Our key innovation is the Spatial Sparse Attention (SSA) mechanism, which greatly enhances the efficiency of Diffusion Transformer (DiT) computations on sparse volumetric data. SSA allows the model to effectively process large token sets within sparse volumes, substantially reducing computational overhead and achieving a 3.9x speedup in the forward pass and a 9.6x speedup in the backward pass. Our framework also includes a variational autoencoder (VAE) that maintains a consistent sparse volumetric format across input, latent, and output stages. Compared to previous methods with heterogeneous representations in 3D VAE, this unified design significantly improves training efficiency and stability. Our model is trained on public available datasets, and experiments demonstrate that Direct3D-S2 not only surpasses state-of-the-art methods in generation quality and efficiency, but also enables training at 1024 resolution using only 8 GPUs, a task typically requiring at least 32 GPUs for volumetric representations at 256 resolution, thus making gigascale 3D generation both practical and accessible. Project page: https://www.neural4d.com/research/direct3d-s2.

3D生成稀疏注意力扩散模型高效训练

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