arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2408.06145cs.CV
收藏

基于稀疏点-体素扩散模型的高效可扩展点云生成

Efficient and Scalable Point Cloud Generation with Sparse Point-Voxel Diffusion Models

University of Patras(帕特雷大学);Industrial Systems Institute (ISI) - Athena Research Center(雅典娜研究中心工业系统研究所);Vrije Universiteit Brussel(布鲁塞尔自由大学)

Ioannis Romanelis, Vlassios Fotis, Athanasios Kalogeras, Christos Alexakos, Konstantinos Moustakas, Adrian Munteanu

AI总结:

本研究提出结合点高分辨率表征与稀疏体素效率的双分支点云U-Net扩散架构SPVD,在多类点云生成及相关任务中取得SOTA结果,生成速度更快且具备可扩展性。

摘要:

我们提出一种用于3D生成建模的新型点云U-Net扩散架构,能够生成高质量、多样化的3D形状,同时保持较快的生成速度。该网络采用双分支架构,结合了点的高分辨率表征与稀疏体素的计算效率。我们最快的模型变体在无条件形状生成(评估点云生成模型最常用的基准测试)任务上优于所有非扩散生成方法;而我们最大的模型在扩散方法中取得了最优结果,运行时间约为此前最优模型PVD的70%。除无条件生成外,我们还开展了广泛评估,包括在ShapeNet所有类别上的条件生成,验证了模型面向更大数据集的可扩展性;以及隐式生成,可让我们的网络在更少时间步下生成高质量点云,进一步缩短生成时间。最后,我们评估了该架构在点云补全和超分辨率任务中的性能。我们的模型在所有任务中表现优异,确立了其作为点云生成建模领域最优扩散U-Net的地位。代码已公开于https://github.com/JohnRomanelis/SPVD.git。

英文摘要:

We propose a novel point cloud U-Net diffusion architecture for 3D generative modeling capable of generating high-quality and diverse 3D shapes while maintaining fast generation times. Our network employs a dual-branch architecture, combining the high-resolution representations of points with the computational efficiency of sparse voxels. Our fastest variant outperforms all non-diffusion generative approaches on unconditional shape generation, the most popular benchmark for evaluating point cloud generative models, while our largest model achieves state-of-the-art results among diffusion methods, with a runtime approximately 70% of the previously state-of-the-art PVD. Beyond unconditional generation, we perform extensive evaluations, including conditional generation on all categories of ShapeNet, demonstrating the scalability of our model to larger datasets, and implicit generation which allows our network to produce high quality point clouds on fewer timesteps, further decreasing the generation time. Finally, we evaluate the architecture's performance in point cloud completion and super-resolution. Our model excels in all tasks, establishing it as a state-of-the-art diffusion U-Net for point cloud generative modeling. The code is publicly available at https://github.com/JohnRomanelis/SPVD.git.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑