Abstract:Recent advances of deep neural networks (DNNs) promote low-level vision applications in real-world scenarios, e.g. , image enhancement, dehazing. Nevertheless, DNN-based methods encounter challenges in terms of high computational and memory requirements, especially when deployed on real-world devices with limited resources. Quantization is one of effective compression techniques that significantly reduces computational and memory requirements by employing low-bit parameters and bit-wise operations. However, low-bit quantization for computational imaging ( Q-Imaging ) remains largely unexplored and usually suffer from a significant performance drop compared with the real-valued counterparts. In this work, through empirical analysis, we identify the main factor responsible for such significant performance drop underlies in the large gradient estimation error from non-differentiable weight quantization methods, and the activation information degeneration along with the activation quantization. To address these issues, we introduce a differentiable quantization search (DQS) method to learn the quantized weights and an information boosting module (IBM) for network activation quantization. Our DQS method allows us to treat the discrete weights in a quantized neural network as variables that can be searched. We achieve this end by using a differential approach to accurately search for these weights. In specific, each weight is represented as a probability distribution across a set of discrete values. During training, these probabilities are optimized, and the values with the highest probabilities are chosen to construct the desired quantized network. Moreover, our IBM module can rectify the activation distribution before quantization to maximize the self-information entropy, which retains the maximum information during the quantization process. Extensive experiments across a range of image processing tasks, including enhancement, super-resolution, denoising and dehazing, validate the effectiveness of our Q-Imaging along with superior performances compared to a variety of state-of-the-art quantization methods. In particular, the method in Q-Imaging also achieves a strong generalization performance when composing a detection network for the dark object detection task.

I $^2$ NQ: Inter and Intra Nonuniform Quantization for Single Image Super-Resolution

2DQuant: Low-bit Post-Training Quantization for Image Super-Resolution

QuantSR: Accurate Low-bit Quantization for Efficient Image Super-Resolution

Fully Quantized Image Super-Resolution Networks

Nonuniform-to-Uniform Quantization: Towards Accurate Quantization via Generalized Straight-Through Estimation

RefQSR: Reference-based Quantization for Image Super-Resolution Networks

Local Means Binary Networks for Image Super-Resolution

Distribution-Flexible Subset Quantization for Post-Quantizing Super-Resolution Networks.

QuantNAS for super resolution: searching for efficient quantization-friendly architectures against quantization noise

Learning Accurate Low-bit Quantization towards Efficient Computational Imaging

A Multi-precision Quantized Super-Resolution Model Framework

Hybrid Post-Training Quantization for Super-Resolution Neural Network Compression

MWQ: Multiscale Wavelet Quantized Neural Networks

Dynamic Dual Trainable Bounds for Ultra-low Precision Super-Resolution Networks

Weighted Quantization and Hamming Search for Fast Image Super-Resolution

HQ-IRN: Quantizing High-frequency Features for Image Rescaling

Efficient Super-Resolution System with Block-Wise Hybridization and Quantized Winograd on FPGA

Overcoming Distribution Mismatch in Quantizing Image Super-Resolution Networks

Outlier-Aware Training for Low-Bit Quantization of Structural Re-Parameterized Networks

AUSN: Approximately Uniform Quantization by Adaptively Superimposing Non-uniform Distribution for Deep Neural Networks