中文

WaMaIR:基于多尺度小波卷积和 Mamba 式通道建模的图像修复方法

计算机视觉与模式识别 2025-10-30 v2

摘要

图像修复是计算机视觉中的一个基本且具有挑战性的任务,CNN 架构显示出显著的计算效率。然而,先前的 CNN 方法常面临在充分恢复细节纹理方面受限,受限于 CNN 结构的小感受野以及缺乏通道特征建模。在本文中,我们提出 WaMaIR,一种具有大感受野的新型框架,用于图像感知并提高恢复图像中纹理细节的重建。具体而言,我们引入全局多尺度小波变换卷积 (GMWTConvs) 以扩大感受野,提取图像特征,同时保留和丰富模型输入中的纹理特征。此外,我们提出 Mamba 式通道感知模块 (MCAM),专为捕获特征通道中的长程依赖而设计,增强模型对颜色、边缘和纹理信息的敏感性。 Additionally,我们提出多尺度纹理增强损失 (MTELoss),用于指导模型有效保留详细纹理结构。大量实验表明,WaMaIR 在图像修复和模型计算性能方面均优于最先进的方法。

关键词

引用

@article{arxiv.2510.16765,
  title  = {WaMaIR: Image Restoration via Multiscale Wavelet Convolutions and Mamba-based Channel Modeling with Texture Enhancement},
  author = {Shengyu Zhu and Congyi Fan and Fuxuan Zhang},
  journal= {arXiv preprint arXiv:2510.16765},
  year   = {2025}
}

备注

Chinese Conference on Pattern Recognition and Computer Vision (PRCV), Oral