ExReg: Wide-range Photo Exposure Correction via a Multi-dimensional Regressor with Attention

TL;DR

ExReg is a novel neural network that simultaneously corrects under- and over-exposed images by using multi-dimensional regression and attention mechanisms, achieving superior results efficiently.

Contribution

It introduces a multi-dimensional regression framework with a multi-exposure generator and attentive neural processes for unified exposure correction.

Findings

Outperforms state-of-the-art in PSNR for exposure correction

Generates visually consistent and physically accurate results

Processes images in 0.05 seconds on an RTX 3090

Abstract

Photo exposure correction is widely investigated, but fewer studies focus on correcting under- and over-exposed images simultaneously. Three issues remain open to handle and correct both under- and over-exposed images in a unified way. First, a locally-adaptive exposure adjustment may be more flexible instead of learning a global mapping. Second, it is an ill-posed problem to determine the suitable exposure values locally. Third, photos with the same content but different exposures may not reach consistent adjustment results. To this end, we proposed a novel exposure correction network, ExReg, to address the challenges by formulating exposure correction as a multi-dimensional regression process. Given an input image, a compact multi-exposure generation network is introduced to generate images with different exposure conditions for multi-dimensional regression and exposure correction in the next stage. An auxiliary module is designed to predict the region-wise exposure values, guiding the proposed Encoder-Decoder ANP (Attentive Neural Processes) to regress the final corrected image. The experimental results show that ExReg can generate well-exposed results and outperform the SOTA method in PSNR for extensive exposure problems. Furthermore, the processing speed, with 0.05 seconds per image on an RTX 3090, is efficient. When tested on the same image under various exposure levels, ExReg also yields results that are visually consistent and physically accurate.