arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2006.00753cs.CV
收藏

TextVQA的结构化多模态注意

Structured Multimodal Attentions for TextVQA

Northwestern Polytechnical University(西北工业大学);University of Adelaide(阿德莱德大学)

Chenyu Gao, Qi Zhu, Peng Wang, Hui Li, Yuliang Liu, Anton van den Hengel, Qi Wu

摘要:

本文提出了一种端到端结构化多模式注意力(SMA)神经网络,主要解决上述前两个问题。SMA首先使用结构图表示对图像中出现的对象-对象、对象-文本和文本-文本关系进行编码,然后设计多模态图注意网络对其进行推理,最后由全局-局部注意应答模块处理上述模块的输出,按照M4 C迭代地将OCR和通用词汇的标记拼接在一起产生应答。我们提出的模型在TextVQA数据集和ST-VQA数据集的两个任务上优于所有模型中的SoTA模型,除了基于预训练的TAP。凭借强大的推理能力,它还赢得了2020年TextVQA挑战赛的第一名。我们广泛地测试了不同的OCR方法在几个推理模型和研究的影响,逐渐提高OCR性能的TextVQA基准。随着更好的OCR结果,不同的模型共享VQA准确性的显着改善,但我们的模型受益最大的是强大的文本视觉推理能力。为了给我们的方法一个上限,并为进一步的工作提供一个公平的测试基础,我们还为TextVQA数据集提供了人工注释的地面实况OCR注释,这在原始版本中没有给出。TextVQA数据集的代码和地面实况OCR注释可在https://github.com/ChenyuGAO-CS/SMA上获得

英文摘要:

In this paper, we propose an end-to-end structured multimodal attention (SMA) neural network to mainly solve the first two issues above. SMA first uses a structural graph representation to encode the object-object, object-text and text-text relationships appearing in the image, and then designs a multimodal graph attention network to reason over it. Finally, the outputs from the above modules are processed by a global-local attentional answering module to produce an answer splicing together tokens from both OCR and general vocabulary iteratively by following M4C. Our proposed model outperforms the SoTA models on TextVQA dataset and two tasks of ST-VQA dataset among all models except pre-training based TAP. Demonstrating strong reasoning ability, it also won first place in TextVQA Challenge 2020. We extensively test different OCR methods on several reasoning models and investigate the impact of gradually increased OCR performance on TextVQA benchmark. With better OCR results, different models share dramatic improvement over the VQA accuracy, but our model benefits most blessed by strong textual-visual reasoning ability. To grant our method an upper bound and make a fair testing base available for further works, we also provide human-annotated ground-truth OCR annotations for the TextVQA dataset, which were not given in the original release. The code and ground-truth OCR annotations for the TextVQA dataset are available at https://github.com/ChenyuGAO-CS/SMA

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑