這篇綜述(Diffusion Models: A Comprehensive Survey of Methods and Applications)來自加州大學&Google Research的Ming-Hsuan Yang、北京大學崔斌實驗室以及CMU、UCLA 、蒙特利爾Mila研究院等眾研究團隊,首次對現有的擴散生成模型(diffusion model)進行了全面的總結分析,從diffusion model演算法細化分類、和其他五大生成模型的關聯以及在七大領域中的應用等面向展開,最後提出了diffusion model的現有limitation和未來的發展方向。
文章連結:https://arxiv.org/abs/2209.00796本綜述diffusion models論文分類彙總github連結:https://github.com/YangLing0818/Diffusion -Models-Papers-Survey-Taxonomy
擴散模型(diffusion models)是深度生成模型中新的SOTA。擴散模型在圖片生成任務中超越了原始SOTA:GAN,並且在許多應用領域都有出色的表現,如計算機視覺,NLP、波形信號處理、多模態建模、分子圖建模、時間序列建模、對抗性淨化等。此外,擴散模型與其他研究領域有著密切的聯繫,如穩健學習、表示學習、強化學習。
然而,原始的擴散模型也有缺點,它的取樣速度慢,通常需要數千個評估步驟才能抽取一個樣本;它的最大似然估計無法和基於似然的模型相比;它泛化到各種資料類型的能力較差。如今許多研究已經從實際應用的角度解決上述限製做出了許多努力,或從理論角度對模型能力進行了分析。
然而,現在缺乏對擴散模型從演算法到應用的最新進展的系統性回顧。為了反映這一快速發展領域的進展,我們對擴散模型進行了首個全面綜述。我們設想我們的工作將闡明擴散模型的設計考慮和先進方法,展示其在不同領域的應用,並指出未來的研究方向。此綜述的摘要如下圖所示:
儘管diffusion model在各類任務中都有著優秀的表現,它仍還有自己的缺點,並有許多研究對diffusion model進行了改善。
為了系統性地闡明diffusion model的研究進展,我們總結了原始擴散模型的三個主要缺點,取樣速度慢,最大化似然差、資料泛化能力弱,並提出將的diffusion models改進研究分為對應的三類:採樣速度提升、最大似然增強和資料泛化增強。
我們首先說明改善的動機,再根據方法的特性將每個改進方向的研究進一步細化分類,從而清楚的展現方法之間的聯繫與區別。在此我們僅選取部分重要方法為例, 我們的工作中對每類方法都做了詳細的介紹,內容如圖所示:
在分析完三類擴散模型後,我們將介紹其他的五種生成模型GAN,VAE,Autoregressive model, Normalizing flow, Energy-based model。
考慮到擴散模型的優良性質,研究者們已經根據其特性將diffusion model與其他生成模型結合,所以為了進一步展現diffusion model 的特點和改進工作,我們詳細地介紹了diffusion model和其他生成模型的結合的工作並闡明了在原始生成模型上的改進之處。
Diffusion model在許多領域都有著優異的表現,並且考慮到不同領域的應用中diffusion model產生了不同的變形,我們系統地介紹了diffusion model的應用研究,其中包含以下領域:電腦視覺,NLP、波形訊號處理、多模態建模、分子圖建模、時間序列建模、對抗性淨化。對於每個任務,我們定義了該任務並介紹利用擴散模型處理任務的工作,我們將本項工作的主要貢獻總結如下:
##生成式建模的一個核心問題是模型的彈性與可計算性之間的權衡。擴散模型的基本概念是正向擴散過程來系統地擾動資料中的分佈,然後透過學習反向擴散過程來恢復資料的分佈,這樣就了產生一個高度靈活且易於計算的生成模型。
(1)Denoising Diffusion Probabilistic Models(DDPM)
一個DDPM由兩個參數化馬可夫鏈組成,並使用變異推斷以在有限時間後產生與原始資料分佈一致的樣本。前向鏈的作用是擾動數據,它根據預先設計的噪聲進度向數據逐漸加入高斯噪聲,直到數據的分佈趨於先驗分佈,即標準高斯分佈。反向鏈從給定的先驗開始並使用參數化的高斯轉換核,學習逐步恢復原始資料分佈。用表示原始資料及其分佈,則前向鏈的分佈是可由下式表達:
這表示前向鍊是馬可夫過程,是加入t步雜訊後的樣本,是事先給定的控制雜訊進度的參數。當 趨於1時,可以近似地認為服從標準高斯分佈。當很小時,逆向過程的轉移核可以近似認為也是高斯的:
我們可以將變分下界作為損失函數來學習:
(2)Score-Based Generative Models(SGM)
上述DDPM可以視為SGM的離散形式。 SGM建構一個隨機微分方程式(SDE)來平滑的擾亂資料分佈,將原始資料分佈轉換到已知的先驗分佈:
##和一個對應的逆向SDE,來將先驗分佈變換回原始資料分佈:
#因此,要逆轉擴散過程並產生數據,我們需要的唯一資訊就是在每個時間點的分數函數。利用score-matching的技巧我們可以透過以下損失函數來學習分數函數:
對兩種方法的進一步介紹和兩者關係的介紹請參見我們的文章。原始擴散模型的三個主要缺點,取樣速度慢,最大化似然差、資料泛化能力弱。最近許多研究都在解決這些缺點,因此我們將改進的擴散模型分為三類:採樣速度提升、最大似然增強和資料泛化增強。在接下來的三、四、五節我們將對這三類模型進行詳細的介紹。
在應用時,為了讓新樣本的品質達到最佳,擴散模型往往需要成千上萬萬步計算來取得一個新樣本。這限制了diffusion model的實際應用價值,因為在實際應用時,我們往往需要產生大量的新樣本,來為下一步處理提供材料。
研究者們在提高diffusion model取樣速度上進行了大量的研究。我們對這些研究進行了詳細的闡述。我們將其細化分類為三種方法:Discretization Optimization,Non-Markovian Process,Partial Sampling。
(1)Discretization Optimization方法最佳化求解diffusion SDE的方法。因為現實中求解複雜SDE只能使用離散解來逼近真正的解,所以該類別方法試圖優化SDE的離散化方法,在確保樣本品質的同時減少離散步數。 SGM 提出了一個通用的方法來求解逆向過程,即對前向和後向過程採取相同的離散方法。如果給定了前向SDE的離散方式:
#那麼我們就可以以相同的方式離散化逆向SDE:
這種方法比樸素DDPM效果略好一點。進一步,SGM在SDE求解器中加入了一個矯正器,讓每個步驟產生的樣本都有正確的分佈。在求解的每一步,求解器給出一個樣本後,矯正器都使用馬可夫鏈蒙特卡羅方法來矯正剛產生的樣本的分佈。實驗顯示向求解器中加入矯正器比直接增加求解器的步數效率更高。
(2)Non-Markovian Process方法突破了原有Markovian Process的限制,其逆過程的每一步可以依賴更多以往的樣本來進行預測新樣本,所以在步長較大時也能做出較好的預測,加速採樣過程。其中主要的工作DDIM,不再假設前向過程是馬可夫過程,而是服從如下分佈:
#DDIM的取樣過程可以視為離散化的神經常微分方程,其取樣過程更有效率,並且支持樣本的內插。進一步的研究發現DDIM可以視為流形上擴散模型PNDM的特殊情況。
(3)Partial Sampling方法透過在generation process中忽略一部分的時間節點,而只使用剩下的時間節點來產生樣本,直接減少了取樣時間。例如,Progressive Distillation從訓練好的擴散模型中蒸餾出效率更高的擴散模型。對於訓練好的一個擴散模型,Progressive Distillation會從新訓練一個擴散模型,使新的擴散模型的一步對應於訓練好的擴散模型的兩步,這樣新模型就可以省去舊模型一半的採樣過程。具體演算法如下:
不斷循環這個蒸餾過程就能讓取樣步驟指數級下降。
擴散模型在最大似然估計的表現差於基於似然函數的生成模型,但最大化似然估計在許多應用場景都有重要意義,例如圖片壓縮, 半監督學習, 對抗性淨化。由於對數似然難以直接計算,研究主要集中在最佳化和分析變分下界(VLB)。我們對提高擴散模型最大似然估計的模型進行了詳細的闡述。我們將其細化分類為三類方法:Objectives Designing,Noise Schedule Optimization,Learnable Reverse Variance。
(1)Objectives Designing方法利用擴散 SDE推倒出生成資料的對數似然與分數函數相符的損失函數的關係。這樣透過適當設計損失函數,就可以最大化 VLB 和對數似然。 Song et al. 證明了可以設計損失函數的權重函數,使得plug-in reverse SDE 生成樣本的似然函數值小於等於損失函數值,即損失函數是似然函數的上界。分數函數擬合的損失函數如下:
我們只需將權重函數設為擴散係數g(t)即可讓損失函數成為似然函數的VLB,即:
#(2)Noise Schedule Optimization透過設計或學習前向過程的噪音進度來增大VLB。 VDM證明了當離散步數接近無窮時,損失函數完全由信噪比函數SNR(t)的端點決定:
##那麼在離散步數接近無窮時,可以透過學習信噪比函數SNR(t)的端點最優化VLB,而透過學習信噪比函數中間部分的函數值來實現模型其他方面的改進。 3.Learnable Reverse Variance方法學習反向過程的方差,從而較少擬合誤差,可以有效最大化VLB。 Analytic-DPM證明,在DDPM和DDIM中存在反向過程中的最優期望和變異數:
##使用上述公式和訓練好的分數函數,在給定前向過程的條件下,最優的VLB可以近似達到。5
數個研究工作將擴散模型推廣到適用於其他資料類型的模型,我們對這些方法進行了詳細地闡釋。我們將其細化分類為兩類方法:Feature Space Unification,Data-Dependent Transition Kernels。
(1)Feature Space Unification方法將資料轉換到統一形式的latent space,然後再latent space上進行擴散。 LSGM提出將資料經由VAE框架先轉換到連續的latent space 上後再在其上進行擴散。這個方法的困難在於如何同時訓練VAE和擴散模型。 LSGM顯示由於潛在先驗是intractable的,因此分數匹配損失不再適用。 LSGM直接使用VAE中傳統的損失函數ELBO作為損失函數,並導出了ELBO和分數匹配的關係:
##該式在忽略常數的意義下成立。透過參數化擴散過程中樣本的分數函數,LSGM可以高效率的學習和最佳化ELBO。
(2)Data-Dependent Transition Kernels方法根據資料類型的特性設計diffusion process 中的transition kernels,使擴散模型可以直接套用於特定的資料類型。 D3PM為離散型資料設計了transition kernel,可以設為lazy random-walk,absorbing state等。 GEODIFF為3D分子圖資料設計了平移-旋轉不變的圖神經網絡,並且證明了具有不變性的初始分佈和transition kernel可以導出具有不變性的邊緣分佈。假設是一個平移-旋轉變換,如:
#那麼產生的樣本分佈也有平移-旋轉不變性:
在下面的每個小節中,我們先介紹其他五類重要的生成模型,並分析它們的優點和限制。然後我們介紹了擴散模型是如何與它們聯繫起來的,並說明透過結合擴散模型來改進這些生成模型。 VAE,GAN,Autoregressive model, Normalizing flow, Energy-based model和擴散模型的聯繫如下圖所示:
Autoregressive model在需要保證資料有一定的結構,這導致設計和參數化自迴歸模型非常困難。擴散模型的訓練啟發了自回歸模型的訓練,透過特定的訓練方式避免了設計的困難。
Energy-based model直接對原始資料的分佈建模,但直接建模導致學習和取樣都比較困難。透過使用擴散恢復似然,模型可以先對樣本加入微小的噪聲,再從有略微噪聲的樣本分佈來推斷原始樣本的分佈,使的學習和採樣過程更簡單和穩定。
7
擴散模型的應用在本節中,我們分別介紹了擴散模型在電腦視覺、自然語言處理、波形訊號處理、多模態學習、分子圖生成、時間序列以及對抗學習等七大應用方向中的應用,並對每類應用中的方法進行了細分並解析。例如在電腦視覺中可以用diffusion model進行影像補全修復(RePaint):
##################在多模態任務中可以用diffusion model進行文字到圖像的生成(GLIDE):######################也可以在分子圖生成中用diffusion model進行藥物分子和蛋白質分子的生成(GeoDiff ):#######應用分類總表見表:
以上是最近火災的Diffusion Model,首篇擴散生成模型綜述!的詳細內容。更多資訊請關注PHP中文網其他相關文章!