S3VD: Semantic-Guidance Spatio-Temporal Scanning for Video Deraining
Organizations: School of Computer Science and Technology, Harbin Institute of Technology, Harbin 150001, China. · Zhengzhou Advanced Research Institute, Harbin Institute of Technology, Zhengzhou 450001, China.
Abstract
Heavy rainfall severely degrades outdoor videos by corrupting high-frequency details and introducing motion blur, critically undermining the reliability of visual tasks. Recently, State Space Models (SSMs), particularly Mamba, have emerged as efficient alternatives for vision tasks with their linear complexity and ability to model long-range dependencies. However, when confronted with the poor visual representations in rainy videos, Mamba still faces difficulties in preserving the integrity of 2D spatial semantics and modeling 3D spatio-temporal correlations. To break these limitations, we introduce S3VD, a Semantic-Guidance Spatio-Temporal Scanning framework for video deraining, featuring two key innovations: Multi-Scale Semantic Fusion (MSSF) Module and Spatio-Temporal Scanning Fusion (STSF) Module. The former integrates temporal semantic priors from DINOv2 to guide precise feature representation and counteract the loss of local semantic context inherent to Mamba's 1D flatten operation, enhancing robustness against extreme degradation. The latter introduces a spatio-temporal scanning mechanism and devises a Decoupled-Gating Mamba (DG-Mamba) layer, which employs two independent gates to adaptively control preceding and subsequent contextual information within the input clip, optimizing intra-frame and inter-frame correlation modeling. Experiments on video deraining benchmarks demonstrate the superiority of S3VD, achieving state-of-the-art performance with an average 0.84 dB PSNR improvement over Mamba-based baselines.
Figures & tables
| VRDS | RainVID&SS | RainSynAll100 | LWDDS | ||||
| Method | PSNR /SSIM /LPIPS | Method | PSNR /SSIM | Method | PSNR /SSIM | Method | PSNR /SSIM |
| CCN 28 | 23.75 / 0.8410 / 0.2091 | FastDerain 15 | 17.88 / 0.4506 | FCRVD 41 | 21.06 / 0.7405 | CCN 28 | 27.53 / 0.9220 |
| ESTINet 46 | 27.17 / 0.8436 / 0.2253 | MSCSC 19 | 19.19 / 0.5250 | RMFD 42 | 25.14 / 0.9172 | VWR 37 | 30.72 / 0.9726 |
| DRSformer 6 | 28.54 / 0.9075 / 0.1143 | PReNet 30 | 24.90 / 0.7464 | BasicVSR++ 3 | 27.67 / 0.9135 | BasicVSR++ 3 | 32.37 / 0.9792 |
| BasicVSR++ 3 | 29.75 / 0.9171 / 0.1023 | S2VD 45 | 29.69 / 0.9136 | NCFL 11 | 28.11 / 0.9235 | ViMP-Net 38 | 34.22 / 0.9784 |
| ViMP-Net 38 | 31.02 / 0.9283 / 0.0862 | VDMamba 35 | 32.69 / 0.9233 | SALN 44 | 29.78 / 0.9315 | SALN 44 | 36.57 / 0.9802 |
| Method | MSCSC | RMFD | S2VD | RainMamba | S3VD |
| WE | 0.054 | 0.043 | 0.033 | 0.029 | 0.013 |
| Feature representation | Mean cosine similarity |
| ConvNeXt features | |
| DINOv2 semantic features |
| Feature configuration | PSNR | SSIM | LPIPS |
| Semantic features only | 30.74 | 0.9291 | 0.0788 |
| Pixel features only | 32.35 | 0.9413 | 0.0642 |
| Pixel features Semantic features | 32.43 | 0.9427 | 0.0610 |
| Method | BasicVSR++ | ESTINet | DRSformer | RainMamba | VDMamba | Ours |
| Parameters(M) | 6.22 | 22.96 | 33.63 | 34.75 | 12.70 | 42.08 |
| FLOPs(G) | 1616.44 | 681.83 | 1101.89 | 118.99 | 389.79 | 143.58 |
| Time(s/frame) | 0.051 | 0.0341 | 0.0381 | 0.0112 | 0.0320 | 0.0089 |
| Model | MSSF | STSF | PSNR | SSIM | LPIPS |
| D1 | 32.04 | 0.9366 | 0.0684 | ||
| D2 | ✓ | 32.10 | 0.9403 | 0.0635 | |
| D3 | ✓ | 32.35 | 0.9413 | 0.0642 | |
| S3VD (Ours) | ✓ | ✓ | 32.43 | 0.9427 | 0.0610 |
| PSNR | SSIM | LPIPS | ||
| 0 | 0 | 31.96 | 0.9388 | 0.0669 |
| 0.3 | 0 | 32.25 | 0.9416 | 0.0629 |
| 0 | 0.1 | 32.18 | 0.9409 | 0.0638 |
| 0.3 | 0.1 | 32.43 | 0.9427 | 0.0610 |
| 0.1 | 0.1 | 32.34 | 0.9420 | 0.0621 |
| 0.3 | 0.05 | 32.38 | 0.9423 | 0.0616 |
| Model | PSNR | SSIM | LPIPS | Param | FLOPS | Runtime |
| DINOv2-small (Ours) | 32.43 | 0.9427 | 0.0610 | 63.08 | 143.58 | 0.0089 |
| DINOv2-small with register | 32.39 | 0.9417 | 0.0622 | 63.08 | 143.58 | 0.0078 |
| DINOv2-base | 32.53 | 0.9431 | 0.0618 | 127.63 | 248.42 | 0.0094 |
| DINOv2-large | 32.58 | 0.9422 | 0.0618 | 345.40 | 602.17 | 0.0143 |
| DINOv1-small | 32.34 | 0.9410 | 0.0632 | 63.30 | 129.43 | 0.0076 |
| CLIP-visual | 32.42 | 0.9413 | 0.0627 | 127.76 | 193.21 | 0.0082 |
| Method | PSNR | SSIM | LPIPS |
| single frame | 32.22 | 0.9420 | 0.0622 |
| cross-frame average | 32.39 | 0.9425 | 0.0611 |
| cross-frame concatenation (Ours) | 32.43 | 0.9427 | 0.0610 |
| Fused levels | PSNR | SSIM | LPIPS |
| 32.19 | 0.9408 | 0.0641 | |
| 32.28 | 0.9415 | 0.0628 | |
| 32.36 | 0.9422 | 0.0617 | |
| 32.43 | 0.9427 | 0.0610 |
| Fusion operator | Params (M) | GFLOPs | PSNR | SSIM | LPIPS |
| Concatenation Conv | 40.31 | 132.75 | 32.17 | 0.9406 | 0.0638 |
| Bidirectional Mamba fusion | 42.36 | 146.91 | 32.31 | 0.9418 | 0.0622 |
| Cross-attention (MSSF) | 42.08 | 143.58 | 32.43 | 0.9427 | 0.0610 |
| Method | PSNR | SSIM | LPIPS |
| T+S | 32.05 | 0.9402 | 0.0638 |
| S+S | 32.13 | 0.9407 | 0.0625 |
| T+T | 32.31 | 0.9414 | 0.0634 |
| S+T (Ours) | 32.43 | 0.9427 | 0.0610 |
| Depth | Params | GFLOPs | Time | PSNR | SSIM | LPIPS |
| 35.42 | 118.62 | 0.0074 | 32.18 | 0.9405 | 0.0642 | |
| 39.76 | 135.47 | 0.0082 | 32.34 | 0.9420 | 0.0621 | |
| 42.08 | 143.58 | 0.0089 | 32.43 | 0.9427 | 0.0610 | |
| 48.91 | 169.84 | 0.0106 | 32.45 | 0.9428 | 0.0609 |
| Shared core | Direction gates | PSNR | SSIM | LPIPS |
| – | – | 32.12 | 0.9410 | 0.0624 |
| ✓ | – | 32.24 | 0.9417 | 0.0619 |
| – | ✓ | 32.29 | 0.9421 | 0.0616 |
| ✓ | ✓ | 32.43 | 0.9427 | 0.0610 |
Video Restoration
Video restoration reconstructs clearer, more faithful, temporally consistent footage from blur, noise, compression, weather, and other degradations.
See all 59 papers