arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2204.08387cs.CLcs.CV
收藏

LayoutLMv3:使用统一文本和图像掩码对文档AI进行预培训

LayoutLMv3: Pre-training for Document AI with Unified Text and Image Masking

Sun Yat-sen University(中山大学);Microsoft Research Asia(微软亚洲研究院)

Yupan Huang, Tengchao Lv, Lei Cui, Yutong Lu, Furu Wei

摘要:

自我监督的预训练技术在文档AI中取得了显着进展。大多数多模态预训练模型使用掩蔽语言建模目标来学习文本模态的双向表示,但它们在图像模态的预训练目标上有所不同。这种差异增加了多模态表示学习的难度。在本文中,我们提出了\textbf{LayoutLMv3}来预训练多模态Transformers,用于具有统一文本和图像掩蔽的文档AI。此外,LayoutLMv3使用单词补丁对齐目标进行预训练,以通过预测文本单词的相应图像补丁是否被掩蔽来学习跨模态对齐。简单的统一架构和训练目标使LayoutLMv3成为以文本为中心和以图像为中心的文档AI任务的通用预训练模型。实验结果表明,LayoutLMv3不仅在以文本为中心的任务(包括表单理解、收据理解和文档视觉问答)中实现了最先进的性能,而且在以图像为中心的任务(如文档图像分类和文档布局分析)中也实现了最先进的性能。代码和模型可在\url{https://aka.ms/layoutlmv3}上公开获得。

英文摘要:

Self-supervised pre-training techniques have achieved remarkable progress in Document AI. Most multimodal pre-trained models use a masked language modeling objective to learn bidirectional representations on the text modality, but they differ in pre-training objectives for the image modality. This discrepancy adds difficulty to multimodal representation learning. In this paper, we propose \textbf{LayoutLMv3} to pre-train multimodal Transformers for Document AI with unified text and image masking. Additionally, LayoutLMv3 is pre-trained with a word-patch alignment objective to learn cross-modal alignment by predicting whether the corresponding image patch of a text word is masked. The simple unified architecture and training objectives make LayoutLMv3 a general-purpose pre-trained model for both text-centric and image-centric Document AI tasks. Experimental results show that LayoutLMv3 achieves state-of-the-art performance not only in text-centric tasks, including form understanding, receipt understanding, and document visual question answering, but also in image-centric tasks such as document image classification and document layout analysis. The code and models are publicly available at \url{https://aka.ms/layoutlmv3}.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑