arXiv:2409.06590cs.CV2024-09被引 1

融合卷积与Transformer的轻量级图像超分辨率网络

Lightweight single-image super-resolution network based on dual paths

  • 双分支结构并行提取局部与全局特征
  • 多阶段特征补全减少深层网络信息丢失
  • 同等参数量下性能优于其他轻量模型

当前基于深度学习的单图像超分辨率(SISR)算法主要分为两类:一类基于卷积神经网络,擅长提取图像局部特征;另一类基于Transformer,通过自注意力机制捕捉像素间的长距离依赖,增强全局特征表达。然而两者均存在局限。为此,本文提出一种基于双向互补卷积与Transformer的轻量级多尺度特征融合网络,通过双分支架构融合二者优势,实现全局与局部信息的互补。同时,针对深层网络训练中低像素图像导致的信息损失问题,设计了多阶段特征补全模块,将浅层与深层特征图进行融合,最大限度保留对图像恢复有益的特征。实验结果表明,在参数量相当的情况下,该模型在图像恢复性能上优于其他轻量级模型。

原文摘要 · Abstract (English)

The single image super-resolution(SISR) algorithms under deep learning currently have two main models, one based on convolutional neural networks and the other based on Transformer. The former uses the stacking of convolutional layers with different convolutional kernel sizes to design the model, which enables the model to better extract the local features of the image; the latter uses the self-attention mechanism to design the model, which allows the model to establish long-distance dependencies between image pixel points through the self-attention mechanism and then better extract the global features of the image. However, both of the above methods face their problems. Based on this, this paper proposes a new lightweight multi-scale feature fusion network model based on two-way complementary convolutional and Transformer, which integrates the respective features of Transformer and convolutional neural networks through a two-branch network architecture, to realize the mutual fusion of global and local information. Meanwhile, considering the partial loss of information caused by the low-pixel images trained by the deep neural network, this paper designs a modular connection method of multi-stage feature supplementation to fuse the feature maps extracted from the shallow stage of the model with those extracted from the deep stage of the model, to minimize the loss of the information in the feature images that is beneficial to the image restoration as much as possible, to facilitate the obtaining of a higher-quality restored image. The practical results finally show that the model proposed in this paper is optimal in image recovery performance when compared with other lightweight models with the same amount of parameters.

超分辨率轻量模型特征融合双分支网络

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。