使用Qwen2-VL-2B模型抽取图1中的信息时,模型原生回答和使用GuidanceOCR后的回答见下表
项目链接:
https://github.com/hzauzxb/guidance-ocr
一、项目背景
视觉信息抽取任务是给定一张单证图片,并从图片中抽取对应的关键字段,如从图1所示的医疗发票中抽取金额合计,住院时间和医保类型。传统的信息抽取模型先利用OCR算法获取图中的文字位置和文字内容,再利用规则引擎或模板匹配获取关键字段。VLM则可以将图片先输入给模型,再用提问题的方式要求模型给出需要抽取的字段。在开放场景的信息抽取中,由于单证板式太多,规则引擎难以维系,因此常常使用基于VLM的方案。
在实际使用时,为了避免显存OOM,VLM常常会将输入的图像大小限制在一定范围。因此VLM在视觉信息抽取任务中,可能会由于文字太小,模糊不清而给出错误的答案。OCR模型分为检测和识别两部分,检测模型负责给出图中的文字位置,而识别模型会根据检测模型给出的文字位置,将图片中的文字一一裁剪下来,并缩放到指定大小,再去识别文字内容。因此传统OCR模型的识字能力通常是高于多模态大模型的识字能力的。
可以看到,同样是"票据代码",输入多模态大模型的分辨率是 16 * 48,而输入传统OCR识别模型的分辨率则为 48 * 320。该特性导致在视觉信息抽取任务中,若使用VLM抽取图中较小的字段,会出现错字的问题,而传统OCR模型则基本都能识别出正确的文字。
为了解决该问题,本文提出了名为Guidance OCR的算法,该算法在不额外训练模型的情况下,先利用OCR算法获取图片中的文字内容,再利用OCR识别出的文字对VLM的生成过程进行约束,从而一定程度上减少VLM做视觉信息抽取任务时出现错字的情况。
二、算法流程
本节将先介绍GuidanceOCR的任务设定,再介绍单个字段信息抽取的算法流程(一次调用VLM只抽取图片中的一个关键字段),最后在介绍多个字段信息抽取的算法流程(一次调用VLM会抽取图片中的多个关键字段,并以Json格式返回结果)。
GuidanceOCR的任务设定
如图4左侧所示,Guidance OCR会将OCR识别出的文字,需要抽取的字段和对应的图片作为输入,并利用多模态大模型(VLM)获取信息抽取结果。Guidance OCR为测试时增强算法(Test Time Argument, TTA), 其并不要求对VLM进行额外的训练。
如图4右侧所示,视觉信息抽取任务有三种常见的情况即:
• 1.需抽取的字段为单个文本框,即为图4左侧OCR Text列表中的某个字符串
• 2.需抽取的字段横跨多个文本框,即为图4左侧OCR Text列表中的多个字符串,按特定顺序组合而成
• 3.需抽取的字段为文本框中的某一部分,即为图4左侧OCR Text列表中的某个字符串的子串
单个字段的信息抽取
在单个字段信息抽取的场景下,多模态大模型的Prompt和Response的要求如下
图5给出了Guidance OCR的算法流程:在VLM每次生成Token的时候,先将候选的token按logit从大到小进行排序(图5 左侧);再使用基于OCR的token判别器逐个判断候选Token是否合法(图五 中间);找到的第一个合法Token即为VLM该次生成的Token(图5 右侧);在进行下一次生成候选token前,Guidance OCR还会根据此次生成的Token更新基于OCR的Token判别器(图5 虚线部分)。
图6给出了基于OCR的Token判别器中的数据结构。在该字典树中,每个被OCR识别出的字符串的开头节点和结尾节点都特殊标识了出来。在介绍如何根据该字典树判断Token是否合法前,我们需要先定义字典树上连续的一条边
字典树上连续的一条边:
• 字典树中从上到下连续的字,如:湖南省,小区,南省长沙市;
• 若灰色节点后接橙色节点,则仍然连续,如:开福区幸福小区、湖南省幸福小区、襄樊湖南省。
被token判别器判定为合法的候选token必须满足:
• 若为大模型首次生成的候选token,对应的文字必须是字典树上连续的一条边或生成结束标识符;
• 若不是大模型首次生成的候选token,大模型之前生成的文字 + 候选token对应的文字,是字典树上连续的一条边或生成结束标识符。
额外的top_k参数:为了节省时间并提高准确率,Guidance OCR会设置top_k参数,在使用图5所示的算法筛选token时,仅筛选logits为前top_k大的token。若前top_k token中无合法token,则直接输出生成结束标识符,结束生成过程。在实际使用中,通常将top_k设定在50到100之间的数值。
小结:在单个字段的信息抽取的场景下,GuidanceOCR会将OCR识别出的文字内容组合成一个字典树并保证大模型的输出内容为字典树上连续的一条边。在这里我们扩展了连续性的定义,这是为了兼容信息抽取时,需抽取的字段横跨多个文本框的情况。
多个字段的信息抽取
在多个字段信息抽取的场景下,多模态大模型的Prompt和Response的要求如下
与上节的方法类似,多个字段的信息抽取仍然是使用图5所示的Token判别器实现的。与上节不同的是,token判别器有json,key和value三个状态
当生成时,token判别器为json状态,随着生成的进行,token判别器会基于启发式的规则,进行状态转移,从而保证能生成json格式的结果。与2.2节一致,若输出生成结束标识符或top_k内没找到符合条件的候选token时,生成过程结束。
三、模型效果
在信息抽取数据集上,使用Qwen2VL-2B进行测试;输入图像进行等比例缩放,并将最长边固定为1024;使用Guidance OCR的多个字段信息抽取模式;使用transformers中AutoModel.from_pretrained()部署模型,进行infer