Enhancing mural inpainting: a dual-branch approach integrating mamba priors and wavelet-sparse transformers
摘要
The Dunhuang murals, an integral part of Chinese cultural heritage, face severe degradation due to natural factors, posing challenges for inpainting. Traditional methods often fail to address these complexities, struggling with global semantic understanding, fine-detail reconstruction, and computational efficiency. To overcome these limitations, we propose MWT-Net (Mamba Prior and Wavelet-Sparse Transformer Network), which combines Mamba priors with a wavelet-based sparse Transformer in a dual-branch architecture. The prior-guided branch captures long-range dependencies and global context via state-space modeling, while the main branch uses multi-scale sparse modeling for enhanced texture reconstruction. A GDT-FF module improves attention expressiveness, preserves fine details, and reduces redundancy. Experiments show that MWT-Net outperforms existing approaches on multiple benchmarks, achieving a 2.67 dB gain in PSNR, a 9.60% increase in SSIM, and reductions of 0.1113 and 0.0250 in LPIPS and FID. These results highlight the potential of MWT-Net to produce coherent and realistic mural inpainting, contributing to cultural heritage preservation. Our code and dataset are available at https://github.com/github662/dunhuang.