arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2403.04473cs.CVcs.AI

TextMonkey:一种用于文档理解的无OCR大型多模态模型

TextMonkey: An OCR-Free Large Multimodal Model for Understanding Document

School of Artificial Intelligence and Automation, Huazhong University of Science and Technology(华中科技大学人工智能与自动化学院);Kingsoft(金山软件)

Yuliang Liu, Biao Yang, Qiang Liu, Zhang Li, Zhiyin Ma, Shuo Zhang, Xiang Bai

AI总结:

TextMonkey是一种无OCR的大型多模态模型,通过移位窗口注意力、标记筛选和文本定位增强,在12个基准上显著提升文档理解性能,并在OCRBench上创下561分的新纪录。

摘要:

我们提出了TextMonkey,一种专为文本中心任务设计的大型多模态模型(LMM)。我们的方法在多个维度上引入了改进:通过采用零初始化的移位窗口注意力(Shifted Window Attention),我们在更高输入分辨率下实现了跨窗口连接,并稳定了早期训练;我们假设图像可能包含冗余标记,通过使用相似性过滤出重要标记,我们不仅能够精简标记长度,还能提升模型性能。此外,通过扩展模型能力以涵盖文本检测与定位(text spotting and grounding),并在响应中融入位置信息,我们增强了可解释性。该模型还通过微调学会了执行截图任务。在12个基准上的评估显示了显著改进:场景文本中心任务(包括STVQA、TextVQA和OCRVQA)提升5.2%,文档导向任务(如DocVQA、InfoVQA、ChartVQA、DeepForm、Kleister Charity和WikiTableQuestions)提升6.9%,关键信息抽取任务(包括FUNSD、SROIE和POIE)提升2.8%。在场景文本检测方面,它以10.9%的提升表现优异,并在OCRBench(一个包含29项OCR相关评估的综合基准)上以561分的成绩树立了新标准,超越了先前开源的用于文档理解的大型多模态模型。代码将在https://github.com/Yuliang-Liu/Monkey发布。

英文摘要:

We present TextMonkey, a large multimodal model (LMM) tailored for text-centric tasks. Our approach introduces enhancement across several dimensions: By adopting Shifted Window Attention with zero-initialization, we achieve cross-window connectivity at higher input resolutions and stabilize early training; We hypothesize that images may contain redundant tokens, and by using similarity to filter out significant tokens, we can not only streamline the token length but also enhance the model's performance. Moreover, by expanding our model's capabilities to encompass text spotting and grounding, and incorporating positional information into responses, we enhance interpretability. It also learns to perform screenshot tasks through finetuning. Evaluation on 12 benchmarks shows notable improvements: 5.2% in Scene Text-Centric tasks (including STVQA, TextVQA, and OCRVQA), 6.9% in Document-Oriented tasks (such as DocVQA, InfoVQA, ChartVQA, DeepForm, Kleister Charity, and WikiTableQuestions), and 2.8% in Key Information Extraction tasks (comprising FUNSD, SROIE, and POIE). It outperforms in scene text spotting with a 10.9\% increase and sets a new standard on OCRBench, a comprehensive benchmark consisting of 29 OCR-related assessments, with a score of 561, surpassing previous open-sourced large multimodal models for document understanding. Code will be released at https://github.com/Yuliang-Liu/Monkey.

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

↑