Visual-Semantic Transformer for Scene Text Recognition论文解读及复现

@zhou

于 2023-04-17 21:59:25 发布

阅读量468

点赞数

分类专栏：论文阅读文章标签： transformer 深度学习人工智能计算机视觉

本文链接：https://blog.csdn.net/weixin_45931455/article/details/130209883

版权

论文阅读专栏收录该内容

2 篇文章 0 订阅

订阅专栏

Visual-Semantic Transformer for Scene Text Recognition论文解读及复现

这篇文章就基于上面的这篇论文VST中结构设计进行阅读，并将其项目Visual-Semantic Transformer进行开源。

该论文中的5个主要构成部分

在这篇论文中作者并没有使用现如今场景文本识别任务中把语言规则建模这一方法，而是通过一个Visual-Semantic Alignment Module对齐模块把经过ConvNet和Visual Module提取到的视觉特征 $Visual_1$ 向Semantic特征对齐得到 $Semantic_1$ 。这里的 $Semantic_1$ 可以认为是我们输出的字符序列的高维特征表示，可以直接使用一个线性映射层便可以将高维特征映射为识别任务的输出，因此该论文并不是基于串行的输出，而是推理速度更快的并行输出，效率更好。之后为了将这两个模态的特征进行交互以获得更好的特征表示，作者想到了Transoformer Encoder它可以实现长距离的特征建模，因此作者提出了一个基于Transformer Encoder的Interaction Module去联合的建模两个模态已达到交互的效果，之后分别得到 $Semantic_2,Visual_2$ 。最后作者再次使用Visual-Semantic Alignment Module将 $Visual_2$ 转化为 $Semantic_3$ 。之后还可以加上一个Semantic Module来对其进行语义的推理最终得到融合之后的 $S e man t i c$ 特征之后接入线性层便可以将输入的Tensor转化为sequence。如下图所示，展示该论文中的所有相关Module和数据流向。
VST Architecture
下面的部分一一介绍上面提到的5个在本文中使用到的Module。

ConvNet

本文的特征提取模块使用了Resnet网络，给定一个输出尺寸为 $\times H \times W$ 的图片，我们将会得到一个 $C_1 \times \frac{H}{8} \times \frac{W}{4}$ 的输出。具体的Resnet一共使用了4个layers(1,2,5,3)。

Visual Module

该模块中作者将上面得到的特诊图进行展平进而得到一个一维的向量输入。之后使用Transformer Encoder结构将这些特征进行全局的建模得到输出我将其记做为 $Visual_1$ 。但是不同于Transform Encoder他这里做了一个小小的改进，便是先Layer Norm之后在进行多头注意力机制。

Visual-Semantic Alignment Module

vsalign
如上图所示：我们使用并行的注意力机制在计算每一个位置上字符的概率分布，因此该论文在推理的过程中是并行输出的而不需要Transform Decoder的那种自回归预测。如下面的公式一样可以计算出 $Semantic_1$ 。
$Semantic_1=softmax({Q \times Visual_1^{T}})Visual_1$
上面的 $Q$ 是一个可训练的参数用来表示视觉和语义之间的关系。作者还认为将该模块进行参数共享的另一个作用就是使得下面的交互模块可以有更好的作用。因为第一次使用该模块以后可以有更大的机会去使用交互模块来对齐进行校正，这是在不使用语言规则之后便可以得到的效果。

Interaction Module

简单来说也就是一个Transformer的编码器结构，只不过在MHSA中计算的公式改为下面的计算：
$[Visual_1; Semantic_1] = softmax(\frac{[Visual_1; Semantic_1]}{\sqrt{K}})[Visual_1; Semantic_1]$
除此之外便没有什么其他的东西。经过这个模块之后得到 $Visual_2; Semantic_2]$ ,多的地方就是再次使用VS Align将得到的 $Visual_2$ 转换为
$Semantic_3 = Vsalgin(Visual_2)$ 。

Semantic Module

该模块将 $Semantic_2, Semantic_3$ 进行在样本长度上Concatenate。之后加上Transformer Encoder进行建模。便可以得到融合后的结果 $S e man t i c$ 。

损失函数

由于上面我们有3可用的Semantic( $Semantic_2, Semantic_3, Semantic$ )因此在训练的过程中一共有三个损失，将其相加便可以得到总的损失函数。在推理的过程中可以使用 $S e man t i c$ 进行推理。

复现结果

前三个epoch

由于数据集较大我在训练三圈之后在IC15， IC13， SVTP，SVT， CUTE, IIIT5k数据上的最好的cwr(也就是每一个样本完全预测正确才算是预测正确)平均正确率是89.06。反观论文中在这几个数据集上的平均正确率为92.483。后续出来更好的结果我会在github和这里进行更新。