DocFormer: End-to-End Transformer for Document Understanding 论文

20212021 IEEE/CVF International Conference on Computer Vision (ICCV)引用 283
Handwritten Text Recognition TechniquesMultimodal Machine Learning ApplicationsDomain Adaptation and Few-Shot Learning

详细信息

发表期刊/会议
2021 IEEE/CVF International Conference on Computer Vision (ICCV)
发表日期
2021-10-01
发表年份
2021

关键词

Handwritten Text Recognition TechniquesMultimodal Machine Learning ApplicationsDomain Adaptation and Few-Shot Learning

摘要

We present DocFormer - a multi-modal transformer based architecture for the task of Visual Document Understanding (VDU). VDU is a challenging problem which aims to understand documents in their varied formats (forms, receipts etc.) and layouts. In addition, DocFormer is pre-trained in an unsupervised fashion using carefully designed tasks which encourage multi-modal interaction. DocFormer uses text, vision and spatial features and combines them using a novel multi-modal self-attention layer. DocFormer also shares learned spatial embeddings across modalities which makes it easy for the model to correlate text to visual tokens and vice versa. DocFormer is evaluated on 4 different datasets each with strong baselines. DocFormer achieves state-of-the-art results on all of them, sometimes beating models 4x its size (in no. of parameters).