Rethinking Underwater Image Enhancement: Learning through Text-Guided Degradation Curriculum
Authors: Hongri Tian, Wenxue Li, Tian Ye, Sixiang Chen, Zongxian Su, Lujia Wang, Xianying Wang, Xinhu Zheng, Gang Hua, Lei Zhu
Organizations: The Hong Kong University of Science and Technology (Guangzhou), Guangzhou, China. · China Geological Survey, Beijing, China. · Director of Applied Science, Amazon.com, Inc., U.S.. · The Hong Kong University of Science and Technology, Hong Kong SAR, China.
Existing paired clean-degraded datasets have advanced learning-based underwater image enhancement, but they remain either limited in scale or difficult to acquire, and still fail to adequately cover the diverse degradations encountered in real-world underwater environments. Such scarce and insufficient data ultimately restrict the generalization of underwater image enhancement models, highlighting the need for realistic and controllable degradation modeling to better capture real underwater conditions and improve enhancement robustness. To this end, we propose a unified degradation-aware training framework for the underwater image enhancement task. A Text-Conditioned Degradation Augmentation Pipeline uses a text-guided diffusion model to synthesize diverse and semantically consistent degraded samples, providing paired data to support the subsequent curriculum learning strategy. The proposed Progressive Degradation Curriculum organizes training across multiple attributes and difficulty levels to promote balanced learning, and a Transmission-Guided One-Step Diffusion Model leverages this progressive process to better address the wide range of degradation severity and restoration ambiguity in underwater images. Extensive experiments on multiple underwater benchmarks demonstrate that our proposed method achieves superior performance. Our code and models will be released upon the publication of this work.
Figures & tables
Figure 1: Top : Illustration of three common routes for constructing paired underwater datasets. Medium : Proposed text-driven degradation diffusion paradigm for controllable underwater degradation synthesis. Bottom : Quantitative results of our method on performance improvement and synthesis quality.
Figure 2: Overview of the proposed Text-Conditioned Degradation Augmentation (TCDA) pipeline. It consists of four stages: (1) Degradation Captioning via MLLMs for generating fine-grained textual descriptions of underwater degradations, (2) Flow-Based Degradation Modeling for learning text-conditioned mappings between clean and degraded representations, (3) Prompt-Space Augmentation for enriching degradation prompts with varying difficulty levels, and (4) Text-Conditioned Degradation Synthesis for controllable degradation generation to support curriculum learning.
Figure 3: Overview of the proposed UIE framework. The model learns progressive degradation-aware enhancement through curriculum sampling and transmission-guided one-step diffusion, combining physical priors and text-conditioned cues for controllable restoration with greater inference efficiency than multi-step diffusion methods.
Figure 4: Examples of learned degradation generation results. Given a clean image and a textual prompt explicitly describing degradation attributes (e.g., turbidity, color cast, lighting, haze), our model synthesizes degraded counterparts that accurately reflect the specified visual conditions and severity levels.
Methods
UIEB
C60
U45
SSIM ↑
PSNR ↑
LPIPS ↓
UCIQE ↑
UIQM ↑
UCIQE ↑
UIQM ↑
UCIQE ↑
UIQM ↑
UIEC 2 - Net Wang et al. (2021)
0.880
17.961
0.252
0.526
3.766
0.487
2.149
0.535
4.231
+ TCDA
0.880 (+0.000)
19.020 (+1.059)
0.229 (-0.023)
0.550 (+0.024)
3.695 (-0.071)
0.525 (+0.038)
2.146 (-0.003)
0.554 (+0.019)
4.156 (-0.075)
SCNet Fu et al. (2022)
0.907
19.560
0.195
0.585
3.869
0.566
2.498
0.570
4.155
+ TCDA
0.909 (+0.002)
20.998 (+1.438)
0.172 (-0.023)
0.593 (+0.008)
3.925 (+0.056)
0.551 (-0.015)
2.452 (-0.046)
0.578 (+0.008)
4.163 (+0.008)
FiveA+ Jiang et al. (2023)
0.903
18.671
0.260
0.558
3.811
0.531
2.229
0.546
4.186
Table 1: Quantitative comparison on UIEB, C60, and U45. Each method is evaluated with its original training data and with TCDA augmentation. Parentheses show the performance change. Best results are in bold.
Figure 5: Visual comparisons with state-of-the-art UIE methods on the UIEB dataset. Our method consistently outperforms state-of-the-art UIE methods for underwater images with varying degrees of degradation.
Table 2: Ablation study on the key components of TG-Diff.
Figure 6: Qualitative ablation study of the proposed TG-Diff components on the UIEB dataset. The full model restores clearer structure, more natural illumination, and balanced color rendition.
Methods SSIM ↑ PSNR ↑ LPIPS ↓ UCIQE ↑ UIEC 2 -Net Wang et al. (2021) 0.857 20.570 0.255 0.552 SCNet Fu et al. (2022) 0.872 20.737 0.215 0.584 FiveA+ Jiang et al. (2023) 0.876 21.655 0.241 0.570 WF-Diff Zhao et al. (2024) 0.895 23.346 0.165 0.590 CE-VAE Pucci and Martinel (2025) 0.851 25.535 0.190 0.578 Phaseformer Khan et al. (2025) 0.891 25.483 0.135 0.579 TG-Diff (Ours) 0.904 25.734 0.130 0.592
Table 3: Quantitative comparison on LSUI Peng et al. (2023) . Best results are in bold.
Figure 7: Visualization of degradation generation across different difficulty levels. Conditioned on text prompts describing easy, moderate, and hard degradations, the model generates visually consistent degradation patterns with increasing severity.
Methods SSIM ↑ PSNR ↑ LPIPS ↓ UCIQE ↑ TG-Diff + Easy only 0.901 21.474 0.149 0.603 TG-Diff + Moderate only 0.905 21.392 0.147 0.608 TG-Diff + Hard only 0.907 21.404 0.150 0.603 TG-Diff + Random Sample 0.925 22.248 0.129 0.608 TG-Diff + Curriculum 0.927 22.341 0.129 0.613
Table 4: Ablation of degradation-difficulty sampling. Curriculum learning yields the best balanced performance.
Method
SSIM ↑
PSNR ↑
LPIPS ↓
UCIQE ↑
w/ UWCNN
0.855
19.491
0.185
0.580
w/ TCDA (Ours)
0.905
21.392
0.147
0.608
Table 6: Comparison between UWCNN-based augmentation and TCDA (moderate only).
Figure 8: Trends of non-reference underwater image quality metrics under different degradation difficulty level settings. Both UCIQE and UIQM consistently decrease as the degradation difficulty increases from easy to hard.
Underwater image enhancement (UIE) aims to recover clear images from observations affected by wavelength-dependent absorption, scattering, and spatially nonuniform degradation. Although existing generative methods can handle complex degradations, severe information loss may lead to semantic drift in the restored results. To address this issue, we propose RPL-UIE, a two-stage teacher--student framework for reliable prior learning. In the teacher stage, the network learns reliable and complementary spatial priors characterizing appearance and photometric properties from paired degraded and reference images. In the student stage, the network takes only degraded images as input and learns to emulate the teacher's prior extraction capability, thereby providing more reliable restoration guidance for the enhancement process without requiring reference images at inference. To reduce the prior-learning discrepancy between the teacher and student models, we further develop Residual Prior Refinement Diffusion (RPRD) and Frequency-Aware Prior Residual Calibration (FPRC). RPRD uses the coarse priors as anchors and progressively predicts the necessary corrections in the residual space. FPRC retains stable low-frequency residual components and selectively modulates high-frequency detail residuals, producing calibrated priors to support high-quality reconstruction. Experiments on multiple UIE benchmarks demonstrate competitive restoration performance. Downstream underwater object detection and instance segmentation experiments further demonstrate the improved utility of enhanced images for visual perception, while tests on real-world data captured by a remotely operated vehicle (ROV) support the practical applicability of RPL-UIE.
Yifan Chen, Jiaming Liu, Ye Zheng +2
College of Future Information Technology, Fudan University, Shanghai 200433, China · Institute of Artificial Intelligence (TeleAI), China Telecom, China · Faculty of Robot Science and Engineering, Northeastern University, Shenyang 110819, China +1
Degradation information in underwater images plays a dual role: its spatial and spectral cues can guide adaptive restoration, while degradation-entangled features may be propagated without explicit regulation during decoding. Existing methods largely overlook this dual role, either underexploiting degradation cues or directly forwarding encoder features through skip connections. To address this issue, we propose PROTEUS, which couples degradation-guided feature adaptation with task?oriented latent control. PROTEUS tackles this problem from two complementary perspectives. At the feature level, the Guided Dynamic Feature Modulation Block exploits spatially varying degradation cues to adapt feature processing across network stages. At the representation level, the task-oriented latent controller learns a structured control code under discriminative regularisation and uses it for channel-wise modulation of skip features, without requiring the code to form a metrically cleaner embedding. Extensive experiments on five paired and four non-reference underwater benchmarks demonstrate that PROTEUS achieves highly competitive restoration performance, with a favourable balance between restoration quality and computational cost.
Xu Zhang, Xuhui Cao, Kangzhe Yuan +5
National Engineering Research Center for Multimedia Software, School of Computer Science, Wuhan University · School of Computer and Artificial Intelligence, Zhengzhou University · Aerospace Information Research Institute, Henan Academy of Sciences +2
In recent years, learning-based underwater image enhancement (UIE) techniques have rapidly evolved. However, distribution shifts between high-quality enhanced outputs and natural images can hinder semantic cue extraction for downstream vision tasks, thereby limiting the adaptability of existing enhancement models. To address this challenge, this work proposes a new learning mechanism that leverages Vision-Language Models (VLMs) to empower UIE models with semantic-sensitive capabilities. To be concrete, our strategy first generates textual descriptions of key objects from a degraded image via VLMs. Subsequently, a text-image alignment model remaps these relevant descriptions back onto the image to produce a spatial semantic guidance map. This map then steers the UIE network through a dual-guidance mechanism, which combines cross-attention and an explicit alignment loss. This forces the network to focus its restorative power on semantic-sensitive regions during image reconstruction, rather than pursuing a globally uniform improvement, thereby ensuring the faithful restoration of key object features. Experiments confirm that when our strategy is applied to different UIE baselines, significantly boosts their performance on perceptual quality metrics as well as enhances their performance on detection and segmentation tasks, validating its effectiveness and adaptability.
Guodong Fan, Shengning Zhou, Genji Yuan +3
Shandong Technology and Business University, Yantai, China · Shandong University of Finance and Economics, Jinan, China · Dalian Maritime University, Dalian, China