BuDDIE: A Business Document Dataset for Multi-task Information Extraction

Ran Zmigrod,Dongsheng Wang,Mathieu Sibue,Yulong Pei,Petr Babkin,Ivan Brugere,Xiaomo Liu,Nacho Navarro,Antony Papadimitriou,William Watson,Zhiqiang Ma,Armineh Nourbakhsh,Sameena Shah

2024-04-05

Abstract:The field of visually rich document understanding (VRDU) aims to solve a multitude of well-researched NLP tasks in a multi-modal domain. Several datasets exist for research on specific tasks of VRDU such as document classification (DC), key entity extraction (KEE), entity linking, visual question answering (VQA), inter alia. These datasets cover documents like invoices and receipts with sparse annotations such that they support one or two co-related tasks (e.g., entity extraction and entity linking). Unfortunately, only focusing on a single specific of documents or task is not representative of how documents often need to be processed in the wild - where variety in style and requirements is expected. In this paper, we introduce BuDDIE (Business Document Dataset for Information Extraction), the first multi-task dataset of 1,665 real-world business documents that contains rich and dense annotations for DC, KEE, and VQA. Our dataset consists of publicly available business entity documents from US state government websites. The documents are structured and vary in their style and layout across states and types (e.g., forms, certificates, reports, etc.). We provide data variety and quality metrics for BuDDIE as well as a series of baselines for each task. Our baselines cover traditional textual, multi-modal, and large language model approaches to VRDU.

Computation and Language

What problem does this paper attempt to address?

The paper aims to address the issue of multi-task information extraction in the field of Visually Rich Document Understanding (VRDU). Specifically, the paper introduces a new dataset called BuDDIE (Business Document Dataset for Information Extraction), which is a multi-task dataset containing 1,665 real-world business documents to support three tasks: Document Classification (DC), Key Entity Extraction (KEE), and Visual Question Answering (VQA).

The characteristics of the BuDDIE dataset are as follows:
1. **Diversity and Complexity**: The documents in the dataset come from government websites across the United States and cover a variety of document types (such as forms, certificates, reports, etc.), which exhibit significant differences in style and layout.
2. **Multi-tasking**: This is the first dataset that supports DC, KEE, and VQA tasks simultaneously, better reflecting the document processing needs in actual application scenarios.
3. **Rich Annotations**: The dataset includes a large number of annotations, including DC category labels, KEE entity labels, and VQA questions and answers, making BuDDIE a densely annotated dataset.
4. **Potential for Expansion**: Due to its rich annotations and multi-task nature, BuDDIE can be further expanded in the future to support more downstream tasks, such as multi-turn question answering or multimodal instruction fine-tuning.

By introducing the BuDDIE dataset, researchers can assess the capability of models in handling diverse documents and promote the research development in the VRDU field. In addition, the paper also provides baseline model results for different tasks on BuDDIE, including traditional text models, multimodal models, and large language models, offering valuable reference points for subsequent research.