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

面向图像复原的上下文感知卷积网络

Towards Context-aware Convolutional Network for Image Restoration

College of Artificial Intelligence, Nankai University(南开大学人工智能学院);Ocean Engineering Research Center, Nankai University(南开大学海洋工程研究中心)

Fangwei Hao, Ji Du, Weiyun Liang, Jing Xu, Xiaoxuan Xu

AI总结:

针对现有图像复原卷积模块特征映射能力有限、感受野不足的问题,提出含高效残差星型模块与大型动态集成模块的上下文感知卷积网络,在低复杂度下实现多项图像复原任务的性能提升。

摘要:

图像复原(IR)是一项从受损观测图像中恢复高质量图像的长期任务。近年来,基于Transformer的算法和一些基于注意力的卷积神经网络(CNN)在多项IR任务中展现出了可观的效果。然而,现有的用于IR的卷积残差构建模块在将输入映射到高维非线性特征空间方面能力有限,且其局部感受野难以像Transformer一样捕捉长距离上下文信息。此外,用于IR的基于CNN的注意力模块要么存在大量静态参数,要么感受野有限。针对第一个问题,我们提出了高效残差星型模块(ERSM),该模块包含上下文感知的“星型运算”(逐元素乘法),可将特征按上下文映射到极高维的非线性特征空间中,极大地增强了表征学习能力。为进一步提升上下文信息的提取效果,针对第二个问题,我们提出了具有超大感受野的大型动态集成模块(LDIM)。因此,LDIM能够动态、高效地整合更多上下文信息,有助于进一步显著提升重建性能。将ERSM和LDIM集成到U型骨干网络中,我们提出了上下文感知卷积网络(CCNet),该网络在上下文高维映射和丰富上下文信息方面具备强大的学习能力。大量实验表明,我们的低模型复杂度CCNet在图像去雾、图像运动去模糊和图像去雪等多项IR任务上,性能优于其他最先进的IR方法。

英文摘要:

Image restoration (IR) is a long-standing task to recover a high-quality image from its corrupted observation. Recently, transformer-based algorithms and some attention-based convolutional neural networks (CNNs) have presented promising results on several IR tasks. However, existing convolutional residual building modules for IR encounter limited ability to map inputs into high-dimensional and non-linear feature spaces, and their local receptive fields have difficulty in capturing long-range context information like Transformer. Besides, CNN-based attention modules for IR either face static abundant parameters or have limited receptive fields. To address the first issue, we propose an efficient residual star module (ERSM) that includes context-aware "star operation" (element-wise multiplication) to contextually map features into exceedingly high-dimensional and non-linear feature spaces, which greatly enhances representation learning. To further boost the extraction of contextual information, as for the second issue, we propose a large dynamic integration module (LDIM) which possesses an extremely large receptive field. Thus, LDIM can dynamically and efficiently integrate more contextual information that helps to further significantly improve the reconstruction performance. Integrating ERSM and LDIM into an U-shaped backbone, we propose a context-aware convolutional network (CCNet) with powerful learning ability for contextual high-dimensional mapping and abundant contextual information. Extensive experiments show that our CCNet with low model complexity achieves superior performance compared to other state-of-the-art IR methods on several IR tasks, including image dehazing, image motion deblurring, and image desnowing.

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

↑