A Lightweight Deep Exclusion Unfolding Network for Single Image Reflection Removal
About
Single Image Reflection Removal (SIRR) is a canonical blind source separation problem and refers to the issue of separating a reflection-contaminated image into a transmission and a reflection image. The core challenge lies in minimizing the commonalities among different sources. Existing deep learning approaches either neglect the significance of feature interactions or rely on heuristically designed architectures. In this paper, we propose a novel Deep Exclusion unfolding Network (DExNet), a lightweight, interpretable, and effective network architecture for SIRR. DExNet is principally constructed by unfolding and parameterizing a simple iterative Sparse and Auxiliary Feature Update (i-SAFU) algorithm, which is specifically designed to solve a new model-based SIRR optimization formulation incorporating a general exclusion prior. This general exclusion prior enables the unfolded SAFU module to inherently identify and penalize commonalities between the transmission and reflection features, ensuring more accurate separation. The principled design of DExNet not only enhances its interpretability but also significantly improves its performance. Comprehensive experiments on four benchmark datasets demonstrate that DExNet achieves state-of-the-art visual and quantitative results while utilizing only approximately 8\% of the parameters required by leading methods.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Single Image Reflection Removal | Real20 (test) | PSNR23.42 | 70 | |
| Single Image Reflection Separation | SIR2 Wild (test) | PSNR26.65 | 20 | |
| Single Image Reflection Separation | SIR2 Postcard (test) | PSNR24.21 | 20 | |
| Single Image Reflection Removal | Nature (test) | PSNR23.39 | 19 | |
| Single Image Reflection Separation | Synthetic Datasets Average (540) (test) | PSNR25.04 | 12 | |
| Single Image Reflection Separation | SIR2 Objects (test) | PSNR25.37 | 12 | |
| Single Image Reflection Separation | OpenRR-1K (test) | PSNR26.1434 | 6 |