2
| 本文作者： 亞萌 | 2017-02-06 16:37 |
雷鋒網(wǎng)按：本文作者鄭華濱，原載于知乎。雷鋒網(wǎng)已獲轉(zhuǎn)載授權(quán)。
在GAN的相關(guān)研究如火如荼甚至可以說是泛濫的今天，一篇新鮮出爐的arXiv論文《Wassertein GAN》卻在Reddit的Machine Learning頻道火了，連Goodfellow都在帖子里和大家熱烈討論，這篇論文究竟有什么了不得的地方呢？
要知道自從2014年Ian Goodfellow提出以來，GAN就存在著訓練困難、生成器和判別器的loss無法指示訓練進程、生成樣本缺乏多樣性等問題。從那時起，很多論文都在嘗試解決，但是效果不盡人意，比如最有名的一個改進DCGAN依靠的是對判別器和生成器的架構(gòu)進行實驗枚舉，最終找到一組比較好的網(wǎng)絡架構(gòu)設(shè)置，但是實際上是治標不治本，沒有徹底解決問題。而今天的主角Wasserstein GAN（下面簡稱WGAN）成功地做到了以下爆炸性的幾點：
徹底解決GAN訓練不穩(wěn)定的問題，不再需要小心平衡生成器和判別器的訓練程度
基本解決了collapse mode的問題，確保了生成樣本的多樣性
訓練過程中終于有一個像交叉熵、準確率這樣的數(shù)值來指示訓練的進程，這個數(shù)值越小代表GAN訓練得越好，代表生成器產(chǎn)生的圖像質(zhì)量越高（如題圖所示）
以上一切好處不需要精心設(shè)計的網(wǎng)絡架構(gòu)，最簡單的多層全連接網(wǎng)絡就可以做到
那以上好處來自哪里？這就是令人拍案叫絕的部分了——實際上作者整整花了兩篇論文，在第一篇《Towards Principled Methods for Training Generative Adversarial Networks》里面推了一堆公式定理，從理論上分析了原始GAN的問題所在，從而針對性地給出了改進要點；在這第二篇《Wassertein GAN》里面，又再從這個改進點出發(fā)推了一堆公式定理，最終給出了改進的算法實現(xiàn)流程，而改進后相比原始GAN的算法實現(xiàn)流程卻只改了四點：
判別器最后一層去掉sigmoid
生成器和判別器的loss不取log
每次更新判別器的參數(shù)之后把它們的絕對值截斷到不超過一個固定常數(shù)c
不要用基于動量的優(yōu)化算法（包括momentum和Adam），推薦RMSProp，SGD也行
算法截圖如下：
改動是如此簡單，效果卻驚人地好，以至于Reddit上不少人在感嘆：就這樣？沒有別的了？ 太簡單了吧！這些反應讓我想起了一個頗有年頭的雞湯段子，說是一個工程師在電機外殼上用粉筆劃了一條線排除了故障，要價一萬美元——畫一條線，1美元；知道在哪畫線，9999美元。上面這四點改進就是作者Martin Arjovsky劃的簡簡單單四條線，對于工程實現(xiàn)便已足夠，但是知道在哪劃線，背后卻是精巧的數(shù)學分析，而這也是本文想要整理的內(nèi)容。
本文內(nèi)容分為五個部分：
原始GAN究竟出了什么問題？（此部分較長）
WGAN之前的一個過渡解決方案
Wasserstein距離的優(yōu)越性質(zhì)
從Wasserstein距離到WGAN
總結(jié)
理解原文的很多公式定理需要對測度論、 拓撲學等數(shù)學知識有所掌握，本文會從直觀的角度對每一個重要公式進行解讀，有時通過一些低維的例子幫助讀者理解數(shù)學背后的思想，所以不免會失于嚴謹，如有引喻不當之處，歡迎在評論中指出。
以下簡稱《Wassertein GAN》為“WGAN本作”，簡稱《Towards Principled Methods for Training Generative Adversarial Networks》為“WGAN前作”。
WGAN源碼實現(xiàn)：martinarjovsky/WassersteinGAN
回顧一下，原始GAN中判別器要最小化如下?lián)p失函數(shù)，盡可能把真實樣本分為正例，生成樣本分為負例：
（公式1 ）
其中
是真實樣本分布，
是由生成器產(chǎn)生的樣本分布。對于生成器，Goodfellow一開始提出來一個損失函數(shù)，后來又提出了一個改進的損失函數(shù)，分別是
（公式2）
（公式3）
后者在WGAN兩篇論文中稱為“the - log D alternative”或“the - log D trick”。WGAN前作分別分析了這兩種形式的原始GAN各自的問題所在，下面分別說明。
一句話概括：判別器越好，生成器梯度消失越嚴重。WGAN前作從兩個角度進行了論證，第一個角度是從生成器的等價損失函數(shù)切入的。
首先從公式1可以得到，在生成器G固定參數(shù)時最優(yōu)的判別器D應該是什么。對于一個具體的樣本，它可能來自真實分布也可能來自生成分布，它對公式1損失函數(shù)的貢獻是
令其關(guān)于
的導數(shù)為0，得
化簡得最優(yōu)判別器為：
（公式4）
這個結(jié)果從直觀上很容易理解，就是看一個樣本
來自真實分布和生成分布的可能性的相對比例。如果
且
，最優(yōu)判別器就應該非常自信地給出概率0；如果
，說明該樣本是真是假的可能性剛好一半一半，此時最優(yōu)判別器也應該給出概率0.5。
然而GAN訓練有一個trick，就是別把判別器訓練得太好，否則在實驗中生成器會完全學不動（loss降不下去），為了探究背后的原因，我們就可以看看在極端情況——判別器最優(yōu)時，生成器的損失函數(shù)變成什么。給公式2加上一個不依賴于生成器的項，使之變成
注意，最小化這個損失函數(shù)等價于最小化公式2，而且它剛好是判別器損失函數(shù)的反。代入最優(yōu)判別器即公式4，再進行簡單的變換可以得到
（公式5)
變換成這個樣子是為了引入Kullback–Leibler divergence（簡稱KL散度）和Jensen-Shannon divergence（簡稱JS散度）這兩個重要的相似度衡量指標，后面的主角之一Wasserstein距離，就是要來吊打它們兩個的。所以接下來介紹這兩個重要的配角——KL散度和JS散度：
（公式6）
（公式7）
于是公式5就可以繼續(xù)寫成
（公式8）
到這里讀者可以先喘一口氣，看看目前得到了什么結(jié)論：根據(jù)原始GAN定義的判別器loss，我們可以得到最優(yōu)判別器的形式；而在最優(yōu)判別器的下，我們可以把原始GAN定義的生成器loss等價變換為最小化真實分布
與生成分布
之間的JS散度。我們越訓練判別器，它就越接近最優(yōu)，最小化生成器的loss也就會越近似于最小化
和
之間的JS散度。
問題就出在這個JS散度上。我們會希望如果兩個分布之間越接近它們的JS散度越小，我們通過優(yōu)化JS散度就能將
“拉向”
，最終以假亂真。這個希望在兩個分布有所重疊的時候是成立的，但是如果兩個分布完全沒有重疊的部分，或者它們重疊的部分可忽略（下面解釋什么叫可忽略），它們的JS散度是多少呢？
答案是
，因為對于任意一個x只有四種可能：
且
且
且
且
第一種對計算JS散度無貢獻，第二種情況由于重疊部分可忽略所以貢獻也為0，第三種情況對公式7右邊第一個項的貢獻是
，第四種情況與之類似，所以最終
。
換句話說，無論
跟
是遠在天邊，還是近在眼前，只要它們倆沒有一點重疊或者重疊部分可忽略，JS散度就固定是常數(shù)
，而這對于梯度下降方法意味著——梯度為0！此時對于最優(yōu)判別器來說，生成器肯定是得不到一丁點梯度信息的；即使對于接近最優(yōu)的判別器來說，生成器也有很大機會面臨梯度消失的問題。
但是
與
不重疊或重疊部分可忽略的可能性有多大？不嚴謹?shù)拇鸢甘牵悍浅４?。比較嚴謹?shù)拇鸢甘牵?strong>當
與
的支撐集（support）是高維空間中的低維流形（manifold）時，
與
重疊部分測度（measure）為0的概率為1。
不用被奇怪的術(shù)語嚇得關(guān)掉頁面，雖然論文給出的是嚴格的數(shù)學表述，但是直觀上其實很容易理解。首先簡單介紹一下這幾個概念：
支撐集（support）其實就是函數(shù)的非零部分子集，比如ReLU函數(shù)的支撐集就是
，一個概率分布的支撐集就是所有概率密度非零部分的集合。
流形（manifold）是高維空間中曲線、曲面概念的拓廣，我們可以在低維上直觀理解這個概念，比如我們說三維空間中的一個曲面是一個二維流形，因為它的本質(zhì)維度（intrinsic dimension）只有2，一個點在這個二維流形上移動只有兩個方向的自由度。同理，三維空間或者二維空間中的一條曲線都是一個一維流形。
測度（measure）是高維空間中長度、面積、體積概念的拓廣，可以理解為“超體積”。
回過頭來看第一句話，“當
與
的支撐集是高維空間中的低維流形時”，基本上是成立的。原因是GAN中的生成器一般是從某個低維（比如100維）的隨機分布中采樣出一個編碼向量，再經(jīng)過一個神經(jīng)網(wǎng)絡生成出一個高維樣本（比如64x64的圖片就有4096維）。當生成器的參數(shù)固定時，生成樣本的概率分布雖然是定義在4096維的空間上，但它本身所有可能產(chǎn)生的變化已經(jīng)被那個100維的隨機分布限定了，其本質(zhì)維度就是100，再考慮到神經(jīng)網(wǎng)絡帶來的映射降維，最終可能比100還小，所以生成樣本分布的支撐集就在4096維空間中構(gòu)成一個最多100維的低維流形，“撐不滿”整個高維空間。
“撐不滿”就會導致真實分布與生成分布難以“碰到面”，這很容易在二維空間中理解：一方面，二維平面中隨機取兩條曲線，它們之間剛好存在重疊線段的概率為0；另一方面，雖然它們很大可能會存在交叉點，但是相比于兩條曲線而言，交叉點比曲線低一個維度，長度（測度）為0，可忽略。三維空間中也是類似的，隨機取兩個曲面，它們之間最多就是比較有可能存在交叉線，但是交叉線比曲面低一個維度，面積（測度）是0，可忽略。從低維空間拓展到高維空間，就有了如下邏輯：因為一開始生成器隨機初始化，所以
幾乎不可能與
有什么關(guān)聯(lián)，所以它們的支撐集之間的重疊部分要么不存在，要么就比
和
的最小維度還要低至少一個維度，故而測度為0。所謂“重疊部分測度為0”，就是上文所言“不重疊或者重疊部分可忽略”的意思。
我們就得到了WGAN前作中關(guān)于生成器梯度消失的第一個論證：在（近似）最優(yōu)判別器下，最小化生成器的loss等價于最小化
與
之間的JS散度，而由于
與
幾乎不可能有不可忽略的重疊，所以無論它們相距多遠JS散度都是常數(shù)
，最終導致生成器的梯度（近似）為0，梯度消失。
接著作者寫了很多公式定理從第二個角度進行論證，但是背后的思想也可以直觀地解釋：
首先，
與
之間幾乎不可能有不可忽略的重疊，所以無論它們之間的“縫隙”多狹小，都肯定存在一個最優(yōu)分割曲面把它們隔開，最多就是在那些可忽略的重疊處隔不開而已。
由于判別器作為一個神經(jīng)網(wǎng)絡可以無限擬合這個分隔曲面，所以存在一個最優(yōu)判別器，對幾乎所有真實樣本給出概率1，對幾乎所有生成樣本給出概率0，而那些隔不開的部分就是難以被最優(yōu)判別器分類的樣本，但是它們的測度為0，可忽略。
最優(yōu)判別器在真實分布和生成分布的支撐集上給出的概率都是常數(shù)（1和0），導致生成器的loss梯度為0，梯度消失。
有了這些理論分析，原始GAN不穩(wěn)定的原因就徹底清楚了：判別器訓練得太好，生成器梯度消失，生成器loss降不下去；判別器訓練得不好，生成器梯度不準，四處亂跑。只有判別器訓練得不好不壞才行，但是這個火候又很難把握，甚至在同一輪訓練的前后不同階段這個火候都可能不一樣，所以GAN才那么難訓練。
實驗輔證如下：
WGAN前作Figure 2。先分別將DCGAN訓練1，20，25個epoch，然后固定生成器不動，判別器重新隨機初始化從頭開始訓練，對于第一種形式的生成器loss產(chǎn)生的梯度可以打印出其尺度的變化曲線，可以看到隨著判別器的訓練，生成器的梯度均迅速衰減。注意y軸是對數(shù)坐標軸。
一句話概括：最小化第二種生成器loss函數(shù)，會等價于最小化一個不合理的距離衡量，導致兩個問題，一是梯度不穩(wěn)定，二是collapse mode即多樣性不足。WGAN前作又是從兩個角度進行了論證，下面只說第一個角度，因為對于第二個角度我難以找到一個直觀的解釋方式，感興趣的讀者還是去看論文吧（逃）。
如前文所說，Ian Goodfellow提出的“- log D trick”是把生成器loss改成
（公式3）
上文推導已經(jīng)得到在最優(yōu)判別器
下
（公式9）
我們可以把KL散度（注意下面是先g后r）變換成含的形式：
（公式10）
由公式3，9，10可得最小化目標的等價變形
注意上式最后兩項不依賴于生成器G，最終得到最小化公式3等價于最小化
（公式11）
這個等價最小化目標存在兩個嚴重的問題。第一是它同時要最小化生成分布與真實分布的KL散度，卻又要最大化兩者的JS散度，一個要拉近，一個卻要推遠！這在直觀上非?；闹?，在數(shù)值上則會導致梯度不穩(wěn)定，這是后面那個JS散度項的毛病。
第二，即便是前面那個正常的KL散度項也有毛病。因為KL散度不是一個對稱的衡量，
與
是有差別的。以前者為例
當
而時
，
，
對貢獻趨近0
當
而
時，
，
對貢獻趨近正無窮
換言之，
對于上面兩種錯誤的懲罰是不一樣的，第一種錯誤對應的是“生成器沒能生成真實的樣本”，懲罰微小；第二種錯誤對應的是“生成器生成了不真實的樣本” ，懲罰巨大。第一種錯誤對應的是缺乏多樣性，第二種錯誤對應的是缺乏準確性。這一放一打之下，生成器寧可多生成一些重復但是很“安全”的樣本，也不愿意去生成多樣性的樣本，因為那樣一不小心就會產(chǎn)生第二種錯誤，得不償失。這種現(xiàn)象就是大家常說的collapse mode。
第一部分小結(jié)：在原始GAN的（近似）最優(yōu)判別器下，第一種生成器loss面臨梯度消失問題，第二種生成器loss面臨優(yōu)化目標荒謬、梯度不穩(wěn)定、對多樣性與準確性懲罰不平衡導致mode collapse這幾個問題。
實驗輔證如下：
WGAN前作Figure 3。先分別將DCGAN訓練1，20，25個epoch，然后固定生成器不動，判別器重新隨機初始化從頭開始訓練，對于第二種形式的生成器loss產(chǎn)生的梯度可以打印出其尺度的變化曲線，可以看到隨著判別器的訓練，藍色和綠色曲線中生成器的梯度迅速增長，說明梯度不穩(wěn)定，紅線對應的是DCGAN相對收斂的狀態(tài)，梯度才比較穩(wěn)定。
原始GAN問題的根源可以歸結(jié)為兩點，一是等價優(yōu)化的距離衡量（KL散度、JS散度）不合理，二是生成器隨機初始化后的生成分布很難與真實分布有不可忽略的重疊。
WGAN前作其實已經(jīng)針對第二點提出了一個解決方案，就是對生成樣本和真實樣本加噪聲，直觀上說，使得原本的兩個低維流形“彌散”到整個高維空間，強行讓它們產(chǎn)生不可忽略的重疊。而一旦存在重疊，JS散度就能真正發(fā)揮作用，此時如果兩個分布越靠近，它們“彌散”出來的部分重疊得越多，JS散度也會越小而不會一直是一個常數(shù)，于是（在第一種原始GAN形式下）梯度消失的問題就解決了。在訓練過程中，我們可以對所加的噪聲進行退火（annealing），慢慢減小其方差，到后面兩個低維流形“本體”都已經(jīng)有重疊時，就算把噪聲完全拿掉，JS散度也能照樣發(fā)揮作用，繼續(xù)產(chǎn)生有意義的梯度把兩個低維流形拉近，直到它們接近完全重合。以上是對原文的直觀解釋。
在這個解決方案下我們可以放心地把判別器訓練到接近最優(yōu)，不必擔心梯度消失的問題。而當判別器最優(yōu)時，對公式9取反可得判別器的最小loss為
其中
和
分別是加噪后的真實分布與生成分布。反過來說，從最優(yōu)判別器的loss可以反推出當前兩個加噪分布的JS散度。兩個加噪分布的JS散度可以在某種程度上代表兩個原本分布的距離，也就是說可以通過最優(yōu)判別器的loss反映訓練進程！……真的有這樣的好事嗎？
并沒有，因為加噪JS散度的具體數(shù)值受到噪聲的方差影響，隨著噪聲的退火，前后的數(shù)值就沒法比較了，所以它不能成為
和
距離的本質(zhì)性衡量。
因為本文的重點是WGAN本身，所以WGAN前作的加噪方案簡單介紹到這里，感興趣的讀者可以閱讀原文了解更多細節(jié)。加噪方案是針對原始GAN問題的第二點根源提出的，解決了訓練不穩(wěn)定的問題，不需要小心平衡判別器訓練的火候，可以放心地把判別器訓練到接近最優(yōu)，但是仍然沒能夠提供一個衡量訓練進程的數(shù)值指標。但是WGAN本作就從第一點根源出發(fā)，用Wasserstein距離代替JS散度，同時完成了穩(wěn)定訓練和進程指標的問題！
作者未對此方案進行實驗驗證。
Wasserstein距離又叫Earth-Mover（EM）距離，定義如下：
（公式12）
解釋如下：
是
和
組合起來的所有可能的聯(lián)合分布的集合，反過來說，
中每一個分布的邊緣分布都是
和
。對于每一個可能的聯(lián)合分布
而言，可以從中采樣
得到一個真實樣本
和一個生成樣本
，并算出這對樣本的距離
，所以可以計算該聯(lián)合分布下樣本對距離的期望值。在所有可能的聯(lián)合分布
中能夠?qū)@個期望值
取到的下界
，就定義為Wasserstein距離。
直觀上可以把
理解為在
這個“路徑規(guī)劃”下把
這堆“沙土”挪到
“位置”所需的“消耗”，而
就是“最優(yōu)路徑規(guī)劃”下的“最小消耗”，所以才叫Earth-Mover（推土機）距離。
Wasserstein距離相比KL散度、JS散度的優(yōu)越性在于，即便兩個分布沒有重疊，Wasserstein距離仍然能夠反映它們的遠近。WGAN本作通過簡單的例子展示了這一點?？紤]如下二維空間中的兩個分布
和
，
在線段AB上均勻分布，
在線段CD上均勻分布，通過控制參數(shù)
可以控制著兩個分布的距離遠近。
此時容易得到（讀者可自行驗證）
（突變）
（突變）
（平滑）
KL散度和JS散度是突變的，要么最大要么最小，Wasserstein距離卻是平滑的，如果我們要用梯度下降法優(yōu)化
這個參數(shù)，前兩者根本提供不了梯度，Wasserstein距離卻可以。類似地，在高維空間中如果兩個分布不重疊或者重疊部分可忽略，則KL和JS既反映不了遠近，也提供不了梯度，但是Wasserstein卻可以提供有意義的梯度。
既然Wasserstein距離有如此優(yōu)越的性質(zhì)，如果我們能夠把它定義為生成器的loss，不就可以產(chǎn)生有意義的梯度來更新生成器，使得生成分布被拉向真實分布嗎？
沒那么簡單，因為Wasserstein距離定義（公式12）中的
沒法直接求解，不過沒關(guān)系，作者用了一個已有的定理把它變換為如下形式
（公式13）
證明過程被作者丟到論文附錄中了，我們也姑且不管，先看看上式究竟說了什么。
首先需要介紹一個概念——Lipschitz連續(xù)。它其實就是在一個連續(xù)函數(shù)
上面額外施加了一個限制，要求存在一個常數(shù)
使得定義域內(nèi)的任意兩個元素
和
都滿足
此時稱函數(shù)
的Lipschitz常數(shù)為
。
簡單理解，比如說
的定義域是實數(shù)集合，那上面的要求就等價于
的導函數(shù)絕對值不超過
。再比如說
就不是Lipschitz連續(xù)，因為它的導函數(shù)沒有上界。Lipschitz連續(xù)條件限制了一個連續(xù)函數(shù)的最大局部變動幅度。
公式13的意思就是在要求函數(shù)
的Lipschitz常數(shù)
不超過
的條件下，對所有可能滿足條件
的取到
的上界，然后再除以
。特別地，我們可以用一組參數(shù)
來定義一系列可能的函數(shù)
，此時求解公式13可以近似變成求解如下形式
（公式14）
再用上我們搞深度學習的人最熟悉的那一套，不就可以把
用一個帶參數(shù)
的神經(jīng)網(wǎng)絡來表示嘛！由于神經(jīng)網(wǎng)絡的擬合能力足夠強大，我們有理由相信，這樣定義出來的一系列
雖然無法囊括所有可能，但是也足以高度近似公式13要求的那個
了。
最后，還不能忘了滿足公式14中
這個限制。我們其實不關(guān)心具體的K是多少，只要它不是正無窮就行，因為它只是會使得梯度變大
倍，并不會影響梯度的方向。所以作者采取了一個非常簡單的做法，就是限制神經(jīng)網(wǎng)絡
的所有參數(shù)
的不超過某個范圍
，比如
，此時所有偏導數(shù)
也不會超過某個范圍，所以一定存在某個不知道的常數(shù)
使得
的局部變動幅度不會超過它，Lipschitz連續(xù)條件得以滿足。具體在算法實現(xiàn)中，只需要每次更新完
后把它clip回這個范圍就可以了。
到此為止，我們可以構(gòu)造一個含參數(shù)
、最后一層不是非線性激活層的判別器網(wǎng)絡
，在限制
不超過某個范圍的條件下，使得
（公式15）
盡可能取到最大，此時
就會近似真實分布與生成分布之間的Wasserstein距離（忽略常數(shù)倍數(shù)
）。注意原始GAN的判別器做的是真假二分類任務，所以最后一層是sigmoid，但是現(xiàn)在WGAN中的判別器
做的是近似擬合Wasserstein距離，屬于回歸任務，所以要把最后一層的sigmoid拿掉。
接下來生成器要近似地最小化Wasserstein距離，可以最小化
，由于Wasserstein距離的優(yōu)良性質(zhì)，我們不需要擔心生成器梯度消失的問題。再考慮到
的第一項與生成器無關(guān)，就得到了WGAN的兩個loss。
（公式16，WGAN生成器loss函數(shù)）
（公式17，WGAN判別器loss函數(shù)）
公式15是公式17的反，可以指示訓練進程，其數(shù)值越小，表示真實分布與生成分布的Wasserstein距離越小，GAN訓練得越好。
WGAN完整的算法流程已經(jīng)貼過了，為了方便讀者此處再貼一遍：
上文說過，WGAN與原始GAN第一種形式相比，只改了四點：
判別器最后一層去掉sigmoid
生成器和判別器的loss不取log
每次更新判別器的參數(shù)之后把它們的絕對值截斷到不超過一個固定常數(shù)c
不要用基于動量的優(yōu)化算法（包括momentum和Adam），推薦RMSProp，SGD也行
前三點都是從理論分析中得到的，已經(jīng)介紹完畢；第四點卻是作者從實驗中發(fā)現(xiàn)的，屬于trick，相對比較“玄”。作者發(fā)現(xiàn)如果使用Adam，判別器的loss有時候會崩掉，當它崩掉時，Adam給出的更新方向與梯度方向夾角的cos值就變成負數(shù)，更新方向與梯度方向南轅北轍，這意味著判別器的loss梯度是不穩(wěn)定的，所以不適合用Adam這類基于動量的優(yōu)化算法。作者改用RMSProp之后，問題就解決了，因為RMSProp適合梯度不穩(wěn)定的情況。
對WGAN作者做了不少實驗驗證，本文只提比較重要的兩點。第一，判別器所近似的Wasserstein距離與生成器的生成圖片質(zhì)量高度相關(guān)，如下所示（此即題圖）：
第二，WGAN如果用類似DCGAN架構(gòu)，生成圖片的效果與DCGAN差不多：
但是厲害的地方在于WGAN不用DCGAN各種特殊的架構(gòu)設(shè)計也能做到不錯的效果，比如如果大家一起拿掉Batch Normalization的話，DCGAN就崩了：
如果WGAN和原始GAN都使用多層全連接網(wǎng)絡（MLP），不用CNN，WGAN質(zhì)量會變差些，但是原始GAN不僅質(zhì)量變得更差，而且還出現(xiàn)了collapse mode，即多樣性不足：
最后補充一點論文沒提到，但是我個人覺得比較微妙的問題。判別器所近似的Wasserstein距離能夠用來指示單次訓練中的訓練進程，這個沒錯；接著作者又說它可以用于比較多次訓練進程，指引調(diào)參，我倒是覺得需要小心些。比如說我下次訓練時改了判別器的層數(shù)、節(jié)點數(shù)等超參，判別器的擬合能力就必然有所波動，再比如說我下次訓練時改了生成器兩次迭代之間，判別器的迭代次數(shù)，這兩種常見的變動都會使得Wasserstein距離的擬合誤差就與上次不一樣。那么這個擬合誤差的變動究竟有多大，或者說不同的人做實驗時判別器的擬合能力或迭代次數(shù)相差實在太大，那它們之間還能不能直接比較上述指標，我都是存疑的。
WGAN前作分析了Ian Goodfellow提出的原始GAN兩種形式各自的問題，第一種形式等價在最優(yōu)判別器下等價于最小化生成分布與真實分布之間的JS散度，由于隨機生成分布很難與真實分布有不可忽略的重疊以及JS散度的突變特性，使得生成器面臨梯度消失的問題；第二種形式在最優(yōu)判別器下等價于既要最小化生成分布與真實分布直接的KL散度，又要最大化其JS散度，相互矛盾，導致梯度不穩(wěn)定，而且KL散度的不對稱性使得生成器寧可喪失多樣性也不愿喪失準確性，導致collapse mode現(xiàn)象。
WGAN前作針對分布重疊問題提出了一個過渡解決方案，通過對生成樣本和真實樣本加噪聲使得兩個分布產(chǎn)生重疊，理論上可以解決訓練不穩(wěn)定的問題，可以放心訓練判別器到接近最優(yōu)，但是未能提供一個指示訓練進程的可靠指標，也未做實驗驗證。
WGAN本作引入了Wasserstein距離，由于它相對KL散度與JS散度具有優(yōu)越的平滑特性，理論上可以解決梯度消失問題。接著通過數(shù)學變換將Wasserstein距離寫成可求解的形式，利用一個參數(shù)數(shù)值范圍受限的判別器神經(jīng)網(wǎng)絡來最大化這個形式，就可以近似Wasserstein距離。在此近似最優(yōu)判別器下優(yōu)化生成器使得Wasserstein距離縮小，就能有效拉近生成分布與真實分布。WGAN既解決了訓練不穩(wěn)定的問題，也提供了一個可靠的訓練進程指標，而且該指標確實與生成樣本的質(zhì)量高度相關(guān)。作者對WGAN進行了實驗驗證。
雷鋒網(wǎng)
相關(guān)文章：
縱覽深度學習技術(shù)前沿，Yoshua Bengio為你解讀如何創(chuàng)造人類水平的AI（附PPT）
Google首席科學家Vincent Vanhoucke：機器人和深度學習正在發(fā)生一些“有趣的融合”| AAAI 2017
雷峰網(wǎng)版權(quán)文章，未經(jīng)授權(quán)禁止轉(zhuǎn)載。詳情見轉(zhuǎn)載須知。
