41: Real-ESRGAN Explained

Real-ESRGAN: Training Real-World Blind Super-Resolution with Pure Synthetic Data by Xintao Wang et al. explained in 5 minutes.

โญ๏ธPaper difficulty: ๐ŸŒ•๐ŸŒ•๐ŸŒ‘๐ŸŒ‘๐ŸŒ‘

Real-ESRGAN: Training Real-World Blind Super-Resolution with Pure Synthetic Data Samples

๐ŸŽฏ At a glance:

While there are many blind image restoration approaches, few can handle complex real-world degradations. Yet Real-ESRGAN by Xintao Wang and his colleagues from ARC, Tencent PCG, Shenzen Institutes, and University of Chinese Academy of Sciences takes real-world image super-resolution (SR) to the next level! The authors propose a new higher-order image degradation model to better simulate real-world data. This idea together with an improved U-Net discriminator allows Real-ESRGAN to demonstrate superior visual performance than prior works on various real datasets.

โŒ›๏ธ Prerequisites:

(Highly recommended reading to understand the core ideas in this paper):

๐Ÿš€ Motivation:

Classical degradation model, which consists of blur, downsampling, noise and JPEG compression is not complex enough to model real-world degradations. Models trained on these synthetic samples will easily fail on real-world tests. The goal of this work is to extend blind SR trained on synthetic data to work on real-world images at inference time. Hence, a more sophisticated degradation model called second-order degradation process is introduced. To compensate for the larger degradation space the VGG-style discriminator is upgraded to a U-Net design. Additionally, spectral normalization (SN) regularization is applied to stabilize training,

๐Ÿ” Main Ideas:

1) Classical Degradation Model: Typically the distortion model consists of a gaussian blur followed by a downsampling operation, additive Poisson noise to simulate camera sensor artifacts. Additionally, differentiable JPEG compression is used for lossy compression and introduces unpleasing block artifacts.

2) Higher-order Degradation Model:
A second-order degradation process is โ€ฆ the classical degradation process applied to an image twice in succession.

3) Ringing and Overshoot Artifacts:
Ringing artifacts often appear as spurious edges near sharp transitions in an image. They visually look like bands or โ€œghostsโ€ near edges. Overshoot artifacts are usually combined with ringing artifacts, which manifest themselves as an increased jump at the edge transition. The main cause of these artifacts is that the signal is bandlimited without high frequencies. A sinc filter is employed to cut off high frequencies to synthesize ringing and overshoot artifacts for training pairs. It is used in the blurring process as well as the last step of the synthesis, where its order is randomly swapped with JPEG compression.

4) Networks and Training:
The generator is the same as ESRGAN with the addition of pixel-unshuffle to reduce the spatial size and increase channel size before the main ESRGAN network. A U-Net discriminator is used in combination with SN regularization for more discriminative power with more stable training dynamics.

๐Ÿ“ˆ Experiment insights / Key takeaways:
  • Real-ESRGAN is trained on 256x256 patches from DIV2K, Flickr2K, and OutdoorSceneTraining datasets
  • It is fine-tuned from ESRGAN with a batch size of 48 for 500k iterations with L1, perceptual, and GAN losses
  • Considered baseline state-of-the-art methods: ESRGAN, DAN, CDC, RealSR, and BSRGAN
  • The baselines are nowhere near Real-ESRGAN in qualitative comparison
  • Real-ESRGAN trained with classical first-order degradation model cannot effectively remove noise or blur
  • If sinc filters are not employed during training, the restored results will amplify the ringing and overshoot artifacts that existed in the input images, especially around the text and lines
  • Most failure cases are supposedly connected with aliasing issues

๐Ÿ–ผ๏ธ Paper Poster:

Real-ESRGAN: Training Real-World Blind Super-Resolution with Pure Synthetic Data paper poster

โœ๏ธMy Notes:
  • Real-ESRGAN is as generic of a name as they get: 2/5
  • I am going to bet that there will be a paper that uses ideas from both ESRGAN and Alias-free GAN soon
  • What do you think about Real-ESRGAN? Write your response in the comments!

Real-ESRGAN arxiv / Real-ESRGAN github

๐Ÿ‘‹ Thanks for reading!

Join Patreon for Exclusive Perks!

If you found this paper digest useful, subscribe and share the post with your friends and colleagues to support Casual GAN Papers!

Join the Casual GAN Papers telegram channel to stay up to date with new AI Papers!

Discuss the paper

By: @casual_gan

P.S. Send me paper suggestions for future posts @KirillDemochkin!