ProlificDreamer:基于变分分数蒸馏的高保真、多样化文本到 3D 生成

ProlificDreamer: High-Fidelity and Diverse Text-to-3D Generation with Variational Score Distillation

朱军 Jun Zhu · Tsinghua University · 2023-05-25 · arXiv:2305.16213 ↗ · 被引 1387

打开互动全文版(逐段中英对照 + 图/公式 + 论文问答)→

摘要 · Abstract

分数蒸馏采样(SDS)通过蒸馏预训练的大规模文本到图像扩散模型,在文本到 3D 生成中显示出巨大潜力,但存在过饱和、过平滑和低多样性问题。在这项工作中,我们提出将 3D 参数建模为随机变量而非 SDS 中的常数,并提出了变分分数蒸馏(VSD),这是一个基于粒子的原则性变分框架,用于解释和解决文本到 3D 生成中的上述问题。我们证明 SDS 是 VSD 的一个特例,并且在 CFG 权重较小和较大时都会导致较差的样本。相比之下,VSD 能够像扩散模型中的祖先采样一样,在各种 CFG 权重下良好工作,并在使用常见 CFG 权重(如 7.5)时同时提高多样性和样本质量。我们进一步提出了文本到 3D 设计空间中的各种改进,例如蒸馏时间表和密度初始化,这些与蒸馏算法正交但尚未得到充分探索。我们的整体方法称为 ProlificDreamer,可以生成高渲染分辨率(即 512×512)和高保真度的 NeRF,具有丰富的结构和复杂效果(如烟雾和液滴)。此外,从 NeRF 初始化后,通过 VSD 微调的网格具有精细的细节和照片级真实感。项目页面和代码:https://ml.cs.tsinghua.edu.cn/prolificdreamer/

Score distillation sampling (SDS) has shown great promise in text-to-3D generation by distilling pretrained large-scale text-to-image diffusion models, but suffers from over-saturation, over-smoothing, and low-diversity problems. In this work, we propose to model the 3D parameter as a random variable instead of a constant as in SDS and present variational score distillation (VSD), a principled particle-based variational framework to explain and address the aforementioned issues in text-to-3D generation. We show that SDS is a special case of VSD and leads to poor samples with both small and large CFG weights. In comparison, VSD works well with various CFG weights as ancestral sampling from diffusion models and simultaneously improves the diversity and sample quality with a common CFG weight (i.e., $7.5$). We further present various improvements in the design space for text-to-3D such as distillation time schedule and density initialization, which are orthogonal to the distillation algorithm yet not well explored. Our overall approach, dubbed ProlificDreamer, can generate high rendering resolution (i.e., $512\times512$) and high-fidelity NeRF with rich structure and complex effects (e.g., smoke and drops). Further, initialized from NeRF, meshes fine-tuned by VSD are meticulously detailed and photo-realistic. Project page and codes: https://ml.cs.tsinghua.edu.cn/prolificdreamer/

核心贡献 · Key contributions

局限 · Limitations

论文章节 · Sections(共 16)

阅读逐段中英对照全文 →