MambaIRv2来了！带有注意力的状态空间复原

文章预览

关注 「 AIWalker 」 并 星标 从此AI不迷路 作者 | 郭航     链接 | https://zhuanlan.zhihu.com/p/9375585949 前段时间我们开发的基于Mamba的图像复原模型MambaIR受到了研究者的关注，并展示了Mamba作为除了CNN和Transformer之外的第三种图像修复backbone的潜力。 这里也感谢社区的反馈与支持来帮助我们的MambaIR项目不断完善。 最近，我们在MambaIR的基础上继续开发了一个新版本MambaIRv2，之后的代码和模型权重我们也会陆续开源。 下面是相关的论文和代码链接。 https://arxiv.org/pdf/2411.15269 https://github.com/csguoh/MambaIR 简单来说，我们的MambaIRv2的主要亮点包括： 我们将原始MamabIR的扫描顺序从4次降低为了1次，同时不影响图像均匀采样 我们的MambaIRv2在lightSR任务上用更少参数实现了比SRformer更好的结果 在经典SR任务上，MambaIRv2用相同的参数量超过了HAT Motivation MambaIRv2要解决的核心 ………………………………