# Two-Stage Teacher-Student Reliable Prior Learning for Robust Underwater Image Enhancement

arXiv: https://arxiv.org/abs/2608.00137 · HTML: https://mlsanity.com/paper/2608.00137/

## Key facts

- No citing papers in the map yet (data through 14 Oct 2026).
- Builds on 15 of its 48 references that are in the map.
- First posted on arXiv 31 Jul 2026.

## Builds on (most cited first, 15 of 15)

- [The Unreasonable Effectiveness of Deep Features as a Perceptual Metric](https://mlsanity.com/paper/1801.03924/) (2018)
- [YOLOv9: Learning What You Want to Learn Using Programmable Gradient Information](https://mlsanity.com/paper/2402.13616/) (2024)
- [An Underwater Image Enhancement Benchmark Dataset and Beyond](https://mlsanity.com/paper/1901.05495/) (2019)
- [Fast Underwater Image Enhancement for Improved Visual Perception](https://mlsanity.com/paper/1903.09766/) (2019)
- [Underwater Image Enhancement via Medium Transmission-Guided Multi-Color Space Embedding](https://mlsanity.com/paper/2104.13015/) (2021)
- [U-shape Transformer for Underwater Image Enhancement](https://mlsanity.com/paper/2111.11843/) (2021)
- [Wavelet-based Fourier Information Interaction with Frequency Diffusion Adjustment for Underwater Image Restoration](https://mlsanity.com/paper/2311.16845/) (2023)
- [Learning A Physical-aware Diffusion Model Based on Transformer for Underwater Image Enhancement](https://mlsanity.com/paper/2403.01497/) (2024)
- [Underwater Image Restoration Through a Prior Guided Hybrid Sense Approach and Extensive Benchmark Analysis](https://mlsanity.com/paper/2501.02701/) (2025)
- [A New Dataset, Poisson GAN and AquaNet for Underwater Object Grabbing](https://mlsanity.com/paper/2003.01446/) (2020)
- [FDCE-Net: Underwater Image Enhancement with Embedding Frequency and Dual Color Encoder](https://mlsanity.com/paper/2404.17936/) (2024)
- [Underwater Image Enhancement by Diffusion Model with Customized CLIP-Classifier](https://mlsanity.com/paper/2405.16214/) (2024)
- [Underwater Image Enhancement with Cascaded Contrastive Learning](https://mlsanity.com/paper/2411.10682/) (2024)
- [Large model enhanced computational ghost imaging](https://mlsanity.com/paper/2503.08710/) (2025)
- [WWE-UIE: A Wavelet & White Balance Efficient Network for Underwater Image Enhancement](https://mlsanity.com/paper/2511.16321/) (2025)

## Abstract (arXiv)

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 robustness and practical applicability of RPL-UIE.