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

场景尺度3D分类数据的扩散概率模型

Diffusion Probabilistic Models for Scene-Scale 3D Categorical Data

Korea Advanced Institute of Science and Technology (KAIST)(韩国科学技术院)

Jumin Lee, Woobin Im, Sebin Lee, Sung-Eui Yoon

摘要:

在本文中,我们学习了一个扩散模型来生成场景尺度的3D数据。具体来说,我们的模型工艺品由多个对象组成的3D场景,而最近的扩散研究集中在一个单一的对象。为了实现我们的目标,我们用离散的类标签表示场景,即,分类分布,将多个对象分配到语义类别中。因此,我们扩展离散扩散模型来学习场景尺度的分类分布。此外,我们验证了一个潜空间扩散模型,可以减少训练和部署的计算成本。据我们所知,我们的工作是第一个应用离散和潜在扩散的三维分类数据的场景规模。我们进一步建议通过使用我们的扩散模型学习条件分布来执行语义场景完成(SSC),其中条件是稀疏点云中的部分观察。在实验中,我们的经验表明,我们的扩散模型不仅产生合理的场景,但也执行场景完成任务比判别模型。我们的代码和模型可在https://github.com/zoomin-lee/scene-scale-diffusion上获得

英文摘要:

In this paper, we learn a diffusion model to generate 3D data on a scene-scale. Specifically, our model crafts a 3D scene consisting of multiple objects, while recent diffusion research has focused on a single object. To realize our goal, we represent a scene with discrete class labels, i.e., categorical distribution, to assign multiple objects into semantic categories. Thus, we extend discrete diffusion models to learn scene-scale categorical distributions. In addition, we validate that a latent diffusion model can reduce computation costs for training and deploying. To the best of our knowledge, our work is the first to apply discrete and latent diffusion for 3D categorical data on a scene-scale. We further propose to perform semantic scene completion (SSC) by learning a conditional distribution using our diffusion model, where the condition is a partial observation in a sparse point cloud. In experiments, we empirically show that our diffusion models not only generate reasonable scenes, but also perform the scene completion task better than a discriminative model. Our code and models are available at https://github.com/zoomin-lee/scene-scale-diffusion

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

↑