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

Munsif Ali; Najmul Hassan; Lucia Ventura; Davide Di Bari; Simonepietro Canese; Jungpil Shin

cs.CVDec 5, 2025

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

Authors: Munsif Ali, Najmul Hassan, Lucia Ventura, Davide Di Bari, Simonepietro Canese, Jungpil Shin

Organizations: Stazione Zoologica Anton Dohrn, Villa Comunale, 80121 Napoli, Italy. · School of Computer Science and Engineering, The University of Aizu, Aizuwakamatsu, Japan.

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.

Figures & tables

Panel of Figure 1 — Figure 1: Component-wise evaluation of AQUA-Net on an underwater image: (a) raw input, (b) encoder–decoder baseline, (c) encoder–decoder with an added frequency branch, and (d) full AQUA-Net combining global correction, frequency-guided enhancement, and illumination estimation.

Figure 2: AQUA-Net architecture. The network combines a frequency enhancement block, an illumination branch, and a multi-level encoder–decoder backbone with Residual Enhancement Modules (REMs). The frequency block output fuses with the encoder input. Skip connections integrate illumination information to enhance feature refinement and restore underwater image quality.

Figure 3: Residual Enhancement Module (REM) structure. The REM consists of two depthwise separable convolution blocks followed by a point-wise convolution with Leaky ReLU activation only in the first stage. Each block performs element-wise addition with its input to form a residual connection.

Methods	PSNR $\uparrow$	SSIM $\uparrow$	UIQM $\uparrow$	UCIQE $\uparrow$
Raw	16.134	0.748	2.346	0.362
Fusion(TIP’17) Ancuti et al. (2017)	18.033	0.861	2.684	0.406
SMBL(TB’20) Song et al. (2020)	16.513	0.781	2.167	0.455
MLLE(TIP’22) Zhang et al. (2022)	18.727	0.790	2.305	0.468
UWCNN(PR’20) Li et al. (2020)	18.147	0.847	2.878	0.357
WaterNet(TIP’19) Li et al. (2019)	19.914	0.859	2.846	0.410

Table 1: Quantitative evaluation on the UEIB dataset using PSNR, SSIM, UIQM, and UCIQE metrics. The top three results are shown in red , green , and blue .

Methods	UIEB-C60		EUVP-T515		RUIE-T78		DeapSea-T80
	UIQM $\uparrow$	UCIQE $\uparrow$	UIQM $\uparrow$	UCIQE $\uparrow$	UIQM $\uparrow$	UCIQE $\uparrow$	UIQM $\uparrow$	UCIQE $\uparrow$
Raw	1.856	0.359	2.217	0.417	2.437	0.321	2.035	0.247
Fusion Ancuti et al. (2017)	2.163	0.378	2.636	0.430	2.772	0.366	2.698	0.393
SMBL Song et al. (2020)	1.724	0.439	1.857	0.513	2.459	0.431	2.407	0.382
MLLE Zhang et al. (2022)	1.956	0.464	2.354	0.461	2.798	0.441	2.843	0.436
WWPF Zhang et al. (2023)	2.1304	0.3943	2.3941	0.4695	2.7388	0.3726	2.485	0.386

Table 2: Quantitative comparison on UIEB-C60, EUVP-T515, RUIE-T78, and DeepSea-T80 in terms of UIQM and UCIQE. The top three scores are marked in red , green , and blue .

Methods	FLOPs (G) $\downarrow$	Parameters (M) $\downarrow$
Fusion (TIP’18) Ancuti et al. (2017)	–	–
SMBL (TB’20) Song et al. (2020)	–	–
MLLE (TIP’22) Zhang et al. (2022)	–	–
UWCNN (PR’20) Li et al. (2020)	11.36	0.04
WaterNet (TIP’19) Li et al. (2019)	310.82	1.09
PUIE (ECCV’22) Fu et al. (2022)	2073.2	0.83

Table 3: Comparison of different UIE models in terms of FLOPs (G) and parameter counts (M). The three lowest results are highlighted in red , green , and blue , respectively.

Panel of Figure 4 — Figure 4: Comparison on the UEIB-T90 dataset showing two samples with greenish and bluish color casts. AQUA-Net effectively corrects the color casts while preserving structural details and natural appearance.

Panel of Figure 5 — Figure 5: Comparison of the UEIB C60 challenging UIWs. AQUA-Net restores colors and maintains visual consistency.

Panel of Figure 6 — Figure 6: Comparison on the RUIE-T78 underwater image subset with strong greenish degradation. AQUA-Net removes the greenish color cast and clearly restores object visibility, including the black sea urchin.

Panel of Figure 7 — Figure 7: Qualitative comparison on the proposed DeepSea-T80 real deep-sea dataset. Four representative images with different color casts are enhanced by classical and DL-based UIE methods, as well as the proposed AQUA-Net. In these examples, AQUA-Net better preserves color balance and illumination and reduces reddish artifacts, resulting in a clearer separation between foreground structures and the background.

Modules	UIEB-T90				UIEB-C60
	PSNR $\uparrow$	SSIM $\uparrow$	UIQM $\uparrow$	UCIQE $\uparrow$	UIQM $\uparrow$	UCIQE $\uparrow$
Raw	16.134	0.748	2.346	0.362	1.856	0.359
Base	18.473	0.832	2.872	0.377	1.602	0.418
Base $+$ Frequency	20.614	0.872	3.089	0.415	2.233	0.436
Base $+$ Illumination	20.730	0.879	3.086	0.391	2.123	0.414
Full Model	21.482	0.890	3.250	0.397	2.313	0.427

Table 4: Quantitative results of the ablation study. The table shows the contribution of each component in the AQUA-Net model. The top scores are marked in red .

Panel of Figure 8 — Figure 8: Visual results of the ablation study. The frequency block improves object visibility, and the illumination block enhances overall scene clarity in the AQUA-Net model.

Panel of Figure 9 — Figure 9: Qualitative comparison on the LSUI-400 test set using different underwater image enhancement methods. The comparison includes the raw input image, reference image, and enhanced results obtained using Fusion, WWPF, MLLE, WFAC, WaterNet, LMF, OUNet-JL, and AQUA-Net.

Methods	PSNR $\uparrow$	SSIM $\uparrow$
Fusion Ancuti et al. (2017)	20.13	0.8294
MLLE Zhang et al. (2022)	16.63	0.6491
WFAC Zhang et al. (2025a)	15.85	0.6665
WWPF Zhang et al. (2023)	16.97	0.6622
WaterNet Li et al. (2019)	22.75	0.8583
LMF Zhou et al. (2025)	21.44	0.8547

Table 5: Comparison with SOTA methods on the LSUI-400 test dataset. The top three results are shown in red , green , and blue , respectively.

Enhancement Method	Precision	Recall	[email protected]	[email protected]:0.95
CCL-Net Liu et al. (2024)	0.856	0.768	0.848	0.658
OUNet-JL Wang et al. (2025)	0.857	0.730	0.823	0.618
AQUA-Net	0.842	0.774	0.828	0.634

Table 6: Downstream object detection comparison of different underwater image enhancement methods on the DUO dataset using YOLO12s.

Panel of Figure 10 — Figure 10: Assessment of underwater object detection on the DUO dataset using different enhancement methods with the YOLO12s detection algorithm. The visual results show that AQUA-Net-enhanced images support reliable underwater object detection compared with alternative enhancement approaches.

Image Enhancement

Image enhancement restores visibility, color, detail, and dynamic range in degraded images for human viewing and downstream vision tasks.

See all 149 papers

Advancing Visual Reliability: Color-Accurate Underwater Image Enhancement for Real-Time Underwater Missions
Mar 17, 2026Yiqiang Zhou, Yifan Chen, Zhe Sun +3Lightweight Neural Networks Image Enhancement
Institute of Artificial Intelligence (TeleAI), China Telecom, P. R. China · College of Computer and Control Engineering, Northeast Forestry University, Harbin, 150040, China · College of Future Information Technology, Fudan University, Shanghai 200433, China +2
Real-Time Underwater Image Enhancement via Frequency-Guided Dual-Path Attention
Jun 29, 2026Leshen Zhang, Ao Li, Ce ZhuImage Enhancement Underwater Image Enhancement
Glasgow College, UESTC, Chengdu, China · University of Glasgow, Glasgow, UK · School of Information and Communication Engineering, UESTC, Chengdu, China
Hero-Mamba: Mamba-based Dual Domain Learning for Underwater Image Enhancement
Apr 17, 2026Tejeswar Pokuri, Shivarth RaiUnderwater Image Enhancement Underwater Imaging
Manipal Institute of Technology, Manipal Academy of Higher Education, Manipal - 576104, India

Mar 17, 2026J/K move · Enter open · S save

Advancing Visual Reliability: Color-Accurate Underwater Image Enhancement for Real-Time Underwater Missions

Yiqiang Zhou, Yifan Chen, Zhe Sun, Jijun Lu+2

Institute of Artificial Intelligence (TeleAI), China Telecom, P. R. China · College of Computer and Control Engineering, Northeast Forestry University, Harbin, 150040, China · College of Future Information Technology, Fudan University, Shanghai 200433, China+2

Lightweight Neural Networks Image Enhancement Underwater Image Enhancement

Open paper AskPDF arXiv

Underwater image enhancement plays a crucial role in providing reliable visual information for underwater platforms, since strong absorption and scattering in water-related environments generally lead to image quality degradation. Existing high-performance methods often rely on complex architectures, which hinder deployment on underwater devices. Lightweight methods often sacrifice quality for speed and struggle to handle severely degraded underwater images. To address this limitation, we present a real-time underwater image enhancement framework with accurate color restoration. First, an Adaptive Weighted Channel Compensation module is introduced to achieve dynamic color recovery of the red and blue channels using the green channel as a reference anchor. Second, we design a Multi-branch Re-parameterized Dilated Convolution that employs multi-branch fusion during training and structural re-parameterization during inference, enabling large receptive field representation with low computational overhead. Finally, a Statistical Global Color Adjustment module is employed to optimize overall color performance based on statistical priors. Extensive experiments on eight datasets demonstrate that the proposed method achieves state-of-the-art performance across seven evaluation metrics. The model contains only 3,880 inference parameters and achieves an inference speed of 409 FPS. Our method improves the UCIQE score by 29.7% under diverse environmental conditions, and the deployment on ROV platforms and performance gains in downstream tasks further validate its superiority for real-time underwater missions.

Abstract

Image Enhancement

Explore similar work

Advancing Visual Reliability: Color-Accurate Underwater Image Enhancement for Real-Time Underwater Missions

Real-Time Underwater Image Enhancement via Frequency-Guided Dual-Path Attention

Hero-Mamba: Mamba-based Dual Domain Learning for Underwater Image Enhancement