Share your thoughts, 1 month free Claude Pro on usSee more
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
1285
Visual Question AnsweringChartQA
Accuracy41.8
371
Chart Question AnsweringChartQA
Accuracy41.8
356
Document ClassificationRVL-CDIP (test)
Accuracy95.3
306
Document Visual Question AnsweringDocVQA
ANLS67.5
263
Visual Question AnsweringAI2D
Accuracy30.8
249
Document Visual Question AnsweringDocVQA (test)
ANLS67.5
213
Chart Question AnsweringChartQA (test)--
176
Table Fact VerificationTabFact (test)
Accuracy54.6
136
Information ExtractionCORD (test)
F1 Score91.6
133
Showing 10 of 75 rows
...

Other info

Code

Follow for update