AQUA-Net: Adaptive Frequency Fusion and Illumination Aware Network for Underwater Image Enhancement

TL;DR

AQUA-Net is a novel deep learning model that enhances underwater images by combining frequency and illumination domain processing, effectively restoring color, contrast, and details while being computationally efficient.

Contribution

The paper introduces AQUA-Net, a new underwater image enhancement network that integrates frequency fusion and illumination awareness, improving image quality with fewer parameters.

Findings

Achieves state-of-the-art performance on benchmark datasets.

Effectively restores color balance and contrast in underwater images.

Demonstrates strong generalization and robustness in real-world conditions.

Abstract

Underwater images often suffer from severe color distortion, low contrast, and a hazy appearance due to wavelength-dependent light absorption and scattering. Simultaneously, existing deep learning models exhibit high computational complexity, which limits their practical deployment for real-time underwater applications. To address these challenges, this paper presents a novel underwater image enhancement model, called Adaptive Frequency Fusion and Illumination Aware Network (AQUA-Net). It integrates a residual encoder decoder with dual auxiliary branches, which operate in the frequency and illumination domains. The frequency fusion encoder enriches spatial representations with frequency cues from the Fourier domain and preserves fine textures and structural details. Inspired by Retinex, the illumination-aware decoder performs adaptive exposure correction through a learned illumination map that separates reflectance from lighting effects. This joint spatial, frequency, and illumination design enables the model to restore color balance, visual contrast, and perceptual realism under diverse underwater conditions. Additionally, we present a high-resolution, real-world underwater video-derived dataset from the Mediterranean Sea, which captures challenging deep-sea conditions with realistic visual degradations to enable robust evaluation and development of deep learning models. Extensive experiments on multiple benchmark datasets show that AQUA-Net performs on par with SOTA in both qualitative and quantitative evaluations while using less number of parameters. Ablation studies further confirm that the frequency and illumination branches provide complementary contributions that improve visibility and color representation. Overall, the proposed model shows strong generalization capability and robustness, and it provides an effective solution for real-world underwater imaging applications.