arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2605.08073cs.CVcs.AI

EmambaIR:用于事件引导图像重建的高效视觉状态空间模型

EmambaIR: Efficient Visual State Space Model for Event-guided Image Reconstruction

Harbin Institute of Technology(哈尔滨理工大学)

Wei Yu, Yunhang Qian

AI总结:

本文提出EmambaIR,一种高效的视觉状态空间模型,通过跨模态Top-k稀疏注意力模块和门控状态空间模块,提升事件流图像重建的效率与性能,实现更高效的全局上下文捕捉。

摘要:

近期基于事件的图像重建方法主要依赖卷积神经网络(CNNs)和视觉变换器(ViTs)来处理互补的事件信息。然而,这些架构面临根本性的局限:CNNs往往无法捕捉全局特征相关性,而ViTs导致二次计算复杂度(例如O(n²)),阻碍了其在高分辨率场景中的应用。为解决这些瓶颈,我们引入EmambaIR,一种高效的视觉状态空间模型,用于使用空间稀疏和时间连续的事件流进行图像重建。我们的框架引入了两个关键组件:跨模态Top-k稀疏注意力模块(TSAM)和门控状态空间模块(GSSM)。TSAM高效地执行像素级Top-k稀疏注意力,以引导跨模态交互,产生丰富但稀疏的融合特征。随后,GSSM利用非线性门控单元来增强常规线性复杂度(O(n))SSMs的时间表示,有效捕捉全局上下文依赖性,而无需典型计算开销。在六个数据集上针对三种不同的图像重建任务——运动模糊消除、去雨和高动态范围(HDR)增强——进行了广泛的实验,结果表明EmambaIR在性能上显著优于现有最先进方法,同时在内存消耗和计算成本上实现了显著的减少。源代码和数据可在:https://github.com/YunhangWickert/EmambaIR 公开获取。

英文摘要:

Recent event-based image reconstruction methods predominantly rely on Convolutional Neural Networks (CNNs) and Vision Transformers (ViTs) to process complementary event information. However, these architectures face fundamental limitations: CNNs often fail to capture global feature correlations, whereas ViTs incur quadratic computational complexity (e.g., $O(n^2)$), hindering their application in high-resolution scenarios. To address these bottlenecks, we introduce EmambaIR, an Efficient visual State Space Model designed for image reconstruction using spatially sparse and temporally continuous event streams. Our framework introduces two key components: the cross-modal Top-k Sparse Attention Module (TSAM) and the Gated State-Space Module (GSSM). TSAM efficiently performs pixel-level top-k sparse attention to guide cross-modal interactions, yielding rich yet sparse fusion features. Subsequently, GSSM utilizes a nonlinear gated unit to enhance the temporal representation of vanilla linear-complexity ($O(n)$) SSMs, effectively capturing global contextual dependencies without the typical computational overhead. Extensive experiments on six datasets across three diverse image reconstruction tasks - motion deblurring, deraining, and High Dynamic Range (HDR) enhancement - demonstrate that EmambaIR significantly outperforms state-of-the-art methods while offering substantial reductions in memory consumption and computational cost. The source code and data are publicly available at: https://github.com/YunhangWickert/EmambaIR

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

↑