arXiv:2605.10830cs.CVcs.LG2026-05被引 1

用扩散模型推理3D结构不确定性,从单图到稀疏数据都能重建。

Predicting 3D structure by latent posterior sampling

论文配图:Predicting 3D structure by latent posterior sampling
图 1 · 摘自论文原文
  • 将3D场景建为随机潜变量,结合扩散模型做后验推断。
  • 支持单视图、多视图、噪声图等多样输入,重建精度高。
  • 适合处理观测不完整时的3D结构预测,尤其在信息稀疏场景。

2D生成模型与3D神经场表示的进展为融合二者优势提供了契机。本文提出一种方法,将基于NeRF的3D场景表示与扩散模型的概率建模相结合,将3D重建视为存在固有不确定性的感知问题,可受益于概率推断。核心思想是将3D场景表示为一个随机潜变量,学习其先验,并基于观测进行后验推断。通过结合扩散模型的基于得分的推断方法与基于体素渲染的重建模型计算似然项,实现后验采样。采用两阶段训练:首先在3D场景数据集上训练重建模型并自动解码潜变量;随后使用扩散模型训练潜变量的先验。通过从后验中生成样本,展示了该方法可完成多种3D重建任务,输入包括单视图、多视图、噪声图像、稀疏像素和稀疏深度数据。不同观测提供的信息量不同,本文方法能有效建模各任务对应的不确定性水平。实验表明,该方法可准确从多样化观测中预测3D结构。

原文摘要 · Abstract (English)

The remarkable achievements of both generative models of 2D images and neural field representations for 3D scenes present a compelling opportunity to integrate the strengths of both approaches. In this work, we propose a methodology that combines a NeRF-based representation of 3D scenes with probabilistic modeling and reasoning using diffusion models. We view 3D reconstruction as a perception problem with inherent uncertainty that can thereby benefit from probabilistic inference methods. The core idea is to represent the 3D scene as a stochastic latent variable for which we can learn a prior and use it to perform posterior inference given a set of observations. We formulate posterior sampling using the score-based inference method of diffusion models in conjunction with a likelihood term computed from a reconstruction model that includes volumetric rendering. We train the model using a two-stage process: first we train the reconstruction model while auto-decoding the latent representations for a dataset of 3D scenes, and then we train the prior over the latents using a diffusion model. By using the model to generate samples from the posterior we demonstrate that various 3D reconstruction tasks can be performed, differing by the type of observation used as inputs. We showcase reconstruction from single-view, multi-view, noisy images, sparse pixels, and sparse depth data. These observations vary in the amount of information they provide for the scene and we show that our method can model the varying levels of inherent uncertainty associated with each task. Our experiments illustrate that this approach yields a comprehensive method capable of accurately predicting 3D structure from diverse types of observations.

3D重建扩散模型概率推断

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