arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.19014cs.CV
收藏

利用大型视觉-语言模型中的对比学习增强视觉文档理解

Enhancing Visual Document Understanding with Contrastive Learning in Large Visual-Language Models

Tencent YouTu Lab(腾讯优图实验室)

Xin Li, Yunfei Wu, Xinghua Jiang, Zhihao Guo, Mingming Gong, Haoyu Cao, Yinsong Liu, Deqiang Jiang, Xing Sun

AI总结:

针对LVLM在文本密集文档中细粒度特征不足的问题,本文提出即插即用的DoCo对比学习框架,通过文档对象特征对齐增强视觉表示,并在多个VDU基准上验证其性能提升。

摘要:

近年来,大型视觉-语言模型(Large Visual-Language Models,LVLMs)的出现受到了多个领域的日益关注,尤其是在视觉文档理解(visual document understanding,VDU)领域。与传统的视觉-语言任务不同,VDU专门关注包含丰富文档元素的文本密集场景。然而,细粒度特征的重要性在LVLM社区中仍基本未被探索,导致其在文本密集场景中的性能欠佳。本文将该问题简称为细粒度特征坍缩问题。为填补这一空白,我们提出了一个对比学习框架,称为Document Object COntrastive learning(DoCo),专门面向VDU下游任务设计。DoCo利用一个辅助多模态编码器获取文档对象的特征,并将这些特征与LVLM视觉编码器生成的视觉特征进行对齐,从而增强文本密集场景中的视觉表示。这表明,视觉整体表示与文档对象的多模态细粒度特征之间的对比学习,能够帮助视觉编码器获得更有效的视觉线索,进而增强LVLM对文本密集文档的理解。我们还证明,所提出的DoCo是一种即插即用的预训练方法,可用于多种LVLM的预训练,并且不会在推理过程中引入任何额外的计算复杂度。在多个VDU基准上的大量实验结果表明,配备本文提出的DoCo后,LVLM能够取得更优性能,并缩小VDU与通用视觉-语言任务之间的差距。

英文摘要:

Recently, the advent of Large Visual-Language Models (LVLMs) has received increasing attention across various domains, particularly in the field of visual document understanding (VDU). Different from conventional vision-language tasks, VDU is specifically concerned with text-rich scenarios containing abundant document elements. Nevertheless, the importance of fine-grained features remains largely unexplored within the community of LVLMs, leading to suboptimal performance in text-rich scenarios. In this paper, we abbreviate it as the fine-grained feature collapse issue. With the aim of filling this gap, we propose a contrastive learning framework, termed Document Object COntrastive learning (DoCo), specifically tailored for the downstream tasks of VDU. DoCo leverages an auxiliary multimodal encoder to obtain the features of document objects and align them to the visual features generated by the vision encoder of LVLM, which enhances visual representation in text-rich scenarios. It can represent that the contrastive learning between the visual holistic representations and the multimodal fine-grained features of document objects can assist the vision encoder in acquiring more effective visual cues, thereby enhancing the comprehension of text-rich documents in LVLMs. We also demonstrate that the proposed DoCo serves as a plug-and-play pre-training method, which can be employed in the pre-training of various LVLMs without inducing any increase in computational complexity during the inference process. Extensive experimental results on multiple benchmarks of VDU reveal that LVLMs equipped with our proposed DoCo can achieve superior performance and mitigate the gap between VDU and generic vision-language tasks.

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

↑