1. 程式人生 > >GAN系列之原理詳解

GAN系列之原理詳解

深度學習交流QQ群:116270156

GAN的基本原理其實非常簡單,這裡以生成圖片為例進行說明。假設我們有兩個網路,G(Generator)和D(Discriminator)。正如它的名字所暗示的那樣,它們的功能分別是:

  • G是一個生成圖片的網路,它接收一個隨機的噪聲z,通過這個噪聲生成圖片,記做G(z)。

  • D是一個判別網路,判別一張圖片是不是“真實的”。它的輸入引數是x,x代表一張圖片,輸出D(x)代表x為真實圖片的概率,如果為1,就代表100%是真實的圖片,而輸出為0,就代表不可能是真實的圖片。

在訓練過程中,生成網路G的目標就是儘量生成真實的圖片去欺騙判別網路D。而D的目標就是儘量把G生成的圖片和真實的圖片分別開來。這樣,G和D構成了一個動態的“博弈過程”。

最後博弈的結果是什麼?在最理想的狀態下,G可以生成足以“以假亂真”的圖片G(z)。對於D來說,它難以判定G生成的圖片究竟是不是真實的,因此D(G(z)) = 0.5。

這樣我們的目的就達成了:我們得到了一個生成式的模型G,它可以用來生成圖片。

以上只是大致說了一下GAN的核心原理,如何用數學語言描述呢?這裡直接摘錄論文裡的公式:

簡單分析一下這個公式:

  • 整個式子由兩項構成。x表示真實圖片,z表示輸入G網路的噪聲,而G(z)表示G網路生成的圖片。

  • D(x)表示D網路判斷真實圖片是否真實的概率(因為x就是真實的,所以對於D來說,這個值越接近1越好)。而D(G(z))是D網路判斷

    G生成的圖片的是否真實的概率

  • G的目的:上面提到過,D(G(z))是D網路判斷G生成的圖片是否真實的概率,G應該希望自己生成的圖片“越接近真實越好”。也就是說,G希望D(G(z))儘可能得大,這時V(D, G)會變小。因此我們看到式子的最前面的記號是min_G。

  • D的目的:D的能力越強,D(x)應該越大D(G(x))應該越小。這時V(D,G)會變大。因此式子對於D來說是求最大(max_D)

下面這幅圖片很好地描述了這個過程:

那麼如何用隨機梯度下降法訓練D和G?論文中也給出了演算法:


於是模型的目標函式就出來了,對於生成器G,目標函式為(梯度下降):


而對於判別器D,目標函式為(梯度上升):


於是模型就成了優化這兩個目標函式的問題了。這樣的話就只需要反向傳播來對模型訓練就好了,沒有像傳統的生成式模型的最大似然函式的計算、馬爾科夫鏈或其他推理等運算了。

這就是這篇文章大概的思路,具體的內容細節以及一些數學公式的推導可以仔細閱讀論文原文。

這裡紅框圈出的部分是我們要額外注意的。第一步我們訓練D,D是希望V(G, D)越大越好,所以是加上梯度(ascending)。第二步訓練G時,V(G, D)越小越好,所以是減去梯度(descending)。整個訓練過程交替進行。

二、DCGAN原理介紹

我們知道深度學習中對影象處理應用最好的模型是CNN,那麼如何把CNN與GAN結合?DCGAN是這方面最好的嘗試之一(點選檢視論文

DCGAN的原理和GAN是一樣的,這裡就不在贅述。它只是把上述的G和D換成了兩個卷積神經網路(CNN)。但不是直接換就可以了,DCGAN對卷積神經網路的結構做了一些改變,以提高樣本的質量和收斂的速度,這些改變有:

  • 取消所有pooling層。G網路中使用轉置卷積(transposed convolutional layer)進行上取樣,D網路中用加入stride的卷積代替pooling。

  • 在D和G中均使用batch normalization

  • 去掉FC層,使網路變為全卷積網路

  • G網路中使用ReLU作為啟用函式,最後一層使用tanh

  • D網路中使用LeakyReLU作為啟用函式

DCGAN中的G網路示意: