arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2403.20106cs.CVcs.LG
收藏

通过选择性状态空间模型学习丰富特征以实现高效图像去模糊

Learning Enriched Features via Selective State Spaces Model for Efficient Image Deblurring

School of Artificial Intelligence, Beijing Normal University(北京师范大学人工智能学院)

Hu Gao, Depeng Dang

AI总结:

针对图像去模糊中长程建模与效率难以兼顾的问题,提出基于选择性SSM的ALGBlock,融合局部注意力与全局状态特征并加权聚合,在基准上达到先进性能。

摘要:

图像去模糊旨在从相应的模糊图像中恢复高质量图像。CNN和Transformer的出现带来了显著进展。然而,这些方法往往面临在消除长程退化扰动与保持计算效率之间的两难。尽管选择性状态空间模型(SSM)在以线性复杂度建模长程依赖方面展现出潜力,但它也遇到局部像素遗忘和通道冗余等挑战。为解决这一问题,我们提出一种高效的图像去模糊网络,利用选择性状态空间模型聚合丰富且准确的特征。具体而言,我们引入局部与全局信息聚合块(ALGBlock),旨在有效捕获并整合局部不变属性与非局部信息。ALGBlock包含两个主要模块:用于捕获局部和全局特征的模块(CLGF)以及特征聚合模块(FA)。CLGF模块由两个分支组成:全局分支通过选择性状态空间模型捕获长程依赖特征,而局部分支采用简化通道注意力对局部连接性进行建模,从而减少局部像素遗忘和通道冗余。此外,我们设计了FA模块,在聚合两个分支进行恢复时通过重新校准权重来突出局部部分。实验结果表明,所提方法在广泛使用的基准上优于最先进方法。

英文摘要:

Image deblurring aims to restore a high-quality image from its corresponding blurred. The emergence of CNNs and Transformers has enabled significant progress. However, these methods often face the dilemma between eliminating long-range degradation perturbations and maintaining computational efficiency. While the selective state space model (SSM) shows promise in modeling long-range dependencies with linear complexity, it also encounters challenges such as local pixel forgetting and channel redundancy. To address this issue, we propose an efficient image deblurring network that leverages selective state spaces model to aggregate enriched and accurate features. Specifically, we introduce an aggregate local and global information block (ALGBlock) designed to effectively capture and integrate both local invariant properties and non-local information. The ALGBlock comprises two primary modules: a module for capturing local and global features (CLGF), and a feature aggregation module (FA). The CLGF module is composed of two branches: the global branch captures long-range dependency features via a selective state spaces model, while the local branch employs simplified channel attention to model local connectivity, thereby reducing local pixel forgetting and channel redundancy. In addition, we design a FA module to accentuate the local part by recalibrating the weight during the aggregation of the two branches for restoration. Experimental results demonstrate that the proposed method outperforms state-of-the-art approaches on widely used benchmarks.

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

↑