SRGAN With WGAN：讓超解析度演算法訓練更穩定 | 附開原始碼
作者丨胡智豪
單位 | 棒谷網路科技影象工程師
經歷丨實現過電商大規模以圖搜圖、超解析度等專案
寫在前面
此文挖坑了很久，專案本在 7 個月前已經結束，但一直沒心思把程式碼整理出來，後來發現有相同思路的團隊把它寫成論文佔坑了，也就更沒動力寫了。昨晚發現我的 Github 專案竟然有星星，感受到了莫大的支援，忽然燃起了寫文章的動力，於是就有了現在這篇。
SRGAN
SRGAN，2017 年 CVPR 中備受矚目的超解析度論文，把超解析度的效果帶到了一個新的高度，而 2017 年超分大賽 NTIRE 的冠軍 EDSR 也是基於 SRGAN 的變體。
▲ 圖片源自原論文
▲ 圖片源自原論文
SRGAN 是基於 GAN 方法進行訓練的
▲ 圖片源自原論文
GAN存在的問題
傳統 GAN 存在一個問題是，你無法知道什麼時候該停止訓練 Generator，什麼時候該停止訓練 Discriminator，如果過度訓練 Discriminator，Generator 就無法學習下去，反之也會導致模型效果差。
如果能有一個 loss 指標來反映訓練情況的話，訓練的難度就會大大降低。而 17 年提出的 WGAN [3] 正是解決這一問題的重要方法。
WGAN 使用 Wasserstein 距離來描述兩個資料集分佈之間的差異程度，只要把模型修改成 WGAN 的形式，就能根據一個唯一的 loss 來監控模型訓練的程度。有關 WGAN 的解釋強烈推薦大家閱讀這篇文章：令人拍案叫絕的 Wasserstein GAN [4]，作者用非常直白明瞭的語言介紹 WGAN。
SRGAN結合WGAN
SRGAN 的一個超讚復現來自 @董豪 之手，他也是 tensorlayer 的作者之一，他的復現專案在 Github 上得到了大量的星星，而筆者的程式碼就正是在他的基礎上進行拓展，首先非常感謝作者的開源。
· 判別器最後一層去掉 sigmoid
· 生成器和判別器的 loss 不取 log
· 每次更新判別器的引數之後把它們的絕對值截斷到不超過一個固定常數 c
· 不要用基於動量的優化演算法（包括 momentum 和 Adam），推薦 RMSProp，SGD 也行
--來自《令人拍案叫絕的Wasserstein GAN》
根據這篇文章的介紹，對程式碼進行以上四項修改，把 GAN 的訓練方式轉換成 WGAN，可以在 tensorboard 中監控 loss 的下降情況。因此，筆者對專案原作的進行了一些修改：
1. 對模型程式碼進行了上面 WGAN 的改造；
2. 增加了 Tensorboard，監控 loss 下降情況；
3. 對作者的 model.py 中，Generator 的最後一層的卷積 kernel 從 1×1 改成 9×9，這是原論文建議的結構。
SRGAN With Wasserstein GAN 的完整程式碼已經開源到筆者的 Github[5]，裡面有詳細的改動介紹和使用說明，覺得好的話就給個 Star 呀！
▲ 修改後的訓練Loss下降情況
下面是一些復現後的超分復原效果：
一個來自工業界的問題
在實際生產使用中，遇到的低解析度圖片並不一定都是 PNG 格式的（無失真壓縮的圖片復原效果最好），而且會帶有不同程度的失真（有失真壓縮導致的 artifacts）。筆者嘗試過很多演算法，例如 SRGAN、EDSR、RAISR、Fast Neural Style 等等，這類圖片目前使用任何一種超分演算法都沒法在提高解析度的同時消除失真。
這個問題我在 @董豪 SRGAN 專案的 issue 中也討論過，同時在知乎也提出過這個問題：SRGAN 超解析度方法對於低清 jpg 格式的圖片復原效果是否比不上對低清 png 格式的復原效果？
可惜沒有很好的答案。目前學術界貌似還沒有很好的演算法，這裡歡迎各位在評論區或者 Github 上來討論。
參考文獻 & 相關連結
[1]. Photo-Realistic Single Image Super-Resolution Using a Generative Adversarial Network
https://www.paperweekly.site/papers/664
[2]. Is the deconvolution layer the same as a convolutional layer?
https://arxiv.org/abs/1609.07009
[3]. Wasserstein GAN
https://www.paperweekly.site/papers/808
[4]. 令人拍案叫絕的Wasserstein GAN
https://zhuanlan.zhihu.com/p/25071913
[5]. 本專案倉庫：SRGAN_Wasserstein
http://github.com/JustinhoCHN/SRGAN_Wasserstein
[6]. Real-Time Single Image and Video Super-Resolution Using an Efficient Sub-Pixel Convolutional Neural Network
https://www.paperweekly.site/papers/386
點選以下標題檢視其他文章：
▲戳我檢視招募詳情
#作 者 招 募#
關於PaperWeekly
PaperWeekly 是一個推薦、解讀、討論、報道人工智慧前沿論文成果的學術平臺。如果你研究或從事 AI 領域，歡迎在公眾號後臺點選「交流群」，小助手將把你帶入 PaperWeekly 的交流群裡。
▽ 點選 | 閱讀原文 | 進入作者知乎專欄