Search papers, labs, and topics across Lattice.
This paper introduces a lightweight framework for real-time underwater image enhancement (UIE) that combines a Multi-Branch Reparameterizable Convolution with Fixed DCT Priors (MBRConv-DCT) and a Frequency-Guided Dual-Path Attention (FGDPA) module. By integrating frequency-sensitive priors and adaptive feature modulation, the model effectively addresses the unique challenges of underwater image degradation while maintaining low latency and computational efficiency. Experimental results demonstrate that the proposed approach achieves state-of-the-art performance with only 4.23K parameters and over 600 FPS, surpassing larger models in both quantitative metrics and visual quality.
Achieving state-of-the-art underwater image enhancement with a model that has only 4.23K parameters and processes images at over 600 FPS.
Real-time underwater image enhancement (UIE) is crucial for mobile underwater photography and autonomous robotic systems, where practical deployment typically requires low latency and compact models under constrained computational resources. Recent ultra-lightweight CNNs based on structural re-parameterization meet these constraints but operate purely in the spatial domain, ignoring the frequency-sensitive nature of underwater degradation. To address this, we propose a lightweight UIE framework that integrates two key components: a Multi-Branch Reparameterizable Convolution with Fixed DCT Priors (MBRConv-DCT) that injects structured directional frequency priors during training, and a Frequency-Guided Dual-Path Attention (FGDPA) module that fuses spatial and spectral cues via a dual-path design for adaptive feature modulation. Both components are fully compatible with structural re-parameterization: the convolution branch introduces zero additional inference cost after re-parameterization, while the attention module incurs only a minimal computational overhead. Experiments show our model achieves state-of-the-art performance with only 4.23K parameters and 600+ FPS, outperforming much larger methods in both quantitative metrics and visual quality. Code is available at https://github.com/LethyZhang/FGDPA.