Overcoming Distribution Mismatch in Quantizing Image Super-Resolution Networks

TL;DR

This paper introduces a quantization-aware training method for image super-resolution networks that effectively addresses distribution mismatch without increasing inference complexity, achieving state-of-the-art results.

Contribution

It proposes a novel training scheme that regularizes feature distributions only when aligned with reconstruction loss, avoiding dynamic range adaptation during inference.

Findings

Reduces distribution mismatch in SR networks

Achieves state-of-the-art super-resolution accuracy

Maintains minimal computational overhead

Abstract

Although quantization has emerged as a promising approach to reducing computational complexity across various high-level vision tasks, it inevitably leads to accuracy loss in image super-resolution (SR) networks. This is due to the significantly divergent feature distributions across different channels and input images of the SR networks, which complicates the selection of a fixed quantization range. Existing works address this distribution mismatch problem by dynamically adapting quantization ranges to the varying distributions during test time. However, such a dynamic adaptation incurs additional computational costs during inference. In contrast, we propose a new quantization-aware training scheme that effectively Overcomes the Distribution Mismatch problem in SR networks without the need for dynamic adaptation. Intuitively, this mismatch can be mitigated by regularizing the distance between the feature and a fixed quantization range. However, we observe that such regularization can conflict with the reconstruction loss during training, negatively impacting SR accuracy. Therefore, we opt to regularize the mismatch only when the gradients of the regularization are aligned with those of the reconstruction loss. Additionally, we introduce a layer-wise weight clipping correction scheme to determine a more suitable quantization range for layer-wise weights. Experimental results demonstrate that our framework effectively reduces the distribution mismatch and achieves state-of-the-art performance with minimal computational overhead.