
摘要
近年来,预训练技术在多种自然语言处理(NLP)任务中得到了成功的验证。尽管预训练模型在NLP应用中被广泛使用,但它们几乎完全集中在文本层面的操作上,而忽视了对文档图像理解至关重要的布局和样式信息。本文提出了一种LayoutLM模型,用于联合建模扫描文档图像中的文本和布局信息之间的交互作用,这有助于许多实际的文档图像理解任务,例如从扫描文档中提取信息。此外,我们还利用图像特征将单词的视觉信息融入到LayoutLM中。据我们所知,这是首次在一个框架内同时学习文本和布局信息以进行文档级别的预训练。该模型在多个下游任务中取得了新的最先进结果,包括表单理解(从70.72提高到79.27)、收据理解(从94.02提高到95.24)和文档图像分类(从93.07提高到94.42)。LayoutLM的代码和预训练模型已公开发布于\url{https://aka.ms/layoutlm}。
代码仓库
thibaultdouzon/long-range-document-transformer
pytorch
GitHub 中提及
facebookresearch/data2vec_vision
pytorch
GitHub 中提及
cydal/LayoutML_pytorch
pytorch
GitHub 中提及
omarsou/layoutlm_CORD
GitHub 中提及
lulia0228/Document_IE
GitHub 中提及
impira/docquery
pytorch
GitHub 中提及
huggingface/transformers
pytorch
GitHub 中提及
nunenuh/layoutlm.pytorch
pytorch
kenAlparslan/Texttract
tf
GitHub 中提及
MS-P3/code3/tree/main/layoutlm
mindspore
doc-analysis/DocBank
GitHub 中提及
PaddlePaddle/PaddleOCR
paddle
microsoft/unilm
官方
pytorch
GitHub 中提及
pwc-1/Paper-9/tree/main/layoutlm
mindspore
BordiaS/layoutlm
pytorch
GitHub 中提及
microsoft/unilm/tree/master/layoutlm
pytorch
GitHub 中提及
基准测试
| 基准 | 方法 | 指标 |
|---|---|---|
| document-image-classification-on-rvl-cdip | Pre-trained LayoutLM | Accuracy: 94.42% Parameters: 160M |
| relation-extraction-on-funsd | LayoutLM | F1: 42.83 |