清華朱軍團隊提出ProlificDreamer：直接文本生成高質(zhì)量3D內(nèi)容

清華大學(xué) TSAIL 團隊最新提出的文生 3D 新算法 ProlificDreamer，在無需任何 3D 數(shù)據(jù)的前提下能夠生成超高質(zhì)量的 3D 內(nèi)容。

ProlificDreamer 算法為文生 3D 領(lǐng)域帶來重大進展。利用 ProlificDreamer，輸入文本 “一個菠蘿”，就能生成非常逼真且高清的 3D 菠蘿：

給出稍微難一些的文本，比如 “一只米開朗琪羅風(fēng)格狗的雕塑，正在用手機讀新聞”，ProlificDreamer 的生成也不在話下：

將 Imagen 生成的照片（下圖靜態(tài)圖）和 ProlificDreamer（基于 Stable-Diffusion）生成的 3D（下圖動態(tài)圖）進行對比。有網(wǎng)友感慨：短短一年時間，高質(zhì)量的生成已經(jīng)能夠從 2D 圖像領(lǐng)域擴展到 3D 領(lǐng)域了！

A blue jay standing on alarge basket of rainbow macarons 這一切都來源于清華大學(xué)計算機系朱軍教授帶領(lǐng)的 TSAIL 團隊近期公開的一篇論文《ProlificDreamer: High-Fidelity and Diverse Text-to-3D Generation with Variational Score Distillation》：

論文鏈接：https://arxiv.org/abs/2305.16213

項目主頁：https://ml.cs.tsinghua.edu.cn/prolificdreamer/

在數(shù)字創(chuàng)作和虛擬現(xiàn)實等領(lǐng)域，從文本到三維模型（Text-to-3D）的技術(shù)具有重要的價值和廣泛的應(yīng)用潛力。這種技術(shù)可以從簡單的文本描述中生成具體的 3D 模型，為設(shè)計師、游戲開發(fā)者和數(shù)字藝術(shù)家提供強大的工具。 然而，為了根據(jù)文本生成準(zhǔn)確的 3D 模型，傳統(tǒng)方法需要大量的標(biāo)記 3D 模型數(shù)據(jù)集。這些數(shù)據(jù)集需要包含多種不同類型和風(fēng)格的 3D 模型，并且每個模型都需要與相應(yīng)的文本描述相關(guān)聯(lián)。創(chuàng)建這樣的數(shù)據(jù)集需要大量的時間和人力資源，目前還沒有現(xiàn)成的大規(guī)模數(shù)據(jù)集可供使用。 由谷歌提出的 DreamFusion [1] 利用預(yù)訓(xùn)練的 2D 文本到圖像擴散模型，首次在無需 3D 數(shù)據(jù)的情況下完成開放域的文本到 3D 的合成。但是 DreamFusion 提出的 Score Distillation Sampling (SDS) [1] 算法生成結(jié)果面臨嚴重的過飽和、過平滑、缺少細節(jié)等問題。高質(zhì)量 3D 內(nèi)容生成目前仍然是非常困難的前沿問題之一。 ProlificDreamer 論文提出了 Variational Score Distillation（VSD）算法，從貝葉斯建模和變分推斷（variational inference）的角度重新形式化了 text-to-3D 問題。具體而言，VSD 把 3D 參數(shù)建模為一個概率分布，并優(yōu)化其渲染的二維圖片的分布和預(yù)訓(xùn)練 2D 擴散模型的分布間的距離?？梢宰C明，VSD 算法中的 3D 參數(shù)近似了從 3D 分布中采樣的過程，解決了 DreamFusion 所提 SDS 算法的過飽和、過平滑、缺少多樣性等問題。此外，SDS 往往需要很大的監(jiān)督權(quán)重（CFG=100），而 VSD 是首個可以用正常 CFG（=7.5）的算法。效果展示ProlificDreamer 可以根據(jù)文本生成非常高質(zhì)量的帶紋理的三維網(wǎng)格：

ProlificDreamer 可以根據(jù)文本生成非常高質(zhì)量的三維神經(jīng)輻射場（NeRF），包括復(fù)雜的效果。甚至 360° 的場景也能生成：

ProlificDreamer 還可以在給出同樣文本的情況下生成具有多樣性的 3D 內(nèi)容：

傳統(tǒng)文生 3D 的優(yōu)化算法給定一個 2D 圖片上預(yù)訓(xùn)練好的擴散模型（例如 stable-diffusion），Dreamfusion [1] 提出可以在不借助任何 3D 數(shù)據(jù)的情況下實現(xiàn)開放域的文到 3D 內(nèi)容（text-to-3D）生成。具體而言，對于一個 3D 物體，文生 3D 任務(wù)的關(guān)鍵是設(shè)計一種優(yōu)化算法，使得 3D 物體在各個視角下投影出來的 2D 圖片與預(yù)訓(xùn)練的 2D 擴散模型匹配，并不斷優(yōu)化 3D 物體。其中，SDS [1] （也稱為 Score Jacobian Chaining (SJC) [3]） 是目前幾乎所有的零樣本開放域文生 3D 工作所使用的算法。該算法將 3D 物體視為一個單點（single point），并通過隨機梯度下降優(yōu)化該 3D 物體，優(yōu)化目標(biāo)是最大化該渲染的 2D 圖像在預(yù)訓(xùn)練擴散模型下的似然值。值得注意的是，該優(yōu)化問題的最優(yōu)解并不等價于從擴散模型中采樣。

傳統(tǒng)文生 3D 的優(yōu)化算法示意圖 實驗中，所有基于 SDS/SJC 的方法目前都有一個嚴重的問題：生成的物體過于平滑、過飽和現(xiàn)象嚴重，并且多樣性不高。例如，開源庫 threestudio [4] 將目前主流的 text-to-3D 工作復(fù)現(xiàn)至與原論文可比水平，如下圖所示：

由 threestuidio [4] 復(fù)現(xiàn)的文生 3D 工作 在此之前，基于 2D 擴散模型的文生 3D 仍然與實踐落地有較大差距。然而，清華大學(xué)朱軍團隊提出的 ProlificDreamer 在算法層面解決了 SDS 的上述問題，能夠生成非常逼真的 3D 內(nèi)容，極大地縮小了這一差距。ProlificDreamer 的原理與以往方法不同，ProlificDreamer 并不單純優(yōu)化單個 3D 物體，而是優(yōu)化 3D 物體對應(yīng)的概率分布。通常而言，給定一個有效的文本輸入，存在一個概率分布包含了該文本描述下所有可能的 3D 物體。

給定文本下的 3D 物體存在一個潛在的概率分布 基于該 3D 概率分布，我們可以進一步誘導(dǎo)出一個 2D 概率分布。具體而言，只需要對每一個 3D 物體經(jīng)過相機渲染到 2D，即可得到一個 2D 圖像的概率分布。

由潛在 3D 分布可以誘導(dǎo)出一個 2D 圖像分布 因此，優(yōu)化 3D 分布可以被等效地轉(zhuǎn)換為優(yōu)化 2D 渲染圖片的概率分布與 2D 擴散模型定義的概率分布之間的距離（由 KL 散度定義）。這是一個經(jīng)典的變分推斷（variational inference）任務(wù)，因此 ProlificDreamer 文中將該任務(wù)及對應(yīng)的算法稱為變分得分蒸餾（Variational Score Distillation，VSD）。

優(yōu)化 3D 分布可以被等效地轉(zhuǎn)換為優(yōu)化 2D 圖片之間的概率分布 具體而言，VSD 的算法流程圖如下所示。其中，3D 物體的迭代更新需要使用兩個模型：一個是預(yù)訓(xùn)練的 2D 擴散模型（例如 Stable-Diffusion），另一個是基于該預(yù)訓(xùn)練模型的 LoRA（low-rank adaptation）。該 LoRA 估計了當(dāng)前 3D 物體誘導(dǎo)的 2D 圖片分布的得分函數(shù)（score function），并進一步用于更新 3D 物體。該算法實際上在模擬 Wasserstein 梯度流，并可以保證收斂得到的分布滿足與預(yù)訓(xùn)練的 2D 擴散模型的 KL 散度最小。

VSD 的訓(xùn)練流程圖 與傳統(tǒng)的 SDS/SJC 算法相比，可以發(fā)現(xiàn) VSD 僅僅需要把原來的高斯噪聲項換成 LoRA 項即可。由于LoRA 提供了比高斯噪聲更精細的更新方向（例如，LoRA 可以利用文本 y、相機視角 c、擴散時間 t 等的先驗信息），VSD 在實踐中可以得到遠超 SDS 的精細結(jié)果。并且，論文作者提出，SDS/SJC 實際上是 VSD 使用一個單點 Dirac 分布作為變分分布的特例，而 VSD 擴展到了由 LoRA 定義的更復(fù)雜的概率分布，因此可以得到更好的結(jié)果。此外，VSD 還對監(jiān)督權(quán)重（CFG）更友好，可以使用與 2D 擴散模型一樣的監(jiān)督權(quán)重（例如 stable-diffusion 常用的 CFG=7.5），因此可以達到和 2D 擴散模型類似的采樣質(zhì)量。這一結(jié)果首次解決了 SDS/SJC 中的超大 CFG（一般為 100）的問題，也同時說明 VSD 這種基于分布優(yōu)化的思想與預(yù)訓(xùn)練的 2D 擴散模型更適配。

SDS/SJC 與 VSD 的更新公式對比 最后，ProlificDreamer 還對 3D 表示的設(shè)計空間做了詳細的研究，提出了如下實現(xiàn)。在實踐中，VSD 可以在 512 渲染分辨率的 NeRF 下訓(xùn)練，并極大地豐富了所得到的 3D 結(jié)果的紋理細節(jié)。

ProlificDreamer 與其它工作的實現(xiàn)細節(jié)比較

審核編輯 ：李倩