4 个月前

CoMER:基于Transformer的手写数学表达式识别覆盖率建模

CoMER:基于Transformer的手写数学表达式识别覆盖率建模

摘要

基于Transformer的编码器-解码器架构在识别手写数学表达式方面最近取得了显著进展。然而,Transformer模型仍然存在覆盖不足的问题,导致其表达式识别率(ExpRate)低于递归神经网络(RNN)模型。覆盖信息记录了过去步骤中的对齐信息,在RNN模型中已被证明是有效的。本文提出了一种名为CoMER的模型,该模型在Transformer解码器中引入了覆盖信息。具体而言,我们设计了一种新颖的注意力精炼模块(Attention Refinement Module, ARM),利用过去的对齐信息来优化注意力权重,同时不影响其并行性。此外,我们将覆盖信息进一步扩展,提出了自覆盖和交叉覆盖机制,这些机制利用了当前层和前一层的过去对齐信息。实验结果表明,CoMER相比当前最先进的模型提高了0.61%、2.09%和1.59%的ExpRate,并在CROHME 2014、2016和2019测试集上分别达到了59.33%、59.81%和62.97%的识别率。

代码仓库

qingzhenduyu/ical
pytorch
GitHub 中提及
Green-Wood/CoMER
官方
pytorch
GitHub 中提及

基准测试

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
CoMER:基于Transformer的手写数学表达式识别覆盖率建模 | 论文 | HyperAI超神经