Toward Comprehensive Semantic Prompt for Region Contrastive Learning Underwater Image Enhancement

Abstract

Underwater image enhancement (UIE) focuses on mitigating image quality degradation due to light absorption and scattering. However, most existing methods enhance images via a global and uniform manner, neglecting the inherent semantic information in different regions, which may cause the network to easily deviate from the region’s original color. Moreover, these methods typically rely on clear images to guide network convergence, a process constrained by the limited availability of real-world datasets, making it extremely challenging to train enhancement models for various degradations. To address these challenges, this paper introduces a semantic guidance and region contrastive constraints network (SRCNet). Initially, we propose a semantic-aware RWKV (Receptance Weighted Key Value) block and a semantic prompt regularization module. These components leverage intra-target semantic correlations to preserve image details and colors within a global perceptual framework, while employing focal loss to emphasize the restoration of severely degraded regions. Subsequently, we introduce a region contrastive learning method that effectively utilizes negative samples to precisely capture features sensitive to degradation factors, thereby fostering robust feature distributions. Finally, experimental results demonstrate that our method outperforms existing state-of-the-art (SOTA) approaches.