人工智能算法基于注意力机制的单图像超分辨率方法区分纹理和平滑

人工智能 算法
人工智能算法基于注意力机制的单图像超分辨率方法区分纹理和平滑摘要: 单图像超分辨率(SISR)的主要挑战是恢复高频细节,如微小纹理。然而,大多数最先进的方法缺乏识别高频区域的特定模块,导致输出图像模糊。

人工智能算法基于注意力机制的单图像超分辨率方法区分纹理和平滑摘要: 单图像超分辨率(SISR)的主要挑战是恢复高频细节,如微小纹理。然而,大多数最先进的方法缺乏识别高频区域的特定模块,导致输出图像模糊。我们提出了一种基于注意力的方法来区分纹理区域和平滑区域。在定位高频细节的位置之后,执行高频补偿。该方法可以与先前提出的SISR网络合并。通过提供高频增强,实现了更好的性能和视觉效果。我们还提出了由DenseRes块组成的自己的SISR网络。该块提供了组合低级功能和高级功能的有效方法。广泛的基准评估表明,我们提出的方法比SISR中最先进的工作有了显着的改进。

人工智能算法基于注意力机制的单图像超分辨率方法区分纹理和平滑介绍:单图像超分辨率(SISR)的任务是从单个低分辨率(LR)输入图像推断出高分辨率(HR)图像。这是一个非常不适合的问题,因为在低通滤波和下采样期间,诸如微小纹理的高频信息会丢失。因此,SISR是一对多映射。我们的任务是找到最合理的HR图像,尽可能地恢复微小的纹理。

为了从LR图像中恢复HR图像,需要大的接收场来从LR图像中获取更多的上下文信息。使用更深的网络是增加感受野的更好方法。深度网络的一个缺点是消失梯度问题,这使得网络难以训练。他等人。 [1]使用剩余学习框架来简化网络培训。跳过连接是另一种通过网络增强梯度和信息流的解决方案。低级特征包含有效信息,可用于重建HR图像。 SISR将受益于不同层面的集体信息。

SISR的难点在于恢复高频细节,例如微小的纹理。输出图像和原始图像之间的均方误差(MSE)通常用作训练卷积神经网络的损失函数。然而,在追求高峰值信噪比(PSNR)的过程中,MSE将返回许多可能解决方案的平均值,因此输出图像看起来模糊且难以置信。为了恢复高频细节,已经提出了鼓励的感知损失[2]网络产生其特征表示相似的图像,从而产生更清晰的图像。 Ledig在al。 [3],[4]结合了对抗性网络,感知损失和纹理损失,以鼓励输出图像恢复高频细节,如微小纹理。但是所有这些网络并没有明确地知道高频细节的位置,他们只是试图盲目地恢复纹理。因此,这些网络的性能并不令人满意。

人工智能算法基于注意力机制的单图像超分辨率方法区分纹理和平滑贡献:为了解决这些问题,首先,基于将每个层连接到每个后续层的denseNet [5],我们提出了一个新的块,称为DenseRes块,由残余构建块(Resblock)[1]组成。每个Resblock的输出都连接到每个其他Resblock,从而增强了信息流并避免了重新学习冗余功能。使用DenseRes块,可以缓解梯度消失问题,并且网络易于训练。其次,我们提供了一种应对高频细节恢复的注意机制。受U-net [6]的启发,用于语义像素分割,我们提出了一种新颖的混合密集连接U-net,以帮助网络区分区域是否充满需要修复的微小纹理或类似于插值图像。它作为一个特征选择器,有选择地增强高频功能。因此,纹理可以尽可能地恢复。

这是第一次将注意机制引入SISR。该方法简单有效。通过选择性地提供高频增强,它减轻了输出图像趋于模糊的问题。注意机制可以与先前提出的SISR网络结合。实现了更高的PSNR和SSIM。另一个贡献是我们提出了DenseRes块,它提供了一种组合低级功能和高级功能的有效方法。它有利于恢复高频细节。

我们在四个公开可用的基准数据集上评估我们的模型。它在PSNR和结构相似性(SSIM)指数方面优于当前最先进的方法。至于PSNR,我们分别比VDSR [7]和DRCN [8]提高了0.54 dB和0.52dB。本文的其余部分安排如下:第二部分介绍了包括超分辨率(SR)算法和注意机制的相关工作,第三部分提出了网络结构。第IV部分提供了实验结果和与最新结果的视觉比较。我们在第五节做出结论。

相关工作:SISR,诸如bicubic和Lanczos [9]等早期方法易于实现且速度很快。但是这些方法通常会产生模糊的结果,缺乏高频细节。提出了许多强大的方法,如稀疏编码[10],以建立低分辨率和高分辨率图像之间的复杂映射。稀疏编码[11],[12]基于以下假设:LR字典上的LR图像的稀疏表示与HR字典上的相应高分辨率图像的稀疏表示相同。

最近,基于卷积神经网络(CNN)的算法取得了很好的效果,并且优于其他算法。董等人。 [13]用双三次插值对输入图像进行了放大,然后端到端地训练了一个浅卷积网络,以学习从LR输入到超分辨率输出的非线性映射。随后,各种工作[3],[7],[8]成功地在SISR中使用了深度网络,并且与浅卷积体系结构相比,获得了更高的PSNR值。最近,Lim等人。 [14]在NTIRE2017超分辨率挑战中获得最佳成绩[15]。他们的网络深度高达32。

在许多用于SISR的深度学习算法中,LR图像通过双三次插值作为网络的输入进行上采样[7],[8]。这意味着SISR操作在高分辨率空间中执行,这是次优的并且增加了计算复杂性。代替内插图像,应用子像素卷积层[16]以将特征图上采样到网络的后续层中的地面实况的大小。这可以在保留模型容量的同时减少计算。

注意机制:基于注意机制的方法在一系列任务中表现出良好的性能。在语音识别领域,基于注意力的循环网络解码器用于将语音话语转录为字符[17]。 Chorowski等。 [18]用注意机制提高对长输入语音的鲁棒性。侯等人。 [19]提出了一种简单但有效的注意机制来实现在线语音识别。在机器翻译领域,Ashish等人。 [20]提出了一个新的简单网络,完全基于注意力机制,完全免除了复发和控制,在机器翻译任务中表现出卓越的品质。其他作品[21],[22]也在关注的帮助下取得了不错的成绩。在...方面Zhang等人的目的是识别和定位吸引人类注意力的特殊区域。 [27]设计一个对称的完全卷积网络来提取显着性特征。李等人。 [28]使用弱监督方法,并使用强监督方法获得可比较的结果。

提出方法:在本节中,我们描述了SISR的建议模型体系结构。该网络旨在学习LR图像和HR图像之间的端到端映射函数F.如图1所示,我们的网络由两部分组成:旨在恢复HR图像的特征重建网络和目的是找到要修复的高频细节的注意力产生网络。通过两个网络输出的相乘,我们将得到HR图像的残差。

人工智能算法基于注意力机制的单图像超分辨率方法区分纹理和平滑结论:我们提出了一种基于注意力的方法来区分纹理区域和平滑区域。当定位高频细节的位置时,注意机构用作特征选择器,其增强高频特征并抑制平滑区域中的噪声。因此,我们的方法避免盲目地恢复高频细节。我们将该机制集成到SISR网络中,包括SRResNet,VDSR和DRCN,并且这些SISR网络的性能都得到了改进。因此,验证了注意机制的有效性。至于特征重建网络,我们提出了DenseRes块,它提供了一种组合低级特征和高级特征的有效方法。通过多个DenseRes块的级联,我们的网络具有大的感知域。因此,捕获来自LR图像的大区域中的有用的上下文信息以恢复HR图像中的高频细节。与最先进的方法相比,我们的方法具有最佳性能。在未来,我们将探索视频超分辨率中注意机制的应用,以产生视觉和数量上的高质量结果。

责任编辑:庞桂玉 来源: 今日头条
相关推荐

2021-01-20 11:14:47

人工智能算法技术

2021-02-24 09:30:44

人工智能PULSE超分辨率算法

2022-01-17 17:01:23

深度学习图像人工智能

2021-03-10 15:49:20

人工智能遗传算法

2018-07-12 10:08:31

图像超分辨率重建技术原理

2017-11-10 11:02:36

2017-04-10 09:17:08

神经网络分辨率像素递归

2020-10-09 07:39:46

智能

2018-06-19 09:38:09

人工智能深度学习图像处理

2018-08-26 22:25:36

自注意力机制神经网络算法

2024-03-05 15:45:10

2024-12-03 13:19:47

2018-07-31 09:20:03

人工智能深度学习机器学习

2024-09-19 10:07:41

2022-02-04 23:22:56

人工智能算法医学

2021-09-10 16:24:00

框架AI开发

2024-07-05 10:41:30

目标检测算法

2022-01-24 14:21:27

人工智能识别图像绘画

2019-03-25 17:59:55

人工智能机器学习技术

2022-02-19 23:05:19

人工智能算法核聚变
点赞
收藏

51CTO技术栈公众号