arXiv:2412.02886cs.CV2024-12中稿 · IEEE/CVF Winter Co…被引 3

用视觉语言模型的不确定度,精准定位扫描文档中的信息片段。

Patchfinder: Leveraging Visual Language Models for Accurate Information Retrieval using Model Uncertainty

  • 基于模型输出的最大softmax概率,构建置信度评分以自适应选择图像块大小。
  • 在190份噪声扫描文档上,准确率达94%,比ChatGPT-4o高18.5个百分点。
  • 适合处理高密度、低质量扫描文档的信息提取任务,尤其对资源有限场景友好。

几十年来,企业和政府一直依赖扫描文档记录海量信息,但信息提取过程因文档量大且结构复杂而缓慢繁琐。视觉语言模型(VLMs)为高效准确地从这些文档中提取信息提供了新可能。当前自动化流程通常分两步:先用OCR提取文本,再用大语言模型处理。然而,面对噪声扫描文档时,该方法面临挑战,常需计算开销大的语言模型应对高信息密度。本文提出PatchFinder算法,基于VLMs改进信息提取。首先,设计一种名为Patch Confidence的置信度评分,依据VLM输出的最大Softmax概率衡量模型预测置信度。利用该指标,PatchFinder自动确定合适图像块大小,将输入文档划分为重叠块,并生成基于置信度的靶向信息预测。实验结果表明,采用Phi-3v(42亿参数)的PatchFinder在190份噪声扫描文档数据集上达到94%准确率,相较ChatGPT-4o提升18.5个百分点。

原文摘要 · Abstract (English)

For decades, corporations and governments have relied on scanned documents to record vast amounts of information. However, extracting this information is a slow and tedious process due to the sheer volume and complexity of these records. The rise of Vision Language Models (VLMs) presents a way to efficiently and accurately extract the information out of these documents. The current automated workflow often requires a two-step approach involving the extraction of information using optical character recognition software and subsequent usage of large language models for processing this information. Unfortunately, these methods encounter significant challenges when dealing with noisy scanned documents, often requiring computationally expensive language models to handle high information density effectively. In this study, we propose PatchFinder, an algorithm that builds upon VLMs to improve information extraction. First, we devise a confidence-based score, called Patch Confidence, based on the Maximum Softmax Probability of the VLMs' output to measure the model's confidence in its predictions. Using this metric, PatchFinder determines a suitable patch size, partitions the input document into overlapping patches, and generates confidence-based predictions for the target information. Our experimental results show that PatchFinder, leveraging Phi-3v, a 4.2-billion-parameter VLM, achieves an accuracy of 94% on our dataset of 190 noisy scanned documents, outperforming ChatGPT-4o by 18.5 percentage points.

信息提取视觉语言模型文档理解置信度评估

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。