通过弥合Text-to-2D和Text-to-3D生成领域之间的差距来创建高质量的3D资产

2023-12-31 12:54 390 阅读 ID:1755
将门
将门

本篇论文提出了X-Dreamer,这是一种用于高质量text-to-3D内容创建的新颖方法,可以有效地弥合text-to-2D和text-to-3D生成之间的领域差距。X-Dreamer的关键组成部分是两种创新设计: Camera-Guided Low-Rank Adaptation (CG-LoRA) 和Attention-Mask Alignment (AMA) 损失。

论文标题:  

X-Dreamer: Creating High-quality 3D Content by Bridging the Domain Gap Between Text-to-2D and Text-to-3D Generation  

论文地址: 

https://arxiv.org/abs/2312.00085[1]  

Github主页:  

https://github.com/xmu-xiaoma666/X-Dreamer[2]  

项目主页:  

https://xmu-xiaoma666.github.io/Projects/X-Dreamer/[3]

一、Introduction

近年来,在预训练的扩散模型[1, 2, 3]的开发推动下,自动text-to-3D内容创建取得了重大进展。其中,DreamFusion [4] 引入了一种有效的方法,该方法利用预训练的2D扩散模型 [5] 从文本中自动生成3D资产,从而无需专门的3D资产数据集。DreamFusion引入的一项关键创新是分数蒸馏采样 (SDS) 算法。该算法利用预训练的2D扩散模型对单个3D表示进行评估,例如NeRF [6],从而对其进行优化,以确保来自任何摄像机视角的渲染图像与给定文本保持较高的一致性。受开创性SDS算法的启发,出现了几项工作 [7,8,9,10,11],通过应用预训练的2D扩散模型来推进text-to-3D生成任务。

虽然text-to-3D的生成通过利用预训练的text-to-2D的扩散模型已经取得了重大进展,但是2D图像和3D资产之间仍存在很大的领域差距。这种区别在图1中清楚地展示出来。首先,text-to-2D模型产生与相机无关的生成结果,专注于从特定角度生成高质量图像,而忽略其他角度。相比之下,3D内容创建与相机参数 (如位置、拍摄角度和视场) 错综复杂地联系在一起。

因此,text-to-3D模型必须在所有可能的相机参数上生成高质量的结果。此外,text-to-2D生成模型必须同时生成前景和背景元素,同时保持图像的整体连贯性。相反,text-to-3D生成模型只需要集中在创建前景对象上。这种区别允许text-to-3D模型分配更多的资源和注意力来精确地表示和生成前景对象。因此,当直接采用预训练的2D扩散模型进行3D资产创建时,text-to-2D和text-to-3D生成之间的域差距构成了显著的性能障碍。

图1 text-to-2D生成模型 (左) 和text-to-3D生成模型 (右) 在同一文本提示下的输出,即

为了解决这个问题,论文提出了X-Dreamer,这是一种用于高质量text-to-3D内容创建的新颖方法,可以有效地弥合text-to-2D和text-to-3D生成之间的领域差距。X-Dreamer的关键组成部分是两种创新设计: Camera-Guided Low-Rank Adaptation (CG-LoRA) 和Attention-Mask Alignment (AMA) 损失。

首先,现有方法 [7,8,9,10] 通常采用2D预训练扩散模型 [5,12] 来进行text-to-3D生成,缺乏与相机参数的固有联系。为了解决此限制并确保X-Dreamer产生直接受相机参数影响的结果,论文引入了CG-LoRA来调整预训练的2D扩散模型。值得注意的是,在每次迭代期间CG-LoRA的参数都是基于相机信息动态生成的,从而在text-to-3D模型和相机参数之间建立鲁棒的关系。其次,预训练的text-to-2D扩散模型将注意力分配给前景和背景生成,而3D资产的创建需要更加关注前景对象的准确生成。

为了解决这一问题,论文提出了AMA损失,使用3D对象的二进制掩码来指导预训练的扩散模型的注意力图,从而优先考虑前景对象的创建。通过合并该模块,X-Dreamer优先考虑前景对象的生成,从而显着提高了生成的3D内容的整体质量。

X-Dreamer对text-to-3D生成领域做出了如下贡献:

  • 论文提出了一种新颖的方法,X-Dreamer,用于高质量的text-to-3D内容创建,有效地弥合了text-to-2D和text-to-3D生成之间的主要差距。
  • 为了增强生成的结果与相机视角之间的对齐,论文提出了CG-LoRA,利用相机信息来动态生成2D扩散模型的特定参数。
  • 为了在text-to-3D模型中优先创建前景对象,论文引入了AMA损失,利用前景3D对象的二进制掩码来引导2D扩散模型的注意图。

二、Approach

X-Dreamer包括两个主要阶段: 几何学习和外观学习。对于几何学习,论文采用DMTET作为3D表示,并利用3D椭球对其进行初始化,初始化时的损失函数采用均方误差 (MSE) 损失。随后,论文使用分数蒸馏采样 (SDS) 损失和论文提出的AMA损失来优化DMTET和CG-LoRA,以确保3D表示和输入文本提示之间的对齐。对于外观学习,论文利用双向反射分布函数 (BRDF) 建模。

具体来说,论文利用具有可训练参数的MLP来预测表面材料。类似于几何学习阶段,论文使用SDS损失和AMA损失来优化MLP和CG-LoRA的可训练参数,以实现3D表示和文本提示之间的对齐。图2展示了X-Dreamer的详细构成。

                                                   图2 X-Dreamer概览,包括几何学习和外观学习。

Geometry Learning(几何学习)

Appearance Learning(外观学习)

Camera-Guided Low-Rank Adaptation(CG-LoRA)

为了解决text-to-2D和text-to-3D的生成任务之间存在的领域差距而导致的次优的3D结果的生成,X-Dreamer提出了Camera-Guided Low-Rank Adaptation。如图3所示,利用摄像机参数和方向感知文本来指导CG-LoRA中参数的生成,使X-Dreamer能够有效地感知摄像机的位置和方向信息。

                                                                图3 摄像机引导的CG-LoRA示意

Attention-Mask Alignment Loss(AMA Loss)

三、Experiments

论文使用四个Nvidia RTX 3090 GPU和PyTorch库进行实验。为了计算SDS损失,利用了通过Hugging Face Diffusers实现的Stable Diffusion模型。对于DMTET和material编码器,将它们分别实现为两层MLP和单层MLP,隐藏层维度为32。

从椭球体开始进行text-to-3D的生成

论文展示了X-Dreamer利用椭球作为初始几何形状的text-to-3D的生成结果,如图4所示。结果证明X-Dreamer具有生成高质量和照片般逼真的3D对象的能力,生成的3D对象与输入的文本提示准确对应。

                                                          图4 从椭球体开始进行text-to-3D的生成

从粗粒度网格开始进行text-to-3D的生成

虽然可以从互联网上下载大量粗粒度网格,但由于缺乏几何细节,直接使用这些网格创建3D内容往往会导致性能较差。然而,与3D椭球体相比,这些网格可以为X-Dreamer提供更好的3D形状先验信息。因此,也可以使用粗粒度引导网格来初始化DMTET,而不是使用椭球。如图5所示,X-Dreamer可以基于给定的文本生成具有精确几何细节的3D资产,即使所提供的粗粒度网格缺乏细节。

                                                   图5 从粗粒度网格开始进行text-to-3D的生成

定性比较

为了评估X-Dreamer的有效性,论文将其与四种SOTA方法进行比较: DreamFusion [4],Magic3D [8],Fantasia3D [7] 和ProlificDreamer [11],如图6所示。当与基于SDS的方法进行比较时 [4,7,8],X-Dreamer在生成高质量和逼真的3D资产方面优于他们。此外,与基于VSD的方法 [11] 相比,X-Dreamer产生的3D内容具有相当甚至更好的视觉效果,同时需要的优化时间明显减少。具体来说,X-Dreamer的几何形状和外观学习过程只需要大约27分钟,而ProlificDreamer则超过8小时。

                                                               图6 与现有技术(SOTA)方法的比较。

消融实验

模块消融

为了深入了解CG-LoRA和AMA损失的能力,论文进行了消融研究,其中每个模块单独加入以评估其影响。如图7所示,消融结果表明,当CG-LoRA被排除在X-Dreamer之外时,生成的3D对象的几何形状和外观质量显著下降。此外,X-Dreamer缺失AMA损失也对生成的3D资产的几何形状和外观保真度产生有害影响。这些消融实验为CG-LoRA和AMA损失在增强生成的3D对象的几何形状、外观和整体质量方面的单独贡献提供了有价值的研究。

                                                                   图7 X-Dreamer的消融研究

有无AMA损失的注意力图比较

引入AMA损失的目的是将去噪过程中的注意力引导到前景对象。这个是通过将SD的注意力图与3D对象的渲染掩码对齐来实现的。为了评估AMA损失在实现这一目标方面的有效性,论文在几何学习和外观学习阶段可视化了有和没有AMA损失的SD的注意力图。如图8所示,可以观察到,加入AMA损失不仅会改善生成的3D资产的几何形状和外观,而且会将SD的注意力特别集中在前景对象区域上。可视化证实了AMA损失在引导SD注意力方面的有效性,从而在几何和外观学习阶段提高了质量和前景对象的聚焦。

                             图8 注意力图、渲染掩码和渲染图像的可视化,包括和不包括AMA损失。

四、Conclusion

这项研究引入了一个名为X-Dreamer的开创性框架,该框架旨在通过解决text-to-2D和text-to-3D生成之间的领域差距来增强text-to-3D的生成。为了实现这一点,论文首先提出了CG-LoRA,这是一个将3D相关信息(包括方向感知文本和相机参数)合并到预训练的Stable Diffusion(SD)模型中的模块。通过这样做,本文能够有效地捕获与3D领域相关的信息。

此外,本文设计了AMA损失,以将SD生成的注意力图与3D对象的渲染掩码对齐。AMA损失的主要目标是引导text-to-3D模型的焦点朝着前景对象的生成方向发展。通过广泛的实验,本文彻底评估了提出方法的有效性,实验始终证明了X-Dreamer能够根据给定的文本提示生成高质量和真实的3D内容。

References

[1] Jonathan Ho, Ajay Jain, and Pieter Abbeel. Denoising diffusion probabilistic models. Advances in neural information processing systems, 33:6840–6851, 2020.

[2] Jascha Sohl-Dickstein, Eric Weiss, Niru Maheswaranathan, and Surya Ganguli. Deep unsupervised learning using nonequilibrium thermodynamics. In International conference on machine learning, pages 2256–2265. PMLR, 2015.

[3] Yang Song, Jascha Sohl-Dickstein, Diederik P Kingma, Abhishek Kumar, Stefano Ermon, and Ben Poole. Score-based generative modeling through stochastic differential equations. arXiv preprint arXiv:2011.13456, 2020.

[4] Ben Poole, Ajay Jain, Jonathan T Barron, and Ben Mildenhall. Dreamfusion: Text-to-3d using 2d diffusion. arXiv preprint arXiv:2209.14988, 2022.

[5] Chitwan Saharia, William Chan, Saurabh Saxena, Lala Li, Jay Whang, Emily L Denton, Kamyar Ghasemipour, Raphael Gontijo Lopes, Burcu Karagol Ayan, Tim Salimans, et al. Photorealistic text-to-image diffusion models with deep language understanding. Advances in Neural Information Processing Systems, 35:36479–36494, 2022.

[6] Ben Mildenhall, Pratul P Srinivasan, Matthew Tancik, Jonathan T Barron, Ravi Ramamoorthi, and Ren Ng. Nerf: Representing scenes as neural radiance fields for view synthesis. Communications of the ACM, 65(1):99–106, 2021.

[7] Rui Chen, Yongwei Chen, Ningxin Jiao, and Kui Jia. Fantasia3d: Disentangling geometry and appearance for high-quality text-to-3d content creation. arXiv preprint arXiv:2303.13873, 2023.

[8] Chen-Hsuan Lin, Jun Gao, Luming Tang, Towaki Takikawa, Xiaohui Zeng, Xun Huang, Karsten Kreis, Sanja Fidler, Ming-Yu Liu, and Tsung-Yi Lin. Magic3d: High-resolution text-to-3d content creation. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pages 300–309, 2023.

[9] Gal Metzer, Elad Richardson, Or Patashnik, Raja Giryes, and Daniel Cohen-Or. Latent-nerf for shape-guided generation of 3d shapes and textures. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pages 12663–12673, 2023.

[10] Haochen Wang, Xiaodan Du, Jiahao Li, Raymond A Yeh, and Greg Shakhnarovich. Score jacobian chaining: Lifting pretrained 2d diffusion models for 3d generation. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pages 12619–12629, 2023.

[11] Zhengyi Wang, Cheng Lu, Yikai Wang, Fan Bao, Chongxuan Li, Hang Su, and Jun Zhu. Prolificdreamer: High-fidelity and diverse text-to-3d generation with variational score distillation. arXiv preprint arXiv:2305.16213, 2023.

[12] Robin Rombach, Andreas Blattmann, Dominik Lorenz, Patrick Esser, and Bjorn Ommer. High-resolution image synthesis with latent diffusion models. In Proceedings of the IEEE/CVF conference on computer vision and pattern recognition, pages 10684–10695, 2022.

Illustration From IconScout By Delesign Graphics

免责声明:作者保留权利,不代表本站立场。如想了解更多和作者有关的信息可以查看页面右侧作者信息卡片。
反馈
to-top--btn