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

Consistent3D:利用确定性采样先验实现一致的高保真文本到3D生成

Consistent3D: Towards Consistent High-Fidelity Text-to-3D Generation with Deterministic Sampling Prior

Nanyang Technological University(南洋理工大学);Singapore Management University(新加坡管理大学);Johns Hopkins University(约翰斯·霍普金斯大学);Sea AI Lab(Sea AI实验室);Skywork AI

Zike Wu, Pan Zhou, Xuanyu Yi, Xiaoding Yuan, Hanwang Zhang

AI总结:

本文针对SDS文本到3D生成中随机采样导致几何坍缩和纹理差的问题,提出Consistent3D,利用ODE确定性轨迹与一致性蒸馏损失实现稳定高保真3D生成。

摘要:

Score distillation sampling(SDS,分数蒸馏采样)及其变体极大推动了文本到3D生成的发展,但仍容易受到几何坍缩和纹理较差的影响。为解决该问题,我们首先深入分析SDS,发现其蒸馏采样过程实际上对应于随机微分方程(SDE)的轨迹采样:SDS沿SDE轨迹进行采样,以产生一个噪声较小的样本,随后该样本作为优化3D模型的引导。然而,SDE采样中的随机性通常会产生多样且不可预测的样本,这些样本并不总是噪声更小,因此并非始终正确的引导,这解释了SDS的脆弱性。由于对于任意SDE,总存在一个常微分方程(ODE),其轨迹采样能够像SDE一样确定性且一致地收敛到期望目标点,我们提出了一种新颖有效的“Consistent3D”方法,该方法探索用于文本到3D生成的ODE确定性采样先验。具体而言,在每次训练迭代中,给定由3D模型渲染的图像,我们首先通过预训练的2D扩散模型估计其期望的3D分数函数,并构建用于轨迹采样的ODE。接下来,我们设计了一致性蒸馏采样损失,该损失沿ODE轨迹采样以生成两个相邻样本,并使用噪声较小的样本引导另一个噪声更大的样本,从而将确定性先验蒸馏到3D模型中。实验结果表明,我们的Consistent3D在生成高保真且多样化的3D对象和大规模场景方面具有有效性,如图1所示。代码可在https://github.com/sail-sg/Consistent3D获取。

英文摘要:

Score distillation sampling (SDS) and its variants have greatly boosted the development of text-to-3D generation, but are vulnerable to geometry collapse and poor textures yet. To solve this issue, we first deeply analyze the SDS and find that its distillation sampling process indeed corresponds to the trajectory sampling of a stochastic differential equation (SDE): SDS samples along an SDE trajectory to yield a less noisy sample which then serves as a guidance to optimize a 3D model. However, the randomness in SDE sampling often leads to a diverse and unpredictable sample which is not always less noisy, and thus is not a consistently correct guidance, explaining the vulnerability of SDS. Since for any SDE, there always exists an ordinary differential equation (ODE) whose trajectory sampling can deterministically and consistently converge to the desired target point as the SDE, we propose a novel and effective "Consistent3D" method that explores the ODE deterministic sampling prior for text-to-3D generation. Specifically, at each training iteration, given a rendered image by a 3D model, we first estimate its desired 3D score function by a pre-trained 2D diffusion model, and build an ODE for trajectory sampling. Next, we design a consistency distillation sampling loss which samples along the ODE trajectory to generate two adjacent samples and uses the less noisy sample to guide another more noisy one for distilling the deterministic prior into the 3D model. Experimental results show the efficacy of our Consistent3D in generating high-fidelity and diverse 3D objects and large-scale scenes, as shown in Fig. 1. The codes are available at https://github.com/sail-sg/Consistent3D.

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

↑