Flow Learning Based Dual Networks for Low-Light Image Enhancement

Siyu Wang,Changhui Hu,Weilin Yi,Ziyun Cai,Mingliang Zhai,Wankou Yang

DOI: https://doi.org/10.1007/s11063-023-11303-3

IF: 2.565

2023-06-07

Neural Processing Letters

Abstract:The deep learning-based low-light image enhancement task aims to learn a mapping that converts low-light images to normally exposed images by training with paired or unpaired datasets. Most of these existing methods are based on convolutional neural networks, which largely limits the network's ability to learn global information. Meanwhile, the reconstruction loss or adversarial loss they adopt often cannot accurately measure the visual distance between the prediction and the target, resulting in blurred regions in the enhancement results. In this paper, we propose a novel flow learning based dual networks (FDN), which consists of a dual network and a flow learning based model. The dual network is mainly composed of a reidual-based Unet encoder and a residual-based Swin Transformer encoder, which can make up for the lack of global information processing and has more advantages in processing deep and shallow information. Moreover, we use a single loss function named negative log-likelihood to train the entire network, which enables the flow models to adequately learn the complex conditional distribution of normally exposed images and avoid blurry outputs. Experimental results on two benchmark datasets show that the proposed FDN method can achieve the sate-of-the-art performances on low-light image enhancement task.

computer science, artificial intelligence