針對圖像編輯中的擴散模型，中科院聯(lián)合Adobe和蘋果公司的研究人員發(fā)布了一篇重磅綜述。

全文長達26頁，共1.5萬余詞，涵蓋297篇文獻，全面研究了圖像編輯的各種前沿方法。

同時，作者還提出了全新的benchmark，為研究者提供了便捷的學習參考工具。

在這份綜述中，作者從理論和實踐層面，詳盡總結(jié)了使用擴散模型進行圖像編輯的現(xiàn)有方法。

作者從學習策略、輸入條件等多個角度對相關(guān)成果進行分類，并展開了深入分析。

為了進一步評估模型性能，作者還提出了一個測評基準，并展望了未來研究的一些潛在方向。

△基于擴散模型的圖像編輯成果速覽

下面，作者將從任務(wù)分類、實現(xiàn)方式、測試基準和未來展望四個方面介紹基于擴散模型的圖像編輯成果。

圖像編輯的分類

除了在圖像生成、恢復(fù)和增強方面取得的重大進步外，擴散模型在圖像編輯方面也實現(xiàn)了顯著突破，相比之前占主導(dǎo)地位的生成對抗網(wǎng)絡(luò)（GANs），前者具有更強的可控性。

不同于“從零開始”的圖像生成，以及旨在修復(fù)模糊圖像、提高質(zhì)量的圖像恢復(fù)和增強，圖像編輯涉及對現(xiàn)有圖像外觀、結(jié)構(gòu)或內(nèi)容的修改，包括添加對象、替換背景和改變紋理等任務(wù)。

在這項調(diào)查中，作者根據(jù)學習策略將圖像編輯論文分為三個主要組別：基于訓練的方法、測試時微調(diào)方法和無需訓練和微調(diào)的方法。

此外，作者還探討了控制編輯過程使用的10種輸入條件，包括文本、掩碼、參考圖像、類別、布局、姿勢、草圖、分割圖、音頻和拖動點。

進一步地，作者調(diào)查了這些方法可以完成的12種最常見的編輯類型，它們被組織成以下三個廣泛的類別：

• 語義編輯：此類別包括對圖像內(nèi)容和敘述的更改，影響所描繪場景的故事、背景或主題元素。這一類別內(nèi)的任務(wù)包括對象添加、對象移除、對象替換、背景更改和情感表達修改。
• 風格編輯：此類別側(cè)重于增強或轉(zhuǎn)換圖像的視覺風格和審美元素，而不改變其敘述內(nèi)容。這一類別內(nèi)的任務(wù)包括顏色更改、紋理更改和整體風格更改，涵蓋藝術(shù)性和現(xiàn)實性風格。
• 結(jié)構(gòu)編輯：此類別涉及圖像內(nèi)元素的空間布局、定位、視角和特征的變化，強調(diào)場景內(nèi)對象的組織和展示。這一類別內(nèi)的任務(wù)包括對象移動、對象大小和形狀更改、對象動作和姿勢更改以及視角/視點更改。

圖像編輯的實現(xiàn)方式

基于訓練的方法

在基于擴散模型的圖像編輯領(lǐng)域，基于訓練的方法已經(jīng)獲得了顯著的突出地位。

這些方法不僅因其穩(wěn)定的擴散模型訓練和有效的數(shù)據(jù)分布建模而著名，而且在各種編輯任務(wù)中表現(xiàn)可靠。

為了徹底分析這些方法，作者根據(jù)它們的應(yīng)用范圍、訓練所需條件和監(jiān)督類型將它們分類為四個主要組別。

根據(jù)核心編輯方法，這些主要組別中的方法又可以細分為不同的類型。

下圖展示了兩種有代表性的CLIP指導(dǎo)方法——DiffusionCLIP和Asyrp的框架圖。

△樣本圖像來自CelebA數(shù)據(jù)集上的Asyrp

下面的圖片，展示的是指令圖像編輯方法的通用框架。

△示例圖像來自InstructPix2Pix、InstructAny2Pix和MagicBrush。

測試時微調(diào)的方法

在圖像生成和編輯中，還會采用微調(diào)策略來增強圖像編輯能力，測試時微調(diào)帶來了精確性和可控制性的重要提升。

如下圖所示，微調(diào)方法的既包括微調(diào)整個去噪模型，也包括專注于特定層或嵌入。

此外，作者還討論了超網(wǎng)絡(luò)的集成和直接圖像表示優(yōu)化

下圖展示了使用不同微調(diào)組件的微調(diào)框架。

△樣本圖像來自Custom-Edit

免訓練和微調(diào)方法

在圖像編輯領(lǐng)域，無需訓練和微調(diào)的方法起點是它們快速且成本低，因為在整個編輯過程中不需要任何形式的訓練（在數(shù)據(jù)集上）或微調(diào)（在源圖像上）。

根據(jù)它們修改的內(nèi)容，可以分為五個類別，這些方法巧妙地利用擴散模型內(nèi)在的原則來實現(xiàn)編輯目標。

下圖是免訓練方法的通用框架。

△樣本圖片來自LEDITS++

圖像inpainting（補全）和outpainting（外擴）

圖像補全和外擴通常被視為圖像編輯的子任務(wù)，可以分為兩大類型——上下文驅(qū)動的補全（上排）與多模態(tài)條件補全（下排）。

△樣本分別來自于Palette和Imagen Editor

全新測試基準

除了分析各種方法的實現(xiàn)原理，評估這些方法在不同編輯任務(wù)中的能力也至關(guān)重要，但現(xiàn)有的圖像編輯測試標準存在局限。

例如，EditBench主要針對文本和掩碼引導(dǎo)的補全，但忽略了涉及全局編輯的任務(wù)（如風格轉(zhuǎn)換）；TedBench雖然擴展了任務(wù)范圍，但缺乏詳細指導(dǎo)；EditVal試圖提供更全面的任務(wù)和方法覆蓋范圍，但圖像通常分辨率低且模糊……

為了解決這些問題，作者提出了EditEval基準，包括一個50張高質(zhì)量圖像的數(shù)據(jù)集，且每張圖像都附有文本提示，可以評估模型在7個常見編輯任務(wù)的性能。

這7種任務(wù)包括物體添加/移除/替換，以及背景、風格和姿勢、動作的改變。

此外，作者還提出了LMM分數(shù)，利用多模態(tài)大模型（LMMs）評估不同任務(wù)上的編輯性能，并進行了真人用戶研究以納入主觀評估。

△LMM Score與用戶研究的皮爾遜相關(guān)系數(shù)

下圖比較了LMM Score/CLIPScore與用戶研究的皮爾遜相關(guān)系數(shù)。

挑戰(zhàn)和未來方向

作者認為，盡管在使用擴散模型進行圖像編輯方面取得了成功，但仍有一些不足需要在未來的工作中加以解決。

減少模型推理步驟

大多數(shù)基于擴散的模型在推理過程中需要大量的步驟來獲取最終圖像，這既耗時又耗費計算資源，給模型部署和用戶體驗帶來挑戰(zhàn)。

為了提高推理效率，已經(jīng)由團隊研究了少步驟或一步生成的擴散模型。

近期的方法通過從預(yù)訓練的強擴散模型中提取知識來減少步驟數(shù)，以便少步驟模型能夠模仿強模型的行為。

一個更具挑戰(zhàn)性的方向是直接開發(fā)少步驟模型，而不依賴于預(yù)訓練的模型（例如一致性模型）。

提高模型效率

訓練一個能夠生成逼真結(jié)果的擴散模型在計算上是密集的，需要大量的高質(zhì)量數(shù)據(jù)。

這種復(fù)雜性使得開發(fā)用于圖像編輯的擴散模型非常具有挑戰(zhàn)性。

為了降低訓練成本，近期的工作設(shè)計了更高效的網(wǎng)絡(luò)架構(gòu)作為擴散模型的骨干。

此外，另一個重要方向是只訓練部分參數(shù)，或者凍結(jié)原始參數(shù)并在預(yù)訓練的擴散模型之上添加一些新層。

復(fù)雜對象結(jié)構(gòu)編輯

現(xiàn)有的工作可以在編輯圖像時合成逼真的顏色、風格或紋理，但處理復(fù)雜結(jié)構(gòu)時仍然會產(chǎn)生明顯的修改痕跡，例如手指、標志和文字。

研究者已經(jīng)在嘗試解決這些問題，常用的策略是把“六個手指”等常見問題作為負面提示，以使模型避免生成此類圖像，這在某些情況下是有效的，但不夠穩(wěn)健。

近期的工作中，已有團隊開始使用布局、邊緣或密集標簽作為指導(dǎo)，編輯圖像的全局或局部結(jié)構(gòu)。

復(fù)雜的光照和陰影編輯

編輯對象的光照或陰影仍然是一個挑戰(zhàn)，因為這需要準確估計場景中的光照條件。

以前的工作（如Total Relighting）使用網(wǎng)絡(luò)組合來估計前景對象的法線、反照率和陰影，以獲得逼真的重新照明效果。

最近，也由有團隊提出將擴散模型用于編輯面部的光照，ShadowDiffusion也探索了基于擴散模型的陰影合成，可以生成合理的對象陰影。

然而，使用擴散模型在不同背景條件下準確編輯對象的陰影仍然是一個未解決的問題。

圖像編輯模型的泛化性

現(xiàn)有基于擴散的圖像編輯模型能夠為給定的一部分條件合成逼真的視覺內(nèi)容，但在許多現(xiàn)實世界場景中仍然會失敗。

這個問題的根本原因在于，模型無法準確地對所有可能的樣本在條件分布空間中進行建模。

如何改進模型以始終生成無瑕疵的內(nèi)容仍然是一個挑戰(zhàn)，解決這個問題有以下幾種思路：

首先是擴大訓練數(shù)據(jù)規(guī)模，以覆蓋具有挑戰(zhàn)性的場景，這種方式效果顯著，但成本較高，如在醫(yī)學圖像、視覺檢測等領(lǐng)域數(shù)據(jù)難以收集。

第二種方法是調(diào)整模型以接受更多條件，如結(jié)構(gòu)引導(dǎo)、3D感知引導(dǎo)和文本引導(dǎo)，以實現(xiàn)更可控和確定性的內(nèi)容創(chuàng)作。

此外，還可以采用迭代細化或多階段訓練的方式，以逐步改進模型的初始結(jié)果。

可靠的評估指標

對圖像編輯進行準確評估，對于確保編輯內(nèi)容與給定條件的對齊至關(guān)重要。

盡管有如FID、KID、LPIPS、CLIP得分、PSNR和SSIM等定量指標，但大多數(shù)現(xiàn)有評估工作仍然嚴重依賴于用戶研究，這既不高效也不可擴展。

可靠的定量評估指標仍然是一個待解決的問題。最近，已經(jīng)有團隊提出了更準確的指標來量化對象的感知相似性。

DreamSim測量了兩幅圖像的中等級別相似性，考慮了布局、姿態(tài)和語義內(nèi)容，并且優(yōu)于LPIPS。

類似的，前景特征平均（FFA）也是一種簡單而有效的方法，可被用于測量對象的相似性。

另外，作者在本文中提出了的LMM score，也是一種有效的圖像編輯度量。

更多有關(guān)用于圖像編輯的擴散模型的詳細信息，可以閱讀原作，同時作者也在GitHub上發(fā)布了附帶資源庫。

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

Github：https://github.com/SiatMMLab/Awesome-Diffusion-Model-Based-Image-Editing-Methods