DeRainMamba: A Frequency-Aware State Space Model with Detail Enhancement for Image Deraining
By: Zhiliang Zhu , Tao Zeng , Tao Yang and more
Potential Business Impact:
Cleans rainy pictures without losing details.
Image deraining is crucial for improving visual quality and supporting reliable downstream vision tasks. Although Mamba-based models provide efficient sequence modeling, their limited ability to capture fine-grained details and lack of frequency-domain awareness restrict further improvements. To address these issues, we propose DeRainMamba, which integrates a Frequency-Aware State-Space Module (FASSM) and Multi-Directional Perception Convolution (MDPConv). FASSM leverages Fourier transform to distinguish rain streaks from high-frequency image details, balancing rain removal and detail preservation. MDPConv further restores local structures by capturing anisotropic gradient features and efficiently fusing multiple convolution branches. Extensive experiments on four public benchmarks demonstrate that DeRainMamba consistently outperforms state-of-the-art methods in PSNR and SSIM, while requiring fewer parameters and lower computational costs. These results validate the effectiveness of combining frequency-domain modeling and spatial detail enhancement within a state-space framework for single image deraining.
Similar Papers
PRE-Mamba: A 4D State Space Model for Ultra-High-Frequent Event Camera Deraining
CV and Pattern Recognition
Cleans up blurry camera images caused by rain.
Traffic Image Restoration under Adverse Weather via Frequency-Aware Mamba
CV and Pattern Recognition
Clears foggy pictures for self-driving cars.
SpectMamba: Integrating Frequency and State Space Models for Enhanced Medical Image Detection
CV and Pattern Recognition
Finds sickness in medical pictures faster.