Depth-Synergized Mamba Meets Memory Experts for All-Day Image Reflection Separation
This addresses a challenging computer vision problem for applications like photography and autonomous driving, with incremental improvements in handling nighttime conditions.
The paper tackles image reflection separation, particularly at night when layer contrasts are similar, by proposing DMDNet which integrates depth-aware scanning with Mamba models and memory experts, achieving state-of-the-art performance on both daytime and nighttime datasets.
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.