SDUIE: Semi-Supervised Diffusion for Underwater Image Enhancement with Quant-Text Dual Control
Xiaofeng Cong, Yu-Xin Zhang, Hao Shen, Yeying Jin, Junming Hou, Jie Gui
Abstract
Underwater images often exhibit dominant blue-green hues due to wavelength-dependent light attenuation. While existing enhancement methods have achieved promising performance, they typically overlook the subjective nature of visual preferences. To address this gap, we propose SDUIE, a level-aware Semi-supervised Diffusion framework for Underwater Image Enhancement that enables dual control through both quantitative and textual inputs. SDUIE-Quant allows continuous, numerical adjustment of enhancement levels via low-rank adaptation weight merging within a dual-branch diffusion model. This model comprises a supervised branch trained on synthetic underwater-terrestrial pairs and a self-supervised branch designed to preserve the natural hues of real-world underwater scenes. Building on this, SDUIE-Text introduces intuitive, languageguided control by aligning semantic prompts with visual enhancement effects, leveraging the learned fusion weights. This dual-modality design offers both precise control and flexible, user-preferred enhancement. Experimental results demonstrate that SDUIE achieves state-of-the-art results while better preserving the aesthetic qualities often missed by conventional methods. The source code is in https://github.com/Xiaofeng-life/SDUIE.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Builds on18
- LoRA: Low-Rank Adaptation of Large Language ModelsEdward J. Hu, Yelong Shen, Phillip Wallis, Zeyuan Allen-Zhu et al.ICLR 2022 · 18,833 citations
- High-Resolution Image Synthesis with Latent Diffusion ModelsRobin Rombach, Andreas Blattmann, Dominik Lorenz, Patrick Esser et al.CVPR 2022 · 13,123 citations
- Underwater Ranker: Learn Which Is Better and How to Be BetterChunle Guo, Ruiqi Wu, Xin Jin, Linghao Han et al.AAAI 2023 · 224 citations
- Unsupervised Underwater Image Restoration: From a Homology PerspectiveZhenqi Fu, Huangxing Lin, Yan Yang, Shu Chai et al.AAAI 2022 · 164 citations
- Nighttime Dehazing with a Synthetic BenchmarkJing Zhang, Yang Cao, Zheng-Jun Zha, Dacheng TaoACM MM 2020 · 137 citations
Related papers
- DACA-Net: A Degradation-Aware Conditional Diffusion Network for Underwater Image EnhancementChang Huang, Jiahang Cao, Jun Ma, Kieren Yu et al.ACM MM 2025 · 4 citations
- Wavelet-based Fourier Information Interaction with Frequency Diffusion Adjustment for Underwater Image RestorationChen Zhao, Weiling Cai, Chenyu Dong, Chengwei HuCVPR 2024 · 116 citations
- Conditional Prompt Learning via Degradation Perception for Underwater Image EnhancementMingze Yao, Zhiying Jiang, Xianping Fu, Huibing WangAAAI 2026
- Learning Underwater Image Enhancement Iteratively Without Reference ImagesYi Tang, Hiroshi Kawasaki, Takafumi Iwaguchi, Yuhang Zhang et al.AAAI 2026
- Adaptive Dual-domain Learning for Underwater Image EnhancementLintao Peng, Liheng BianAAAI 2025 · 9 citations
