让视觉模型学会识别图中箭头,提升对技术图表的理解能力
Can Visual Encoder Learn to See Arrows?
- 用无文本和位置偏见的图表数据训练视觉编码器,避免模型依赖错误线索
- 微调后模型在图文检索、描述生成等任务上均优于CLIP和GPT-4o
- 为提升视觉语言模型理解科技图表提供了可复制的训练路径
图表是通过边(线或箭头)表示关系的视觉表达,广泛用于工业与科学交流。尽管识别图表对视觉语言模型(VLMs)理解领域知识至关重要,但近期研究发现许多VLMs无法准确识别图像中的边。我们假设这些失败源于对文本和位置偏见的过度依赖,阻碍了模型学习显式的边缘特征。基于此,我们通过在人工生成的无文本和位置偏见的图表-标题数据集上进行对比学习,训练一个图像编码器,并在三个任务(探针测试、图像检索、图文生成)上评估其图表相关特征。结果表明,微调后的模型在所有任务中均优于预训练的CLIP,且在图文生成任务中超越零样本的GPT-4o和LLaVA-Mistral。这些发现证实,消除文本与位置偏见有助于实现更准确的边缘识别,为提升图表理解能力提供了一条可行路径。
原文摘要 · Abstract (English)
The diagram is a visual representation of a relationship illustrated with edges (lines or arrows), which is widely used in industrial and scientific communication. Although recognizing diagrams is essential for vision language models (VLMs) to comprehend domain-specific knowledge, recent studies reveal that many VLMs fail to identify edges in images. We hypothesize that these failures stem from an over-reliance on textual and positional biases, preventing VLMs from learning explicit edge features. Based on this idea, we empirically investigate whether the image encoder in VLMs can learn edge representation through training on a diagram dataset in which edges are biased neither by textual nor positional information. To this end, we conduct contrastive learning on an artificially generated diagram--caption dataset to train an image encoder and evaluate its diagram-related features on three tasks: probing, image retrieval, and captioning. Our results show that the finetuned model outperforms pretrained CLIP in all tasks and surpasses zero-shot GPT-4o and LLaVA-Mistral in the captioning task. These findings confirm that eliminating textual and positional biases fosters accurate edge recognition in VLMs, offering a promising path for advancing diagram understanding.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。