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

用于TextVQA的空间感知多模态Transformers

Spatially Aware Multimodal Transformers for TextVQA

Georgia Institute of Technology(佐治亚理工学院);Facebook AI Research (FAIR)(Facebook人工智能研究院(FAIR));Google(谷歌);University of Illinois, Urbana-Champaign(伊利诺伊大学厄巴纳-香槟分校);Allen Institute for AI(艾伦人工智能研究所)

Yash Kant, Dhruv Batra, Peter Anderson, Alex Schwing, Devi Parikh, Jiasen Lu, Harsh Agrawal

摘要:

文本提示对于购买杂货和使用公共交通等日常任务至关重要。为了开发这种辅助技术,我们研究了TextVQA任务,即,推理图像中的文本来回答问题。现有的方法在空间关系的使用方面受到限制,并且依赖于完全连接的transformer式架构来隐式地学习场景的空间结构。相比之下,我们提出了一种新的空间感知的自我注意力层,使得每个视觉实体只看由空间图定义的相邻实体。此外,我们的多头自我关注层中的每个头都专注于不同的关系子集。我们的方法有两个优点:(1)每个头考虑本地上下文,而不是分散注意力在所有视觉实体;(2)我们避免学习冗余的功能。我们表明,我们的模型在改进的基线上将当前最先进的TextVQA方法的绝对准确率提高了2.2%,在涉及空间推理的问题上提高了4.62%,并且可以使用OCR令牌正确回答。同样,在ST-VQA上,我们将绝对精度提高了4.2%。我们进一步表明,空间意识的自我注意力提高视觉接地。

英文摘要:

Textual cues are essential for everyday tasks like buying groceries and using public transport. To develop this assistive technology, we study the TextVQA task, i.e., reasoning about text in images to answer a question. Existing approaches are limited in their use of spatial relations and rely on fully-connected transformer-like architectures to implicitly learn the spatial structure of a scene. In contrast, we propose a novel spatially aware self-attention layer such that each visual entity only looks at neighboring entities defined by a spatial graph. Further, each head in our multi-head self-attention layer focuses on a different subset of relations. Our approach has two advantages: (1) each head considers local context instead of dispersing the attention amongst all visual entities; (2) we avoid learning redundant features. We show that our model improves the absolute accuracy of current state-of-the-art methods on TextVQA by 2.2% overall over an improved baseline, and 4.62% on questions that involve spatial reasoning and can be answered correctly using OCR tokens. Similarly on ST-VQA, we improve the absolute accuracy by 4.2%. We further show that spatially aware self-attention improves visual grounding.

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

↑