DI-Retinex: Digital-Imaging Retinex Theory for Low-Light Image Enhancement
Shangquan Sun, Wenqi Ren, Jingyang Peng, Fenglong Song, Xiaochun Cao
TL;DR
This work extends Retinex theory for LLIE to account for digital-imaging distortions by proposing Digital-Imaging Retinex (DI-Retinex), which introduces a non-zero-mean offset term $\boldsymbol{\beta}$ in a pixel-wise linear enhancement model $\mathbf{I}_h \approx \boldsymbol{\alpha} \mathbf{I}_l + \boldsymbol{\beta}$. It develops a zero-shot enhancement pipeline using a lightweight three-convolutional-layer network to predict a and b in a two-parameter contrast-brightness mapping, guided by a masked reverse degradation loss in Gamma space and a variance suppression loss to curb variance amplification. The method achieves state-of-the-art performance among unsupervised/zero-shot LLIE methods on LOL-v1 and LOL-v2 while maintaining small model size and fast inference, and it improves downstream face-detection in low light. The approach offers a practical, lightweight preprocessing step for real-time imaging and downstream vision tasks in challenging lighting conditions, with clear theoretical grounding in the extended DI-Retinex framework.
Abstract
Many existing methods for low-light image enhancement (LLIE) based on Retinex theory ignore important factors that affect the validity of this theory in digital imaging, such as noise, quantization error, non-linearity, and dynamic range overflow. In this paper, we propose a new expression called Digital-Imaging Retinex theory (DI-Retinex) through theoretical and experimental analysis of Retinex theory in digital imaging. Our new expression includes an offset term in the enhancement model, which allows for pixel-wise brightness contrast adjustment with a non-linear mapping function. In addition, to solve the lowlight enhancement problem in an unsupervised manner, we propose an image-adaptive masked reverse degradation loss in Gamma space. We also design a variance suppression loss for regulating the additional offset term. Extensive experiments show that our proposed method outperforms all existing unsupervised methods in terms of visual quality, model size, and speed. Our algorithm can also assist downstream face detectors in low-light, as it shows the most performance gain after the low-light enhancement compared to other methods.
