arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2606.06176cs.CV

RQUL-UIE: 通过数据集内自监督重振质量不稳定标签用于水下图像增强

RQUL-UIE: Revitalizing Quality-Unstable Labels for Underwater Image Enhancement via In-Dataset Self-Supervision

The Hong Kong Polytechnic University(香港理工大学)

Haochen Hu, Yanrui Bin, Chih-yung Wen, Bing Wang

AI总结:

提出一种基于扩散模型的数据集内自监督学习策略,通过评估标签质量并量化噪声级别进行分步去噪监督,结合傅里叶细化网络,有效利用不稳定标签提升水下图像增强质量。

摘要:

水下图像增强对于减轻水介质引起的退化至关重要。尽管基于学习的方法取得了显著进展,但大多数依赖于具有不稳定标签质量的配对数据集,这限制了模型性能。本文提出了一种基于扩散的数据集内自监督学习策略,旨在利用训练标签的质量分布。具体地,我们通过预训练扩散模型的语义感知嵌入以无需训练的方式评估标签质量。这些质量分数随后被量化为噪声级别索引,指导多步去噪过程以进行级别监督。该机制防止低质量标签降低模型性能,同时最大化其在训练中的效用。此外,引入基于傅里叶的细化网络以显式重建高频分量。大量评估表明,我们的方法在恢复质量上始终优于最先进的方法。代码和预训练模型将在接收后提供链接。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑