视频超分，Transformer再下一城之VSR-Transformer

论文：https://arxiv.org/pdf/2106.06847.pdf

代码：https://github.com/caojiezhang/VSR-Transformer

编者言：ETHz(苏黎世联邦理工大学)出品，第一篇在VSR中使用的Transformer模型，方法与思想值得学习。模型与实验没有文章中大批量的数学公式那样惊艳，有数学推理癖好的读者建议直接阅读原文。

看点

Transformer应用在VSR中有两大问题：

全连接的自注意层由于依赖线性层来计算注意图而忽略了对局部信息的利用。
由于词前馈层独立的处理每个输入，导致其缺乏特征对齐的能力。

为了解决第一个问题，本文提出了一个时空卷积自注意层以利用局部信息。对于第二个问题，本文设计了一个双向光流的前馈层来挖掘不同帧之间的相关性并对齐特征。本文是第一个在VSR中应用Transformer的模型。

方法

overview

本文提出的VSR-Transformer由特征提取器、Transformer编码器和重建网络组成。其中特征提取器由堆叠的残差块组成。

时空卷积自注意层

由于全连接的自注意层不能使用每个帧的空间信息，本文提出了一个新的时空卷积自注意层(STCSA)。给定特征图X，使用三个独立的CNN来捕捉每一帧的空间信息。Q，K，V的生成通过如下公式：

其中，是展开操作，然后计算相似度矩阵σ并与值V聚合以获得特征图。最后，采用折叠操作将这些滑动局部patches组合起来，并使用输出层来输出最终特征图，公式如下：

其中，τ◦是reshape操作τ和展开操作的组合。本文中使用单头（h=1）获得了良好的性能。

时空位置编码

VSR-Transformer的结构是置换不变的，而VSR任务需要精确的时空位置信息。为了解决这个问题，本文使用3D固定位置编码将其添加到注意层的输入中。具体地，位置编码包含两个空间位置信息（水平和垂直）和一个时间位置信息，表述如下：

其中，，k是[0,d/6)中的一个整数，‘pos’表示维度位置，d是通道维度。### 双向光流前馈层传统Transformer中的全连接前馈层忽略了利用不同帧的相关性，为了解决这个问题，本文通过使用光流进行空间对齐，提出了双向光流前馈层，如下图左所示。

具体地说，给定时空卷积自注意层输出的特征映射X，首先学习相邻帧之间的双向光流

然后通过翘曲得到后向特征

和前向特征

在本文的实验中使用SPyNet来进行光流估计。然后将视频帧和扭曲的特征图进行聚合。作者修改了全连接的前馈层为：

其中，是一个融合模块。在实验中将两层网络的情形推广到由残差ReLU网络组成的多层神经网络和，公式如下：

与ViT相比，本文能够捕获不同帧之间的相关性。相比BasicVSR，本文避免了对光流和特征的递归，因此允许并行计算。

实验

REDS上的测试在都用5帧来训练的模型上，VSR-Transformer拥有较好的性能，这是因为BasicVSR和IconVSR更加适应长期的视频序列，但是VSR-Transformer的参数量缺惊人的高，这是一大待处理的弊端。

Vid4上的测试

在Vid4上的BI降采样的定量结果，作者分析不如BasicVSR的原因可能是BasicVSR和IconVSR在Vid4的所有帧上测试，而VSR-Transformer只在7帧上测试，亦或是Vimeo-90K-T和Vid4之间可能存在分布偏差。

在只有7帧的Vid4上进行测试。（这个实验有点牵强，自己模型处理长期视频效果不好就得把测试集剪切来证明自己的优越性，建议作者直接改成一帧的然后和单图超分去比）

消融实验

使用定性的方式来进行光流的消融实验。

在REDS上进行其他组件的消融实验。

备注：SR

超分辨率交流群

扫码备注拉你入群。

在看，让更多人看到