CN112329760A - 基于空间变换网络端到端印刷体蒙古文识别翻译的方法 - Google Patents
基于空间变换网络端到端印刷体蒙古文识别翻译的方法 Download PDFInfo
- Publication number
- CN112329760A CN112329760A CN202011290754.7A CN202011290754A CN112329760A CN 112329760 A CN112329760 A CN 112329760A CN 202011290754 A CN202011290754 A CN 202011290754A CN 112329760 A CN112329760 A CN 112329760A
- Authority
- CN
- China
- Prior art keywords
- network
- attention
- layer
- mongolian
- transformation
- Prior art date
- Legal status (The legal status is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the status listed.)
- Granted
Links
Images
Classifications
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06V—IMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
- G06V10/00—Arrangements for image or video recognition or understanding
- G06V10/20—Image preprocessing
- G06V10/22—Image preprocessing by selection of a specific region containing or referencing a pattern; Locating or processing of specific regions to guide the detection or recognition
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06F—ELECTRIC DIGITAL DATA PROCESSING
- G06F40/00—Handling natural language data
- G06F40/40—Processing or translation of natural language
- G06F40/58—Use of machine translation, e.g. for multi-lingual retrieval, for server-side translation for client devices or for real-time translation
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/04—Architecture, e.g. interconnection topology
- G06N3/045—Combinations of networks
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/08—Learning methods
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06V—IMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
- G06V10/00—Arrangements for image or video recognition or understanding
- G06V10/20—Image preprocessing
- G06V10/26—Segmentation of patterns in the image field; Cutting or merging of image elements to establish the pattern region, e.g. clustering-based techniques; Detection of occlusion
- G06V10/267—Segmentation of patterns in the image field; Cutting or merging of image elements to establish the pattern region, e.g. clustering-based techniques; Detection of occlusion by performing operations on regions, e.g. growing, shrinking or watersheds
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06V—IMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
- G06V30/00—Character recognition; Recognising digital ink; Document-oriented image-based pattern recognition
- G06V30/10—Character recognition
- G06V30/14—Image acquisition
- G06V30/148—Segmentation of character regions
- G06V30/153—Segmentation of character regions using recognition of characters or words
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06V—IMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
- G06V30/00—Character recognition; Recognising digital ink; Document-oriented image-based pattern recognition
- G06V30/10—Character recognition
Landscapes
- Engineering & Computer Science (AREA)
- Theoretical Computer Science (AREA)
- Physics & Mathematics (AREA)
- General Physics & Mathematics (AREA)
- General Health & Medical Sciences (AREA)
- Health & Medical Sciences (AREA)
- Multimedia (AREA)
- Artificial Intelligence (AREA)
- Computational Linguistics (AREA)
- General Engineering & Computer Science (AREA)
- Biomedical Technology (AREA)
- Molecular Biology (AREA)
- Computing Systems (AREA)
- Evolutionary Computation (AREA)
- Data Mining & Analysis (AREA)
- Mathematical Physics (AREA)
- Software Systems (AREA)
- Biophysics (AREA)
- Life Sciences & Earth Sciences (AREA)
- Computer Vision & Pattern Recognition (AREA)
- Audiology, Speech & Language Pathology (AREA)
- Image Analysis (AREA)
Abstract
本发明一种基于空间变换网络端到端印刷体蒙古文识别翻译的方法,从蒙古文字的特点出发,使用四阶段的文字识别网络对蒙古文进行识别,空间变换阶段使用空间变换网络对蒙古文字图片进行标准化,其次,综合上下文信息使用CBAM‑GRCNN提取文字特征,以及使用Mogrifier LSTM进行序列建模。在预测阶段,使用聚焦注意力机制解决注意力漂移问题,并结合GRU网络进行预测;在翻译过程中,采用Reformer模型,Reformer模型通过将传统的多头注意力机制改为基于局部敏感哈希的注意力机制,以及将传统残差网络替换为可逆残差网络,并对前馈网络进行分块,从而降低模型的时间与空间复杂度,缓解训练长序列数据内存不足、速度慢的问题。
Description
技术领域
本发明属于光学字符识别(OCR)与机器翻译技术领域,特别涉及一种基于空间变换网络端到端印刷体蒙古文识别翻译的方法。
背景技术
光学字符识别是利用计算机将图片中的文字提取为文本的一种技术,是解决图片转文字问题的最有效手段之一,机器翻译能够将一种语言转换为目标语言,是解决语言障碍的有效方式。随着深度学习的日渐发展,使用深度学习进行光学字符识别任务与机器翻译任务已成为主流,谷歌、百度、有道等针对光学字符识别与机器翻译进行了大量研究,并已开发出实用化的应用程序。
在端到端文字识别网络与序列到序列文字识别网络出现前,印刷体蒙古文识别网络多采用BP、CNN等网络进行文字识别,但传统蒙古语的词汇量巨大,日常使用的词汇可达数十万个。若使用分类器来对每个单词进行分类时不可能的,且不能识别词汇表以外的单词。为了解决上面所述的问题,目前主流方法是使用端到端的网络或者序列到序列的网络对文字进行识别,这两种网络将输入单词图像看作一个图像帧序列,而将输出单词视为一个字母序列。网络可以获得字母和字形之间的关系,先识别字母,然后将其组合成单词。2019年,Weiyuan Wang等人基于双向LSTM和CTC提出端到端的蒙古文印刷体识别。该项工作将蒙古文OCR转换为一个序列到序列的映射问题,无需对蒙古文单词进行切分,LSTM网络将输入图片编码为多维向量,CTC网络将LSTM网络的输出转换为识别结果。该模型可以解决蒙古文数据词汇量不足问题。
在翻译方面,谷歌研究人员提出Transformer模型,一种只使用注意力机制的机器翻译模型。该模型摒弃了传统的编码器-解码器模型必须结合RNN或者CNN的固有模式,使用完全基于注意力机制的方式,在主流语料上达到了不错的效果。
目前光学字符识别在主流语言方面已经发展成熟,且对扭曲变形的文字有着良好的识别率,但在小语种方面研究较少,尤其是数据库严重缺乏,识别翻译没有达到很好的效果。
发明内容
为了克服上述现有技术的缺点,本发明的目的在于提供一种基于空间变换网络端到端印刷体蒙古文识别翻译的方法,使用空间变换网络对数据进行标准化,可以识别扭曲变形的蒙古文字,并可同步将识别出的蒙古文翻译为中文,同时更易于并行化,并且需要更少的时间进行训练。
为了实现上述目的,本发明采用的技术方案是:
一种基于空间变换网络端到端印刷体蒙古文识别翻译的方法,包括:
文字识别,利用基于空间变换网络端到端的印刷体蒙古文识别网络实现,由空间变换、特征提取、序列建模、预测四个阶段组成,其中所述空间变换阶段是利用空间变换网络与薄板样条插值将倾斜扭曲的文本图像矫正,对蒙古文字图片进行标准化,为后续网络训练提供良好的输入;特征提取阶段由CBAM-GRCNN网络实现,负责提取图像中的文字特征,同时抑制不相关的特征,CBAM从空间上与通道上分别关注图像特征,GRCNN使用门结构控制RCL中的上下文信息,同时平衡前馈信息和循环信息;序列建模阶段由Mogrifier LSTM网络实现,负责捕获字符序列中的上下文信息,以便更好地预测每个字符,而不是独立进行预测;预测阶段由聚焦注意力网络与GRU实现,负责根据图像特征预测输出字符序列;
文字翻译,利用Reformer模型,将识别出的蒙古文翻译为汉语。
在文字识别之前,为便于深度学习神经网络更好地提取特征,先对数据进行预处理,所述对数据进行预处理是对印刷体蒙古文进行版面分析以及分割。
所述空间变换网络是一种基于仿射变换的、利用神经网络进行自适应校正的架构,所述仿射变换包含平移、缩放、旋转、错切四种图像变换操作,空间变换网络关注图像区域最主要的特征,抑制图像背景,提高分类性能,该网络使用反向传播算法进行训练,或进行端到端的训练;
所述CBAM-GRCNN网络使用基于卷积注意力模块与门结构的循环卷积层分别从通道与空间上提取图像特征;
所述Mogrifier LSTM网络是使当前时间的输入xt和上一个时间的隐藏状态ht-1在输入LSTM网络之前先进行多次交互,然后将交互后得的向量再作为LSTM网络的输入,从而增强整体模型的上下文建模能力,Mogrifier LSTM网络,通过前馈输入跟隐藏状态的相互交互,能够获得对上下文相关的表达;
所述聚焦注意力网络通过在注意力模块添加局部的监督信息来促使注意力特征与实际标签序列的对齐,主要包括两个步骤:1),计算每个预测标签的注意力中心;2),通过在注意力区域生成概率分布,将注意力集中到目标区域中。
所述空间变换网络包含定位网络、网格生成器、网格采样器三部分,定位网络用于从输入的原始图像或特征图中估计仿射变换的参数矩阵,采样网格生成器通过对一个固定的、均匀的网格应用仿射变换以生成采样网格,网格采样器则使用薄板样条插值算法,根据采样网格对输入图像进行采样,以最终实现自适应校正功能;
所述CBAM是卷积神经网络的注意力模块,CBAM从通道和空间两个维度计算卷积神经网络生成的特征图的注意力图,然后将注意力图与特征图相乘,进行特征的自适应学***衡上下文信息。
所述Reformer模型基于局部敏感哈希的注意力机制,并将可逆残差网络(RevNet)加入到Transformer以减少参数对内存的占用,对Transformer中前馈网络分段处理,以减少在全连接层的内存占用量。
所述Reformer模型基于编码器-解码器框架搭建,主要由词向量编码模块、位置编码模块、编码器和解码器组成,其中编码器由多个编码器子模块堆叠而成,每个编码器子模块中包含一个多头多轮局部敏感哈希注意力层和一个分块前馈网络,在多头多轮局部敏感哈希注意力层和分块前馈网络之后均设置有输出可逆残差层和层标准化层;解码器由多个解码器子模块堆叠组成,每个解码器子模块除包含多头多轮局部敏感哈希注意力层和分块前馈网络之外,还在可逆残差层与层标准化层之前增加一层共享查询关键字向量自掩码层,在多头多轮局部敏感哈希注意力层、分块前馈网络以及共享查询关键字向量自掩码层之后均设置有输出可逆残差层和层标准化层。
与现有技术相比,本发明的有益效果是:
1、本发明使用四阶段端到端的文字识别网络作为印刷体蒙古文识别模型,因为它无需对蒙古文字进行分割,同时可以识别出数据集以外的文字,从而提升识别准确率。
2、本发明以印刷体蒙古文为研究对象,针对倾斜扭曲变形的蒙古文图片识别困难的问题,拟使用空间变换网络对输入进行标准化,为后续的模型训练提供有力的支持。
3、本发明针对上下文信息平衡问题,在特征提取层使用带有门结构的循环卷积层GRCNN提取图像特征,通过使用门结构来控制RCL中的上下文信息,同时平衡前馈信息和循环信息,并使用CBAM机制关注图像特征。
4、本发明针对注意力漂移(attention drift)问题,使用聚焦注意力网络(Focusing Attention Network))结合GRU对序列建模阶段的特征进行预测,得到识别结果。
5、本发明针对Transformer模型多头注意力机制计算时间复杂度高的问题,使用基于局部敏感哈希的注意力机制。局部敏感哈希通过使用哈希函数将超大集合划分为多个较小的子集,每个子集中的元素个数较小且相邻,查找数据时先找到查询点所在的子集空间,然后在子集空间内进行查找,这样便将在超大集合内查找相邻元素的问题转化为在小集合内查找相邻元素的问题,大幅减少了查找时间与计算量。
6、本发明针对Transformer模型中ResNet层数加深后,需要储存每一层的输入用于反向传播,导致内存消耗过大的问题,将可逆残差网络加入到Transformer里面来减少参数对内存的占用。采用这种方式,不需要记录中间层的输入,而只需储存最后一层的输出,从而通过模型的特定结构,反推出中间层的结果
7、本发明针对Transformer模型中FFN的全连接层内存占用量问题,对FFN进行分段的处理。
附图说明
图1为端到端的四阶段印刷体蒙古文识别流程示意图。
图2为基于Reformer的蒙汉翻译流程示意图。
图3为空间变换网络架构示意图。
图4为卷积注意力模块示意图。
图5为GRCL架构示意图。
图6为5轮更新的Mogrifier LSTM示意图。
图7为聚焦注意力机制原理示意图。
具体实施方式
下面结合附图和实施例详细说明本发明的实施方式。
本发明一种基于空间变换网络端到端印刷体蒙古文识别翻译的方法,包括文字识别和文字翻译两大步骤。并且,在文字识别之前,为便于深度学习神经网络更好地提取特征,可先对数据进行预处理,预处理主要是对印刷体蒙古文进行版面分析以及分割。
文字识别利用基于空间变换网络端到端的印刷体蒙古文识别网络实现,参考图1,本发明从蒙古文字的特点出发,由空间变换(Trans.)、特征提取(Feat.)、序列建模(Seq.)、预测(Pred.)四个阶段实现识别,其中空间变换阶段是利用空间变换网络与薄板样条插值将倾斜扭曲的文本图像矫正,对蒙古文字图片进行标准化、规范化,为后续网络训练提供良好的输入;特征提取阶段由CBAM-GRCNN网络实现,综合上下文信息提取图像中的文字特征,同时抑制不相关的特征,如字体、颜色、大小、背景,CBAM从空间上与通道上分别关注图像特征,GRCNN使用门结构控制RCL中的上下文信息,同时平衡前馈信息和循环信息;序列建模阶段由Mogrifier LSTM网络实现,负责捕获字符序列中的上下文信息,以便更好地预测每个字符,而不是独立进行预测;预测阶段由聚焦注意力网络与GRU实现,负责根据图像特征预测输出字符序列,聚焦注意力机制能够较好地解决注意力漂移问题,提升模型识别准确率,结合GRU网络可进行预测。
在文字翻译环节,为解决Transformer模型空间与时间复杂度高的问题,利用Reformer模型进行蒙汉翻译,将识别出的蒙古文翻译为汉语。参考图2,Reformer模型将传统的多头注意力机制改为基于局部敏感哈希的注意力机制(locality-sensitve hashing(LSH)attention),以及将可逆残差网络(RevNet)的思想加入到Transformer里面来减少参数对内存的占用,对Transformer中前馈网络(feed forward neural network,FFN)分段处理,因为FFN中的输入之间互相独立,进行分段的处理可以减少在全连接层的内存占用量。RevNet的提出是为解决ResNet层数加深后,需要储存每一层的输入用于反向传播,导致内存消耗过大的问题。在Transformer中也存在相同的问题,采用这种方式,不需要记录中间层的输入,而只需储存最后一层的输出,从而通过模型的特定结构,反推出中间层的结果。Reformer使用以上三个方法将Transformer的复杂度由O(L2)降低为O(LlogL),同时模型表现效果与Transformer几乎相同。Reformer模型可降低模型的时间与空间复杂度,缓解训练长序列数据内存不足、速度慢的问题。
具体地,Reformer模型与Transformer模型类似,是基于编码器-解码器框架搭建的,模型主要由词向量(Word Embedding)编码模块、位置编码模块(PositionalEmbedding)、编码器(Encoder)和解码器(Decoder)组成,其中编码器是由多个编码器子模块堆叠而成的,每个编码器子模块中包含一个多头多轮局部敏感哈希注意力层(Multi-Round LSH Attention)和一个分块前馈网络(Chunking Feed Forward),在这两个子层之后均设置有输出可逆残差层和层标准化层,以进行输出可逆残差连接和层标准化(LayerNormalization)操作。解码器结构与编码器类似,也是由多个解码器子模块堆叠组成,每个解码器子模块除包含多头多轮局部敏感哈希注意力层和分块前馈网络之外,还在二者之前增加一层共享查询关键字向量自掩码层(Causal Masking For Shared-QK Attention),增加掩码的目的是为在解码时预测当前词只与前面解码出的词有关系,从而使模型的训练和预测保持一致。同样的,在解码器子模块中的上述三个子层之后,均设置有输出可逆残差层和层标准化层,保证模型的正常收敛。
其中,空间变换网络是一种基于仿射变换的、利用神经网络进行自应校正的架构,仿射变换包含平移、缩放、旋转、错切四种图像变换操作,空间变换网络关注图像区域最主要的特征,抑制图像背景,提高分类性能,同时该网络可以使用反向传播算法进行训练,也可进行端到端的训练。参考图3,一个完整的空间变换网络(STN)模块包含定位网络、网格生成器、网格采样器三部分,定位网络用于从输入的原始图像或特征图中估计仿射变换的参数矩阵A,采样网格生成器通过对一个固定的、均匀的网格G0应用参数为A的仿射变换以生成采样网格G,网格采样器则使用薄板样条插值算法(TPS),根据采样网格G对输入图像I进行采样,以最终实现STN的自适应校正功能。空间变换网络在训练过程中几乎与原有的网络模型训练时间相同,它相当于一个加入的组件块,该组件块的空间变换能力对训练数据的旋转、平移、尺度变换、扭曲等多样性有很好的纠正能力提升了网络的学习能力。
CBAM-GRCNN网络使用基于卷积注意力模块与门结构的循环卷积层分别从通道与空间上提取图像特征,通过使用门结构来控制RCL中的上下文信息,同时平衡前馈信息和循环信息;卷积块注意力模型(Convolutional Block Attention Module,CBAM),是一种为卷积神经网络设计的简单有效的注意力模块,CBAM从通道和空间两个维度计算卷积神经网络生成的特征图的注意力图,然后将注意力图与特征图相乘,进行特征的自适应学习,对特征重新赋予权重,权重高的特征即注意力的注意点;这种通道信息与空间信息两个维度相结合的学习方式,可以强调局部关键特征,抑制无关特征,增强网络表达特征的能力。
循环卷积神经网络在卷积神经网络的基础上引入循环连接,循环连接在不同的位置共享权值且具有局部连通性。随着循环连接次数不断增加,循环卷积层中的每个单元可以体现更丰富的上下文信息。通过将门结构加入到RCNN中,可以抑制上下文无关信息,更好的平衡上下文信息,加入门结构的循环卷积神经网络称为GRCNN。
图4所示为卷积注意力模块。卷积注意力模块主要由通道注意力模块和空间注意力模块组成。输入特征图F分别经过全局最大池化和全局平均池化,之后将其输入到多层感知器,将多层感知器输出的两个特征向量相加并使用sigmoid进行激活,得到通道注意力矩阵Mc。将通道注意力权重矩阵与输入特征图F点乘生成通道注意力模块的特征图F'。空间注意力模块首先在通道维度上采用最大池化和平均池化对特征图F'进行处理,之后将两个特征图串行联结为一个特征图,使用卷积层进一步提取特征,最后经过sigmoid函数进行激活,生成空间注意力特征矩阵Ms。将空间注意力特征矩阵和特征图F'进行点乘,得到特征图F”,将输入的特征图F与经过CBAM的特征图F”相加得到下一层卷积的输入。
图5所示为T=2时的GRCL架构。当t=0时,只进行前馈计算,当t=1时,由前馈输入与t=0时的步长状态决定门的输出并作用于循环组件。当所有门的输出为1时,网络变为RCL,当所有门的输出为0时,循环信号丢失,网络成为卷积层。GRCL是RCL的一种变体,可以动态的调整上下文信息,且上层特征图中每个GRCL单元的有效感受野随网络迭代次数增加而变大。
Mogrifier LSTM网络是LSTM的进一步优化,其通过前馈输入跟隐藏状态的相互交互能够获得对上下文相关的表达,从而解决上下文不相关问题。Mogrifier LSTM并不是对LSTM作出真正意义上的修改或者形变,只是使当前时间的输入xt和上一个时间的隐藏状态ht-1在输入LSTM网络之前先进行多次交互,然后将交互后得的向量再作为LSTM网络的输入,通过这种预交互的方法可以增强整体模型的上下文建模能力。
图6所示为5轮更新的Mogrifier LSTM。前一状态h0=hprev与门x-1=x通过sigmoid激活函数产生x1。线性变换x1与门h0产生h2,在多次重复门控循环后,h*与x*序列最后的值被输入到一个LSTM细胞。
聚焦注意力网络通过在注意力模块添加局部的监督信息来促使注意力特征与实际标签序列的对齐,在一定程度上解决了注意力漂移问题。该过程主要包括两个步骤:1),计算每个预测标签的注意力中心;2),通过在注意力区域生成概率分布,将注意力集中到目标区域中。
图7所示为聚焦注意力机制原理。a,c,g,+分别表示校准因子,每个特征在输入图像的中心,瞥见向量和聚焦操作,蓝色网格和绿色网格分别表示每个像素上的裁剪特征与预测结果。为了预测第t-th目标,首先对CNN-LSTM中每个特征向量hj求中心位置ct,j,然后计算所有中心的加权和,得到一个加权位置ct,然后从输入图像或卷积输出中裁剪一小块特征,并对关注区域进行聚焦操作。
具体地,本发明的完整流程包括:
1、端到端的四阶段印刷体蒙古文识别网络,包括以下几个部分:
A、空间变换阶段
空间变换阶段采用空间变换网络对输入图片进行矫正,使得网络在不需要数据增强的情况下,根据具体任务自适应地完成图像数据的变换、对齐等操作,增强网络的空间不变性,提高识别的准确性。空间变换网络主要由定位网络、网格生成器、网格采样器三部分组成。
定位网络的目的是得到空间变换所需的参数λ,输入为特征图U∈RH×W×C,其中,H为特征图的高,W为特征图的宽,C为特征图的通道数,λ为输出变换参数。之后,具有参数λ的变换函数Tλ(G)将作用于原图。定位网络由全连接层构成,最后包含一个回归层产生变换参数。
网格生成器的主要任务为生成参数化的采样网格,为了对输入执行变换,通过对输入图像的特定位置应用采样核来计算每个输出像素。输出像素被定义在规则网格上G={Gi},像素输出像素的集合形成输出图像M∈RH'×W'×C,其中H'和W'分别为输出网格的高度和宽度,输入和输出的通道数一致。
上式中:为规则网格的目标坐标;为输入图像的源坐标,它定义了所需的采样点,网络根据坐标采样以获得期望的变换输出。网格生成器首先创建形状为(H,W)的归一化网格,即覆盖整个输入的一组索引由于算法应用了2D仿射变换,将定位网络生成的参数λ重塑为2×3矩阵,并执行上式的乘法运算,从而产生所需的参数化采样网格。
使用如下变换找到输入图像中感兴趣的区域,如下:
上式允许通过改变V、tx和ty来进行裁剪、平移和缩放操作。
采样器利用采样网格和输入的特征图同时作为输入产生输出,并通过采样核对输入特征图中的像素进行采样,将采样的像素值放入输出特征图中对应的坐标位置。公式如下:
薄板样条插值法将插值问题模拟为一个薄金属板在点约束下的弯曲变形。用离散点数据插值得到曲面,薄板样条插值具有光滑、连续、弹性好的特点。
其中,a0,a1,a2,Fi(i=1,2,…,n)为待定系数;ψ为调节曲率大小的经验参数;ri 2为待插点到已知点距离的平方:
ri 2=(x-xi)2+(y-yi)2
上式中的n+3个未知数可以通过下式求得:
其中,cj=16πD/kj,kj是关于点j的弹性常数,一般取cj=0,以使求出的曲面样条函数在已知点与原始数据吻合。
对上述方程组求解,得到待定参数值后,即可求得平面内任一点的近似值W(x,y)。
B、卷积注意力模块
卷积注意力模块包括通道注意力模块和空间注意力模块,下面分别对其进行讨论。
(1)通道注意力模块
输入特征图F分别经过全局最大池化和全局平均池化,得到两个1×1×N的特征向量,之后将其输入到多层感知器,将多层感知器输出的两个特征向量相加并使用sigmoid进行激活,得到通道注意力矩阵Mc。通道注意力矩阵可以表示为通道的权重,输入特征图中重要信息所占通道权重大,不重要信息所占通道权重小。将通道注意力权重矩阵与输入特征图F点乘生成通道注意力模块的特征图F'。
通道注意力模块运算过程如下公式:
Mc(F)=σ(MLP(AvgPoll(F))+MLP(MaxPoll(F)))
F'=Mc(F)×F
其中,F为输入特征图,σ为sigmoid激活函数,MLP为多层感知器,Avgpoll为平均池化,Maxpoll为最大池化,Mc为通道注意力矩阵,F'为通道注意力机制输出的特征。
(2)空间注意力模块
空间注意力模块首先在通道维度上采用最大池化和平均池化对特征图F'进行处理,之后将两个W×H×1的特征图串行联结为一个W×H×2的特征图,使用卷积层进一步提取特征,得到W×H×1的特征图,最后经过sigmoid函数进行激活,生成空间注意力特征矩阵Ms。空间注意力特征矩阵表示为每个像素所占通道的权重,具有重要信息的像素所占通道权重大,具有不重要信息的像素所占通道权重小。将空间注意力特征矩阵和特征图F'进行点乘,得到特征图F”,将输入的特征图F与经过CBAM的特征图F”相加得到下一层卷积的输入。
空间注意力模块运算过程如下式:
Ms(F')=σ(f[AvgPoll(F');MaxPoll(F')])
F”=Ms(F')×F'
其中,f为卷积层,;为串行联结,Ms为空间注意力模块,F”为空间注意力机制输出特征。
经过卷积注意力模块,网络更加关注重要的局部特征,过滤不重要的局部特征,增强网络对特征的学习能力。
C、特征提取阶段使用GRCNN网络提取标准化后的图像特征,RCNN模型通过在前馈输入与时间域上展开研究,尽可能的获取上下文信息。在时间域上展开的RCNN是一个从输入层到输出层之间有多条路径的CNN,这个模块称之为循环卷积层。
一个具有前馈输入ωt的RNN模型,内部状态xt公式如下:
其中*表示卷积,ω(t)和x(t-1)分别为前馈输入与循环输入,wf和wr分别表示前馈权重与循环权重。
门控循环卷积层(GRCL)是GRCNN的基本模块,这个模块使用门来控制RCL的上下文信息,可以抑制无关的上下文信息,GRCL的门可以如下式表示:
上式中,⊙为element-wise乘法,BN在每个卷积操作与element-wise后应用,在不同的时间步长上,BN的参数与统计量不共享。
向量交互更新公式如下:
xi=2σ(Qihi-1)xi-2 for odd i∈[1...r]
hi=2σ(Rixi-1)hi-2 for even i∈[1...r]
其中,轮数r为超参数,r=0时模型退化为LSTM,矩阵Qi,Ri为随机初始化矩阵,式中乘以一个常数2,是因为经过sigmoid激活函数后,其值分布在(0,1),这样反复乘下去,值会逐渐趋近于0。因此乘以一个2保证其数值的稳定性。
E、预测阶段使用聚焦注意力机制与GRU,聚焦注意力网络主要包括两个步骤:1.计算注意力中心2.关注目标区域。
计算注意力中心:在卷积或者最大池化操作中,将输入定义为N×Di×Hi×Wi,输出定义为N×DO×HO×WO,其中N,D,H,W分别表示批次大小、通道数量、特征图的高度和宽度。根据卷积的参数kernal,stride,pad,那么:
HO=(Hi+2×padH-kernelH)/strideH+1
WO=(Wi+2×padW-kernelW)/strideW+1
因此,第L层(x,y)处对于第L-1层的感受野r=(xmin,xmax,ymin,ymax)计算如下:
xmin=(x-1)×strideW+1-padW
xmax=(x-1)×strideW-padW+kernelW
ymin=(y-1)×strideH+1-padH
ymax=(y-1)×strideH-padH+kernelH
对于时间步长t,根据上式,可以计算hj(编码器输出的第j个特征向量),对于输入图片的感受野,并且选择感受野的中心作为注意力中心:
ct,j=location(j)
其中j是hj的索引,location表示计算感受野中心的函数。因此,输入图片对应的注意力中心yt计算如下:
其中,αt,j表示解码器在时间步长t对于特征向量hj的注意力权重。
通过裁剪后的特征图,计算注意力区域的能量分布如下:
其中,K为类别数量。
那么,定义聚焦损失函数为:
2、基于Reformer的蒙汉翻译模型,包括以下几个部分:
A、局部敏感哈希的注意力机制
重新定义Transformer的注意力机制公式,对于一个查询位置i的时间为:
在散列过程中,会存在一个小概率的情况将相同项分到不同的桶中,这个概率可以通过用nrounds个不同的散列函数{h(1),h(2),...}做多轮散列来降低,这种方法称为Multi-round LSH attention,公式如下:
B、可逆残差网络
RevNet的核心思想是每层的激活可以根据下层的激活推导获得,从而不需要在内存中储存激活,以减少模型对内存的需求。在标准残差层中,输出激活是由y=x+R(x)得到。其中R是残差函数。
在RevNet中,先将输入x分为两个部分x1和x2,然后通过残差函数:R(·)和G(·),得到输出y1和y2,公式如下:
y1=x1+R(x2)
y2=x2+G(x1)
x2=y2-G(y1)
x1=y1-R(x2)
通过将RevNet应用到Transformer中,将sub-encoder block之中的ResNet替换为RevNet,从而减少内存的消耗。我们令F函数作为我们的attention层,G函数作为FFN层,公式如下:
Y1=X1+Attention(X2)
Y2=X2+FeedForward(Y1)
C、分块前馈网络
在FFN中,通常中间隐藏层的维度会非常大,如df f=4k或者更大。FFN的输入是独立的,为了降低内存的使用,可以进行拆分计算,每次计算一块,公式如下:
本发明的整体流程如下:
(1)对印刷体蒙古文图片进行版面分析与分割;
(2)使用空间变换网络对输入标准化;
(3)使用CBAM-GRCNN提取图像特征;
(4)使用Mogrifier LSTM进行序列建模;
(5)使用聚焦注意力机制与GRU进行预测;
(6)使用准确率评价印刷体蒙古文识别模型;
(7)对蒙古语进行预处理;
(8)搭建Reformer的解码器;
(9)搭建Reformer的编码器;
(10)进行网络训练;
(11)使用BLEU值评价蒙汉翻译模型。
Claims (6)
1.一种基于空间变换网络端到端印刷体蒙古文识别翻译的方法,其特征在于,包括:
文字识别,利用基于空间变换网络端到端的印刷体蒙古文识别网络实现,由空间变换、特征提取、序列建模、预测四个阶段组成,其中所述空间变换阶段是利用空间变换网络与薄板样条插值将倾斜扭曲的文本图像矫正,对蒙古文字图片进行标准化,为后续网络训练提供良好的输入;特征提取阶段由CBAM-GRCNN网络实现,负责提取图像中的文字特征,同时抑制不相关的特征,CBAM从空间上与通道上分别关注图像特征,GRCNN使用门结构控制RCL中的上下文信息,同时平衡前馈信息和循环信息;序列建模阶段由Mogrifier LSTM网络实现,负责捕获字符序列中的上下文信息,以便更好地预测每个字符,而不是独立进行预测;预测阶段由聚焦注意力网络与GRU实现,负责根据图像特征预测输出字符序列;
文字翻译,利用Reformer模型,将识别出的蒙古文翻译为汉语。
2.根据权利要求1所述基于空间变换网络端到端印刷体蒙古文识别翻译的方法,其特征在于,在文字识别之前,为便于深度学习神经网络更好地提取特征,先对数据进行预处理,所述对数据进行预处理是对印刷体蒙古文进行版面分析以及分割。
3.根据权利要求1所述基于空间变换网络端到端印刷体蒙古文识别翻译的方法,其特征在:
所述空间变换网络是一种基于仿射变换的、利用神经网络进行自适应校正的架构,所述仿射变换包含平移、缩放、旋转、错切四种图像变换操作,空间变换网络关注图像区域最主要的特征,抑制图像背景,提高分类性能,该网络使用反向传播算法进行训练,或进行端到端的训练;
所述CBAM-GRCNN网络使用基于卷积注意力模块与门结构的循环卷积层分别从通道与空间上提取图像特征;
所述Mogrifier LSTM网络是使当前时间的输入xt和上一个时间的隐藏状态ht-1在输入LSTM网络之前先进行多次交互,然后将交互后得的向量再作为LSTM网络的输入,从而增强整体模型的上下文建模能力,Mogrifier LSTM网络,通过前馈输入跟隐藏状态的相互交互,能够获得对上下文相关的表达;
所述聚焦注意力网络通过在注意力模块添加局部的监督信息来促使注意力特征与实际标签序列的对齐,主要包括两个步骤:1),计算每个预测标签的注意力中心;2),通过在注意力区域生成概率分布,将注意力集中到目标区域中。
4.根据权利要求3所述基于空间变换网络端到端印刷体蒙古文识别翻译的方法,其特征在于,所述空间变换网络包含定位网络、网格生成器、网格采样器三部分,定位网络用于从输入的原始图像或特征图中估计仿射变换的参数矩阵,采样网格生成器通过对一个固定的、均匀的网格应用仿射变换以生成采样网格,网格采样器则使用薄板样条插值算法,根据采样网格对输入图像进行采样,以最终实现自适应校正功能;
所述CBAM是卷积神经网络的注意力模块,CBAM从通道和空间两个维度计算卷积神经网络生成的特征图的注意力图,然后将注意力图与特征图相乘,进行特征的自适应学***衡上下文信息。
5.根据权利要求1所述基于空间变换网络端到端印刷体蒙古文识别翻译的方法,其特征在于,所述Reformer模型基于局部敏感哈希的注意力机制,并将可逆残差网络加入到Transformer以减少参数对内存的占用,对Transformer中前馈网络分段处理,以减少在全连接层的内存占用量。
6.根据权利要求1或5所述基于空间变换网络端到端印刷体蒙古文识别翻译的方法,其特征在于,所述Reformer模型基于编码器-解码器框架搭建,主要由词向量编码模块、位置编码模块、编码器和解码器组成,其中编码器由多个编码器子模块堆叠而成,每个编码器子模块中包含一个多头多轮局部敏感哈希注意力层和一个分块前馈网络,在多头多轮局部敏感哈希注意力层和分块前馈网络之后均设置有输出可逆残差层和层标准化层;解码器由多个解码器子模块堆叠组成,每个解码器子模块除包含多头多轮局部敏感哈希注意力层和分块前馈网络之外,还在可逆残差层与层标准化层之前增加一层共享查询关键字向量自掩码层,在多头多轮局部敏感哈希注意力层、分块前馈网络以及共享查询关键字向量自掩码层之后均设置有输出可逆残差层和层标准化层。
Priority Applications (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202011290754.7A CN112329760B (zh) | 2020-11-17 | 2020-11-17 | 基于空间变换网络端到端印刷体蒙古文识别翻译的方法 |
Applications Claiming Priority (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202011290754.7A CN112329760B (zh) | 2020-11-17 | 2020-11-17 | 基于空间变换网络端到端印刷体蒙古文识别翻译的方法 |
Publications (2)
Publication Number | Publication Date |
---|---|
CN112329760A true CN112329760A (zh) | 2021-02-05 |
CN112329760B CN112329760B (zh) | 2021-12-21 |
Family
ID=74321220
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
CN202011290754.7A Active CN112329760B (zh) | 2020-11-17 | 2020-11-17 | 基于空间变换网络端到端印刷体蒙古文识别翻译的方法 |
Country Status (1)
Country | Link |
---|---|
CN (1) | CN112329760B (zh) |
Cited By (10)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN113065432A (zh) * | 2021-03-23 | 2021-07-02 | 内蒙古工业大学 | 一种基于数据增强和ECA-Net的手写体蒙古文识别方法 |
CN113420760A (zh) * | 2021-06-22 | 2021-09-21 | 内蒙古师范大学 | 一种基于分割和形变lstm的手写体蒙古文检测和识别方法 |
CN113505775A (zh) * | 2021-07-15 | 2021-10-15 | 大连民族大学 | 一种基于字符定位的满文单词识别方法 |
CN113539297A (zh) * | 2021-07-08 | 2021-10-22 | 中国海洋大学 | 一种用于声音分类的联合注意力机制模型、方法及应用 |
CN113537459A (zh) * | 2021-06-28 | 2021-10-22 | 淮阴工学院 | 一种药品库房温湿度预测方法 |
CN113627349A (zh) * | 2021-08-12 | 2021-11-09 | 南京信息工程大学 | 一种基于自注意力变换网络的动态人脸表情识别方法 |
CN113807340A (zh) * | 2021-09-07 | 2021-12-17 | 南京信息工程大学 | 一种基于注意力机制的不规则自然场景文本识别方法 |
CN113822276A (zh) * | 2021-09-30 | 2021-12-21 | 中国平安人寿保险股份有限公司 | 基于神经网络的图片矫正方法、装置、设备及介质 |
CN114898439A (zh) * | 2022-06-01 | 2022-08-12 | 桂林电子科技大学 | 一种基于分段策略和多头卷积注意力的视频人脸识别方法 |
CN117710969A (zh) * | 2024-02-05 | 2024-03-15 | 安徽大学 | 一种基于深度神经网络的细胞核分割和分类方法 |
Citations (10)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
JP2009045597A (ja) * | 2007-08-22 | 2009-03-05 | Nippon Steel Corp | タール改質用触媒、当該触媒の製造方法、及び当該触媒を用いたタールの水蒸気改質方法 |
CN107967262A (zh) * | 2017-11-02 | 2018-04-27 | 内蒙古工业大学 | 一种神经网络蒙汉机器翻译方法 |
CN109492232A (zh) * | 2018-10-22 | 2019-03-19 | 内蒙古工业大学 | 一种基于Transformer的增强语义特征信息的蒙汉机器翻译方法 |
CN109508462A (zh) * | 2018-10-25 | 2019-03-22 | 内蒙古工业大学 | 一种基于编码器-解码器的神经网络蒙汉机器翻译方法 |
CN110399798A (zh) * | 2019-06-25 | 2019-11-01 | 朱跃飞 | 一种基于深度学习的离散图片文件信息提取***及方法 |
CN110598221A (zh) * | 2019-08-29 | 2019-12-20 | 内蒙古工业大学 | 利用生成对抗网络构造蒙汉平行语料提高蒙汉翻译质量的方法 |
CN110738062A (zh) * | 2019-09-30 | 2020-01-31 | 内蒙古工业大学 | 一种gru神经网络蒙汉机器翻译方法 |
CN110765966A (zh) * | 2019-10-30 | 2020-02-07 | 哈尔滨工业大学 | 一种面向手写文字的一阶段自动识别与翻译方法 |
CN111368035A (zh) * | 2020-03-03 | 2020-07-03 | 新疆大学 | 一种基于神经网络的汉维-维汉机构名词典的挖掘*** |
CN111382583A (zh) * | 2020-03-03 | 2020-07-07 | 新疆大学 | 一种混合多种策略的汉语-维吾尔人名翻译*** |
-
2020
- 2020-11-17 CN CN202011290754.7A patent/CN112329760B/zh active Active
Patent Citations (10)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
JP2009045597A (ja) * | 2007-08-22 | 2009-03-05 | Nippon Steel Corp | タール改質用触媒、当該触媒の製造方法、及び当該触媒を用いたタールの水蒸気改質方法 |
CN107967262A (zh) * | 2017-11-02 | 2018-04-27 | 内蒙古工业大学 | 一种神经网络蒙汉机器翻译方法 |
CN109492232A (zh) * | 2018-10-22 | 2019-03-19 | 内蒙古工业大学 | 一种基于Transformer的增强语义特征信息的蒙汉机器翻译方法 |
CN109508462A (zh) * | 2018-10-25 | 2019-03-22 | 内蒙古工业大学 | 一种基于编码器-解码器的神经网络蒙汉机器翻译方法 |
CN110399798A (zh) * | 2019-06-25 | 2019-11-01 | 朱跃飞 | 一种基于深度学习的离散图片文件信息提取***及方法 |
CN110598221A (zh) * | 2019-08-29 | 2019-12-20 | 内蒙古工业大学 | 利用生成对抗网络构造蒙汉平行语料提高蒙汉翻译质量的方法 |
CN110738062A (zh) * | 2019-09-30 | 2020-01-31 | 内蒙古工业大学 | 一种gru神经网络蒙汉机器翻译方法 |
CN110765966A (zh) * | 2019-10-30 | 2020-02-07 | 哈尔滨工业大学 | 一种面向手写文字的一阶段自动识别与翻译方法 |
CN111368035A (zh) * | 2020-03-03 | 2020-07-03 | 新疆大学 | 一种基于神经网络的汉维-维汉机构名词典的挖掘*** |
CN111382583A (zh) * | 2020-03-03 | 2020-07-07 | 新疆大学 | 一种混合多种策略的汉语-维吾尔人名翻译*** |
Non-Patent Citations (4)
Title |
---|
JEONGHUN BAEK等: "What Is Wrong With Scene Text Recognition Model Comparisons?Dataset and Model Analysis", 《IEEE》 * |
WEIYUAN WANG等: "End-to-End Model Based on Bidirectional LSTM and CTC for Segmentation-free Traditional Mongolian Recognition", 《IEEE》 * |
XIANGCHENG DU等: "SCENE TEXT RECOGNITION WITH TEMPORAL CONVOLUTIONAL ENCODER", 《ARXIV》 * |
高芬等: "基于 Transformer的蒙汉神经机器翻译研究", 《计算机应用与软件》 * |
Cited By (16)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN113065432A (zh) * | 2021-03-23 | 2021-07-02 | 内蒙古工业大学 | 一种基于数据增强和ECA-Net的手写体蒙古文识别方法 |
CN113420760A (zh) * | 2021-06-22 | 2021-09-21 | 内蒙古师范大学 | 一种基于分割和形变lstm的手写体蒙古文检测和识别方法 |
CN113537459B (zh) * | 2021-06-28 | 2024-04-26 | 淮阴工学院 | 一种药品库房温湿度预测方法 |
CN113537459A (zh) * | 2021-06-28 | 2021-10-22 | 淮阴工学院 | 一种药品库房温湿度预测方法 |
CN113539297A (zh) * | 2021-07-08 | 2021-10-22 | 中国海洋大学 | 一种用于声音分类的联合注意力机制模型、方法及应用 |
CN113505775A (zh) * | 2021-07-15 | 2021-10-15 | 大连民族大学 | 一种基于字符定位的满文单词识别方法 |
CN113505775B (zh) * | 2021-07-15 | 2024-05-14 | 大连民族大学 | 一种基于字符定位的满文单词识别方法 |
CN113627349A (zh) * | 2021-08-12 | 2021-11-09 | 南京信息工程大学 | 一种基于自注意力变换网络的动态人脸表情识别方法 |
CN113627349B (zh) * | 2021-08-12 | 2023-12-05 | 南京信息工程大学 | 一种基于自注意力变换网络的动态人脸表情识别方法 |
CN113807340A (zh) * | 2021-09-07 | 2021-12-17 | 南京信息工程大学 | 一种基于注意力机制的不规则自然场景文本识别方法 |
CN113807340B (zh) * | 2021-09-07 | 2024-03-15 | 南京信息工程大学 | 一种基于注意力机制的不规则自然场景文本识别方法 |
CN113822276A (zh) * | 2021-09-30 | 2021-12-21 | 中国平安人寿保险股份有限公司 | 基于神经网络的图片矫正方法、装置、设备及介质 |
CN114898439A (zh) * | 2022-06-01 | 2022-08-12 | 桂林电子科技大学 | 一种基于分段策略和多头卷积注意力的视频人脸识别方法 |
CN114898439B (zh) * | 2022-06-01 | 2024-03-19 | 桂林电子科技大学 | 一种基于分段策略和多头卷积注意力的视频人脸识别方法 |
CN117710969A (zh) * | 2024-02-05 | 2024-03-15 | 安徽大学 | 一种基于深度神经网络的细胞核分割和分类方法 |
CN117710969B (zh) * | 2024-02-05 | 2024-06-04 | 安徽大学 | 一种基于深度神经网络的细胞核分割和分类方法 |
Also Published As
Publication number | Publication date |
---|---|
CN112329760B (zh) | 2021-12-21 |
Similar Documents
Publication | Publication Date | Title |
---|---|---|
CN112329760B (zh) | 基于空间变换网络端到端印刷体蒙古文识别翻译的方法 | |
CN110443143B (zh) | 多分支卷积神经网络融合的遥感图像场景分类方法 | |
CN110287800B (zh) | 一种基于sgse-gan的遥感图像场景分类方法 | |
CN111783705B (zh) | 一种基于注意力机制的文字识别方法及*** | |
CN108960143B (zh) | 一种高分辨率可见光遥感图像中的舰船检测深度学习方法 | |
CN110490946B (zh) | 基于跨模态相似度和生成对抗网络的文本生成图像方法 | |
CN111950453B (zh) | 一种基于选择性注意力机制的任意形状文本识别方法 | |
CN111428718B (zh) | 一种基于图像增强的自然场景文本识别方法 | |
CN113657124B (zh) | 基于循环共同注意力Transformer的多模态蒙汉翻译方法 | |
CN111210443A (zh) | 基于嵌入平衡的可变形卷积混合任务级联语义分割方法 | |
CN110909673A (zh) | 一种基于自然语言描述的行人再识别方法 | |
CN112347888B (zh) | 基于双向特征迭代融合的遥感图像场景分类方法 | |
CN110533041B (zh) | 基于回归的多尺度场景文本检测方法 | |
CN112348036A (zh) | 基于轻量化残差学习和反卷积级联的自适应目标检测方法 | |
CN112150493A (zh) | 一种基于语义指导的自然场景下屏幕区域检测方法 | |
CN111259940A (zh) | 一种基于空间注意力地图的目标检测方法 | |
CN110347857B (zh) | 基于强化学习的遥感影像的语义标注方法 | |
CN111401374A (zh) | 基于多任务的模型训练方法、字符识别方法及装置 | |
CN115222998B (zh) | 一种图像分类方法 | |
CN112507800A (zh) | 一种基于通道注意力机制和轻型卷积神经网络的行人多属性协同识别方法 | |
CN113011243A (zh) | 基于胶囊网络的面部表情分析方法 | |
CN113255602A (zh) | 基于多模态数据的动态手势识别方法 | |
CN117152416A (zh) | 一种基于detr改进模型的稀疏注意力目标检测方法 | |
CN112883931A (zh) | 基于长短期记忆网络的实时真假运动判断方法 | |
CN115116074A (zh) | 一种手写体文字识别、及模型训练方法和装置 |
Legal Events
Date | Code | Title | Description |
---|---|---|---|
PB01 | Publication | ||
PB01 | Publication | ||
SE01 | Entry into force of request for substantive examination | ||
SE01 | Entry into force of request for substantive examination | ||
GR01 | Patent grant | ||
GR01 | Patent grant |