Conditional variational underwater image enhancement with kernel decomposition and adaptive hybrid normalization

Abstract

Enhancing underwater images has gained significant interest due to its wide range of applications in ocean engineering and marine robotics. However, underwater images often suffer from ambiguous degradations, making it difficult to construct a deterministic map between distorted underwater images and reference images. In addition, the distortions and artifacts affect the quality of underwater images in local details and global structures. To this end, we propose a novel network based on Conditional Variational Autoencoder (CVAE) for Underwater Image Enhancement, named CVUIE. Specifically, to capture inherent uncertainties in underwater scenes and generate robust enhanced output, we propose a novel network structure that combines the CVAE with adversarial learning. Then, we present a Kernel Decomposition Attention (KDA) module to process and enhance features over a broader respective field. Moreover, to balance the complex details and global structures of enhanced images, we design a Probabilistic Adaptive Hybrid Normalization (PAHN) module. Evaluations conducted on multiple benchmark datasets prove that the proposed network performs qualitatively and quantitatively better than existing state-of-the-art methods. Real-world experiments have also demonstrated the promising future application prospects of our method.