Approaching the Limit of Image Rescaling via Flow Guidance

TL;DR

This paper introduces an invertible flow guidance module to improve image rescaling, enabling better joint optimization of downscaling and upscaling for superior reconstruction quality.

Contribution

It proposes a novel invertible flow guidance module that replaces traditional LR guidance, allowing end-to-end optimization of image rescaling modules for enhanced performance.

Findings

Achieves state-of-the-art results in image reconstruction quality.

Enables end-to-end training of downscaling and upscaling modules.

Outperforms existing methods on multiple benchmarks.

Abstract

Image downscaling and upscaling are two basic rescaling operations. Once the image is downscaled, it is difficult to be reconstructed via upscaling due to the loss of information. To make these two processes more compatible and improve the reconstruction performance, some efforts model them as a joint encoding-decoding task, with the constraint that the downscaled (i.e. encoded) low-resolution (LR) image must preserve the original visual appearance. To implement this constraint, most methods guide the downscaling module by supervising it with the bicubically downscaled LR version of the original high-resolution (HR) image. However, this bicubic LR guidance may be suboptimal for the subsequent upscaling (i.e. decoding) and restrict the final reconstruction performance. In this paper, instead of directly applying the LR guidance, we propose an additional invertible flow guidance module (FGM), which can transform the downscaled representation to the visually plausible image during downscaling and transform it back during upscaling. Benefiting from the invertibility of FGM, the downscaled representation could get rid of the LR guidance and would not disturb the downscaling-upscaling process. It allows us to remove the restrictions on the downscaling module and optimize the downscaling and upscaling modules in an end-to-end manner. In this way, these two modules could cooperate to maximize the HR reconstruction performance. Extensive experiments demonstrate that the proposed method can achieve state-of-the-art (SotA) performance on both downscaled and reconstructed images.