arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2103.01458cs.CV
收藏

三维点云生成的扩散概率模型

Diffusion Probabilistic Models for 3D Point Cloud Generation

Peking University(北京大学);Wangxuan Institute of Computer Technology(王选计算机研究所)

Shitong Luo, Wei Hu

摘要:

我们提出了一个点云生成的概率模型,这是各种三维视觉任务,如形状完成,上采样,合成和数据增强的基础。受非平衡热力学中扩散过程的启发,我们将点云中的点视为与热浴接触的热力学系统中的粒子,这些粒子从原始分布扩散到噪声分布。因此,点云生成相当于学习将噪声分布变换为期望形状的分布的逆扩散过程。具体来说,我们建议模拟点云的反向扩散过程作为马尔可夫链条件下的某些形状的潜在的。我们推导出封闭形式的训练变分界限,并提供该模型的实现。实验结果表明,我们的模型在点云生成和自动编码方面具有竞争力的性能。代码可在\url{https://github.com/luost26/diffusion-point-cloud}上获得。

英文摘要:

We present a probabilistic model for point cloud generation, which is fundamental for various 3D vision tasks such as shape completion, upsampling, synthesis and data augmentation. Inspired by the diffusion process in non-equilibrium thermodynamics, we view points in point clouds as particles in a thermodynamic system in contact with a heat bath, which diffuse from the original distribution to a noise distribution. Point cloud generation thus amounts to learning the reverse diffusion process that transforms the noise distribution to the distribution of a desired shape. Specifically, we propose to model the reverse diffusion process for point clouds as a Markov chain conditioned on certain shape latent. We derive the variational bound in closed form for training and provide implementations of the model. Experimental results demonstrate that our model achieves competitive performance in point cloud generation and auto-encoding. The code is available at \url{https://github.com/luost26/diffusion-point-cloud}.

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

↑