SEA-PACE: Semi-Supervised Underwater Image Enhancement via Gaussian Process-Assisted Self-Paced Learning
Jingyang Wang, Hengyue Bi, Jingchao Cao, Feng Gao, Junyu Dong
Abstract
The scarcity of paired data severely limits the performance and generalization of learning-based underwater image enhancement (UIE) methods. This challenge is particularly prominent in scenes with complex degradations. Semi-supervised learning has emerged as a promising solution by enabling the utilization of large-scale unlabeled data. However, its effectiveness is limited by the use of static, model-agnostic metrics for pseudo-label reliability assessment. To address this, we propose SEA-PACE, a novel semi-supervised framework that integrates model-aware uncertainty modeling and self-paced consistency learning to fully exploit unlabeled data for UIE. Specifically, we design a Model-Aware Reliability Estimator (MARE) that quantifies the uncertainty of the teacher model's predictions through Gaussian Process Regression in latent feature space. The resulting uncertainty is then transformed into reliability weights via a rank-based mapping. Additionally, we apply the Self-Paced Consistency Learning (SPCL) strategy that employs a loss-aware schedule to dynamically prioritize high-confidence pseudo-labels, gradually incorporating more challenging samples during training. Extensive experiments on several public UIE benchmarks demonstrate that SEA-PACE consistently surpasses state-of-the-art methods in both visual quality and generalization capability.
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 on10
- Depth Anything V2Lihe Yang, Bingyi Kang, Zilong Huang, Zhen Zhao et al.NeurIPS 2024 · 2,305 citations
- Improved Knowledge Distillation via Teacher AssistantSeyed-Iman Mirzadeh, Mehrdad Farajtabar, Ang Li, Nir Levine et al.AAAI 2020 · 1,361 citations
- MUSIQ: Multi-scale Image Quality TransformerJunjie Ke, Qifei Wang, Yilin Wang, Peyman Milanfar et al.ICCV 2021 · 1,325 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
- From Synthetic to Real: Image Dehazing Collaborating with Unlabeled Real DataYe Liu, Lei Zhu, Shunda Pei, Huazhu Fu et al.ACM MM 2021 · 197 citations
Related papers
- Learning Underwater Image Enhancement Iteratively Without Reference ImagesYi Tang, Hiroshi Kawasaki, Takafumi Iwaguchi, Yuhang Zhang et al.AAAI 2026
- Contrastive Semi-Supervised Learning for Underwater Image Restoration via Reliable BankShirui Huang, Keyan Wang, Huan Liu, Jun Chen et al.CVPR 2023
- Adaptive Dual-domain Learning for Underwater Image EnhancementLintao Peng, Liheng BianAAAI 2025 · 9 citations
- SDUIE: Semi-Supervised Diffusion for Underwater Image Enhancement with Quant-Text Dual ControlXiaofeng Cong, Yu-Xin Zhang, Hao Shen, Yeying Jin et al.CVPR 2026
- Conditional Prompt Learning via Degradation Perception for Underwater Image EnhancementMingze Yao, Zhiying Jiang, Xianping Fu, Huibing WangAAAI 2026
