Search papers, labs, and topics across Lattice.
This paper addresses the challenges of Underwater Image Enhancement (UIE) by introducing a novel diffusion-based self-supervised learning approach that leverages the quality distribution of training labels. By utilizing semantic perception embeddings from a pre-trained diffusion model, the authors quantify label quality and guide a multi-step denoising process, effectively mitigating the impact of low-quality labels on model performance. Extensive evaluations reveal that this method significantly surpasses state-of-the-art techniques in restoration quality, showcasing its potential for practical applications in underwater imaging.
Self-supervised learning can transform low-quality labels into a powerful asset for enhancing underwater images, leading to unprecedented restoration quality.
Underwater Image Enhancement (UIE) is essential for mitigating degradations caused by water medium. Although learning-based methods have advanced significantly, most rely on paired datasets with unstable label quality, which bottlenecks model performance. This paper proposes a diffusion-based, in-dataset self-supervised learning strategy designed to exploit the quality distribution of training labels. Specifically, we evaluate label quality via semantic perception embeddings from a pre-trained diffusion model in a training-free manner. These quality scores are subsequently quantized into noise-level indices, guiding a multi-step denoising process for level-wise supervision. This mechanism prevents low-quality labels from degrading the model while maximizing their utility during training. Furthermore, a Fourier-based refinement network is incorporated to explicitly reconstruct high-frequency components. Extensive evaluations demonstrate that our method consistently outperforms SOTA approaches in restoration quality. The code and pre-trained model will be available once accepted in link.
