SDXL Turbo與LCM帶來AI畫圖的即時生成時代:速度跟打字一樣快,影像瞬間呈現

PHPz
發布: 2023-11-30 14:14:50
轉載
1396 人瀏覽過

Stability AI在本週二推出了新一代的圖像合成模型——Stable Diffusion XL Turbo,這款模型引起了人們的熱烈反響。許多人表示,使用該模型進行圖像到文本生成變得前所未有的容易

在輸入框中輸入你的想法,SDXL Turbo 將快速響應並生成相應的內容,無需其他操作。無論你輸入的內容增多或減少,都不會影響它的速度

SDXL Turbo和LCM带来AI画图的实时生成时代:速度跟打字一样快,图像瞬间呈现

SDXL Turbo和LCM带来AI画图的实时生成时代:速度跟打字一样快,图像瞬间呈现

你可以利用現有的圖像,更精細地完成創作。只要拿一張白紙,告訴SDXL Turbo 你想要一隻白貓,還沒打完字,小白貓就已經出現在你手中了

SDXL Turbo和LCM带来AI画图的实时生成时代:速度跟打字一样快,图像瞬间呈现

##SDXL Turbo 模型的速度達到了近乎「即時」的程度,讓人不禁開始暢想:影像產生模型是否可以有其他用途呢

有人直接連著遊戲,獲得了2fps 的風格遷移畫面:

SDXL Turbo和LCM带来AI画图的实时生成时代:速度跟打字一样快,图像瞬间呈现

#根據官方部落格介紹,在A100 上,SDXL Turbo 可在207 毫秒內產生512x512 影像(即時編碼單一去噪步驟解碼,fp16),其中單一UNet 前向評估佔用了67 毫秒。

如此,我們可以判斷,文生圖已經進入「即時」時代。

這樣的「即時生成」效率,與前不久爆火的清華 LCM 模型看起來有些相似,但是它們背後的技術內容卻有所不同。 Stability 在同期發布的一篇研究論文中詳細介紹了該模型的內部工作原理。該研究重點提出了一種名為對抗擴散蒸餾(Adversarial Diffusion Distillation,ADD)的技術。 SDXL Turbo 聲稱的優勢之一是它與生成對抗網路(GAN)的相似性,特別是在生成單步影像輸出方面。

SDXL Turbo和LCM带来AI画图的实时生成时代:速度跟打字一样快,图像瞬间呈现

論文地址:https://static1.squarespace.com/static/6213c340453c3f502425776e/t/65663480a92fba51d0e1023f/1701197769659/adversarial_diffusion_distillation.pdf

論文細節

簡單來說,對抗擴散蒸餾是一種通用方法,可將預訓練擴散模型的推理步驟數量減少到1- 4 個取樣步,同時保持高取樣保真度,並有可能進一步提高模型的整體效能。 

為此,研究者引入了兩個訓練目標的組合:(i)對抗損失和(ii)與 SDS 相對應的蒸餾損失。對抗損失迫使模型在每次前向傳遞時直接生成位於真實影像流形上的樣本,避免了其他蒸餾方法中常見的模糊和其他偽影。蒸餾損失使用另一個預先訓練(且固定)的 擴散模型作為教師,有效利用其廣泛知識,並保留在大型擴散模型中觀察到的強組合性。在推理過程中,研究者未使用無分類器指導,進一步減少了記憶體需求。他們保留了模型透過迭代細化來改進結果的能力,這比先前基於 GAN 的單步方法更具優勢。

訓練步驟如圖2 所示:

SDXL Turbo和LCM带来AI画图的实时生成时代:速度跟打字一样快,图像瞬间呈现

#表1 展示了消融實驗的結果,下列為主要結論:

SDXL Turbo和LCM带来AI画图的实时生成时代:速度跟打字一样快,图像瞬间呈现

接下來是與其他 SOTA 模型的對比,此處研究者沒有採用自動化指標,而是選擇了更可靠的使用者偏好評估方法,目標是評估 prompt 遵循情況和整體影像。

為了比較多個不同的模型變體(StyleGAN-T 、OpenMUSE、IF-XL、SDXL和LCM-XL),實驗使用相同的提示產生輸出。在盲測中,SDXL Turbo以單步擊敗LCM-XL的4步配置,並且僅用4步擊敗SDXL的50步配置。透過這些結果,可以看出SDXL Turbo的性能優於最先進的多步驟模型,同時計算需求明顯降低,而不需要犧牲影像品質

SDXL Turbo和LCM带来AI画图的实时生成时代:速度跟打字一样快,图像瞬间呈现

##這裡呈現的是關於推理速度的ELO分數的可視化圖表

SDXL Turbo和LCM带来AI画图的实时生成时代:速度跟打字一样快,图像瞬间呈现

在表2中,對使用相同基礎模型的不同few-step採樣和蒸餾方法進行了比較。結果表明,ADD方法的性能優於所有其他方法,包括8步的標準DPM求解器

SDXL Turbo和LCM带来AI画图的实时生成时代:速度跟打字一样快,图像瞬间呈现

作為定量實驗結果的補充,論文也展示了部分定性實驗結果,展示了ADD-XL 在初始樣本基礎上的改進能力。圖 3 將 ADD-XL(1 step)與 few-step 方案中目前最佳基線進行了比較。圖 4 介紹了 ADD-XL 的迭代採樣過程。圖 8 將 ADD-XL 與其教師模型 SDXL-Base 進行了直接比較。如使用者研究所示,ADD-XL 在品質和 prompt 對齊方面都優於教師模式。

SDXL Turbo和LCM带来AI画图的实时生成时代:速度跟打字一样快,图像瞬间呈现

SDXL Turbo和LCM带来AI画图的实时生成时代:速度跟打字一样快,图像瞬间呈现

SDXL Turbo和LCM带来AI画图的实时生成时代:速度跟打字一样快,图像瞬间呈现

如欲了解更多的研究細節,請參考原始論文######

以上是SDXL Turbo與LCM帶來AI畫圖的即時生成時代:速度跟打字一樣快,影像瞬間呈現的詳細內容。更多資訊請關注PHP中文網其他相關文章!

相關標籤:
來源:51cto.com
本網站聲明
本文內容由網友自願投稿,版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容,請聯絡admin@php.cn
最新問題
熱門教學
更多>
最新下載
更多>
網站特效
網站源碼
網站素材
前端模板