SARM: Scene-Aware Retinex Mamba for Underwater Image Enhancement

Underwater image enhancement is essential for marine visual perception tasks. However, the highly heterogeneous optical degradations in real-world waters, the scarcity of paired training data, and the inherent dilemma for existing models in balancing long-range dependency modeling with computational overhead pose significant challenges. To address these issues, this paper proposes a prior-guided, self-supervised underwater image enhancement framework called Scene-Aware Retinex Mamba (SARM). This framework seamlessly integrates Retinex theoretical priors with state space models (SSMs) and operates without paired supervision by employing a prior-guided pseudo-labeling strategy to guide network optimization. Architecturally, SARM deeply couples the physical Retinex prior with SSM. Its core module integrates multi-color space features and leverages a 2D selective scan mechanism to achieve global context modeling with linear complexity O(HW), effectively removing complex color casts and suppressing non-uniform scattering noise. To further overcome the generalization bottlenecks in cross-domain underwater testing, this paper introduces a Scene-Aware Adapter (SAA), which facilitates dynamic loss scheduling and adaptive feature gating by quantifying scene-specific degradation characteristics. Comprehensive evaluations on multiple benchmark datasets, including UIEB, EUVP, and UCCS, demonstrate that SARM achieves state-of-the-art subjective and objective enhancement quality (e.g., yielding a URanker score of 2.491 and a CCF score of 35.76), while maintaining an ultra-fast inference speed of 136.52 FPS on the UIEB dataset. Furthermore, extended experiments reveal that SARM can significantly boost the performance of downstream vision tasks, validating its potential as a robust preprocessing module for various practical marine vision applications.