arXivDaily arXiv每日学术速递 周一至周五更新
​通知:联邦节假日调休:哥伦布日 / 原住民日(Columbus Day / Indigenous Peoples' Day),今日(13日)无更新。
arXiv 2409.00410cs.CV

一种用于单图像去雨的混合Transformer-Mamba网络

A Hybrid Transformer-Mamba Network for Single Image Deraining

Institute of Information Engineering, Chinese Academy of Sciences(中国科学院信息工程研究所);University of Chinese Academy of Sciences(中国科学院大学);School of Cyber Science and Technology, Shenzhen Campus of Sun Yat-sen University(中山大学深圳校区网络空间科学与技术学院);Yunnan Normal University(云南师范大学)

Shangquan Sun, Wenqi Ren, Juxiang Zhou, Jianhou Gan, Rui Wang, Xiaochun Cao

AI总结:

本文提出双分支混合网络TransMamba,结合谱带Transformer与Mamba双向状态空间模型捕获长程依赖,并通过谱相干损失提升单图像去雨性能。

摘要:

现有的去雨Transformer采用固定范围窗口或沿通道维度的自注意力机制,限制了对非局部感受野的利用。针对这一问题,我们提出了一种新颖的双分支混合Transformer-Mamba网络,称为TransMamba,旨在有效捕获长程的雨相关依赖关系。基于雨退化与背景具有不同谱域特征的先验,我们在第一个分支上设计了谱带Transformer块。自注意力在谱域通道维度的组合内执行,以提高长程依赖建模能力。为增强频率特定信息,我们提出了一个谱增强前馈模块,在谱域中聚合特征。在第二个分支中,Mamba层配备级联双向状态空间模型模块,以额外捕获局部和全局信息的建模。在编码器和解码器的每个阶段,我们对双分支特征进行通道级拼接,并通过通道缩减实现特征融合,从而更有效地整合来自Transformer和Mamba分支的多尺度信息。为更好地重建干净图像中固有的信号级关系,我们还开发了一种谱相干损失。在多种数据集和真实世界图像上的大量实验表明,与最先进方法相比,我们的方法具有优越性。

英文摘要:

Existing deraining Transformers employ self-attention mechanisms with fixed-range windows or along channel dimensions, limiting the exploitation of non-local receptive fields. In response to this issue, we introduce a novel dual-branch hybrid Transformer-Mamba network, denoted as TransMamba, aimed at effectively capturing long-range rain-related dependencies. Based on the prior of distinct spectral-domain features of rain degradation and background, we design a spectral-banded Transformer blocks on the first branch. Self-attention is executed within the combination of the spectral-domain channel dimension to improve the ability of modeling long-range dependencies. To enhance frequency-specific information, we present a spectral enhanced feed-forward module that aggregates features in the spectral domain. In the second branch, Mamba layers are equipped with cascaded bidirectional state space model modules to additionally capture the modeling of both local and global information. At each stage of both the encoder and decoder, we perform channel-wise concatenation of dual-branch features and achieve feature fusion through channel reduction, enabling more effective integration of the multi-scale information from the Transformer and Mamba branches. To better reconstruct innate signal-level relations within clean images, we also develop a spectral coherence loss. Extensive experiments on diverse datasets and real-world images demonstrate the superiority of our method compared against the state-of-the-art approaches.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑