Efficient Monte Carlo denoising via state-space model with low GPU memory overhead
摘要
Monte Carlo denoising plays a pivotal role in enhancing image quality in physically based rendering. Despite the widespread adoption of learning-based methods, achieving high denoising accuracy while minimizing GPU memory usage remains a formidable challenge. In this paper, we introduce Denoising Mamba, a novel approach that leverages a memory-efficient state-space model architecture, circumventing the limitations of attention-based and diffusion-based methods. Our framework incorporates an optimized Fast Fourier Convolution Extractor for seamless local–global feature extraction and a Multi-Scale Residual Multiway Mamba Encodethat captures long-range spatial dependencies through linear-complexity multi-directional scanning with progressively increasing kernel sizes. This culminates in a lightweight decoder for reconstructing high-resolution, denoised images. Compared to mainstream CNN-base), attention-based, and diffusion-based learning methods, experimental results on real datasets demonstrate that our method achieves superior quantitative performance while requiring low computational resources. Our code and datasets are publicly available at [https://github.com/YUJIAJING0408/DenoisingMamba].