Feb, 2024

MambaIR:一种基于状态空间模型的简单图像恢复基准方法

TL;DR通过引入 Residual State Space Block 作为核心组件,该研究提出了一种名为 MambaIR 的简单而强大的基准模型,利用卷积和通道注意力增强了 vanilla Mamba 的能力,从而利用局部补丁的重复特性和通道交互产生特定于恢复的特征表示。广泛的实验证明了我们方法的优越性,例如,MambaIR 在类似计算成本但具有全局感受野的情况下,比基于 Transformer 的基准模型 SwinIR 提高了 0.36dB。