Depth-Synergized Mamba Meets Memory Experts for All-Day Image Reflection Separation
About
Image reflection separation aims to disentangle the transmission layer and the reflection layer from a blended image. Existing methods rely on limited information from a single image, tending to confuse the two layers when their contrasts are similar, a challenge more severe at night. To address this issue, we propose the Depth-Memory Decoupling Network (DMDNet). It employs the Depth-Aware Scanning (DAScan) to guide Mamba toward salient structures, promoting information flow along semantic coherence to construct stable states. Working in synergy with DAScan, the Depth-Synergized State-Space Model (DS-SSM) modulates the sensitivity of state activations by depth, suppressing the spread of ambiguous features that interfere with layer disentanglement. Furthermore, we introduce the Memory Expert Compensation Module (MECM), leveraging cross-image historical knowledge to guide experts in providing layer-specific compensation. To address the lack of datasets for nighttime reflection separation, we construct the Nighttime Image Reflection Separation (NightIRS) dataset. Extensive experiments demonstrate that DMDNet outperforms state-of-the-art methods in both daytime and nighttime.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Image Reflection Separation (Transmission Layer) | NightIRS 1.0 (test) | PSNR25.24 | 12 | |
| Transmission Layer Separation | Nature 20 images (test) | PSNR26.68 | 12 | |
| Transmission Layer Separation | Wild 55 images (test) | PSNR27.7 | 12 | |
| Transmission Layer Separation | Solid 200 images (test) | PSNR27.07 | 12 | |
| Transmission Layer Separation | Real 20 images (test) | PSNR24.6 | 12 | |
| Transmission Layer Separation | Postcard 199 images (test) | PSNR25.32 | 12 | |
| Image Reflection Separation (Reflection Layer) | NightIRS 1.0 (test) | PSNR28.37 | 9 | |
| Image Reflection Separation (Transmission Layer) | Public Datasets (test) | PSNR26.27 | 3 | |
| Image Reflection Separation (Transmission Layer) | NightIRS (test) | PSNR25.24 | 3 |