arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2407.18613cs.CVeess.IV

用于图像恢复的膨胀条带注意力网络

Dilated Strip Attention Network for Image Restoration

Nankai University(南开大学)

Fangwei Hao, Jiesheng Wu, Ji Du, Yinjie Wang, Jing Xu

AI总结:

针对图像恢复中注意力模块感受野受限或参数量大的问题,提出膨胀条带注意力网络(DSAN),通过水平和垂直方向的膨胀条带注意力机制及多尺度感受野高效整合上下文信息,在多项任务上超越现有最先进算法。

摘要:

图像恢复是一项长期存在的任务,旨在从退化的对应物中恢复出潜在的清晰图像。由于自注意力机制在捕获长程依赖方面具有强大的能力,基于transformer的方法或一些基于注意力的卷积神经网络在近年来的许多图像恢复任务中展现出了令人瞩目的效果。然而,现有的注意力模块面临着感受野受限或参数量庞大的问题。为了更有效且高效地整合上下文信息,本文提出了一种用于图像恢复的膨胀条带注意力网络(DSAN)。具体而言,为了从同一行或列的相邻像素中为每个像素收集更多的上下文信息,我们精心提出了一种膨胀条带注意力(DSA)机制。通过在水平和垂直方向上应用DSA操作,每个位置都能从更广泛的区域中获取上下文信息。此外,我们在DSA中的不同特征组之间利用多尺度感受野来提升表示学习。大量实验表明,我们的DSAN在多个图像恢复任务上超越了现有最先进的算法。

英文摘要:

Image restoration is a long-standing task that seeks to recover the latent sharp image from its deteriorated counterpart. Due to the robust capacity of self-attention to capture long-range dependencies, transformer-based methods or some attention-based convolutional neural networks have demonstrated promising results on many image restoration tasks in recent years. However, existing attention modules encounters limited receptive fields or abundant parameters. In order to integrate contextual information more effectively and efficiently, in this paper, we propose a dilated strip attention network (DSAN) for image restoration. Specifically, to gather more contextual information for each pixel from its neighboring pixels in the same row or column, a dilated strip attention (DSA) mechanism is elaborately proposed. By employing the DSA operation horizontally and vertically, each location can harvest the contextual information from a much wider region. In addition, we utilize multi-scale receptive fields across different feature groups in DSA to improve representation learning. Extensive experiments show that our DSAN outperforms state-of-the-art algorithms on several image restoration tasks.

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

↑