Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

OCR-free Document Understanding Transformer

About

Understanding document images (e.g., invoices) is a core but challenging task since it requires complex functions such as reading text and a holistic understanding of the document. Current Visual Document Understanding (VDU) methods outsource the task of reading text to off-the-shelf Optical Character Recognition (OCR) engines and focus on the understanding task with the OCR outputs. Although such OCR-based approaches have shown promising performance, they suffer from 1) high computational costs for using OCR; 2) inflexibility of OCR models on languages or types of document; 3) OCR error propagation to the subsequent process. To address these issues, in this paper, we introduce a novel OCR-free VDU model named Donut, which stands for Document understanding transformer. As the first step in OCR-free VDU research, we propose a simple architecture (i.e., Transformer) with a pre-training objective (i.e., cross-entropy loss). Donut is conceptually simple yet effective. Through extensive experiments and analyses, we show a simple OCR-free VDU model, Donut, achieves state-of-the-art performances on various VDU tasks in terms of both speed and accuracy. In addition, we offer a synthetic data generator that helps the model pre-training to be flexible in various languages and domains. The code, trained model and synthetic data are available at https://github.com/clovaai/donut.

Geewook Kim, Teakgyu Hong, Moonbin Yim, Jeongyeon Nam, Jinyoung Park, Jinyeong Yim, Wonseok Hwang, Sangdoo Yun, Dongyoon Han, Seunghyun Park• 2021

Related benchmarks

TaskDatasetResultRank
Visual Question AnsweringTextVQA
Accuracy43.5
1117
Document ClassificationRVL-CDIP (test)
Accuracy95.3
306
Visual Question AnsweringChartQA
Accuracy41.8
239
Chart Question AnsweringChartQA
Accuracy41.8
229
Document Visual Question AnsweringDocVQA (test)
ANLS67.5
192
Visual Question AnsweringAI2D
Accuracy30.8
174
Document Visual Question AnsweringDocVQA
ANLS67.5
164
Information ExtractionCORD (test)
F1 Score91.6
133
Chart Question AnsweringChartQA (test)--
129
Visual Question AnsweringTextVQA (test)
Accuracy43.5
124
Showing 10 of 74 rows
...

Other info

Code

Follow for update