PP-ChatOCRv3新升级：多页PDF信息抽取支持自定义提示词工程，拓展大语言模型功能边界

news2025/4/24 20:14:47

文本图像信息抽取技术在自动化办公、建筑工程、教育科研、金融风控、医疗健康等行业领域具有广泛应用场景。2024年9月，飞桨低代码开发工具PaddleX中新增文本图像智能产线PP-ChatOCRv3，充分结合PaddleOCR的文本图像版面解析能力和文心一言语言理解优势，实现了高效的文本图像信息抽取。
近期，飞桨研发团队对飞桨低代码开发工具PaddleX中文本图像智能产线PP-ChatOCRv3进行升级，一方面实现了基于标准OpenAI接口的大语言模型调用，支持不同类型大模型灵活更换，另一方面针对文本图像信息抽取，丰富了自定义提示词工程的能力，从而为文本图像信息抽取注入新动力。

在这里插入图片描述
PP-ChatOCRv3 系统流程图

PP-ChatOCRv3进阶使用传送门
https://github.com/PaddlePaddle/PaddleX/blob/release/3.0-rc/docs/practical_tutorials/document_scene_information_extraction(deepseek)_tutorial.md

本次升级具体特色功能如下：

新模型可快速适配多页 PDF 文件，高效抽取信息

由于大语言模型的推理时间相对较长，从多页 PDF 文件中抽取信息往往成本较高。
然而，PP-ChatOCRv3 集成了向量检索技术，先从 PDF 中筛选出与查询信息相关的内容，再借助大语言模型进行信息抽取，有效提升了处理效率。同时为了有效节约视觉特征和建立向量库的时间成本，PP-ChatOCRv3提供了数据缓存加载方法。在完成初次视觉特征和向量库建立后，后续的信息抽取过程即可基于缓存数据进行，有效的避免了频繁的视觉推理和建立索引库过程。
详细的实现如下：

import os
from paddlex import create_pipeline

pipeline = create_pipeline(pipeline="PP-ChatOCRv3-doc",initial_predictor=False)

output_dir = "output"
if not os.path.exists(output_dir):
    os.makedirs(output_dir)
visual_predict_res_path = os.path.join(output_dir, "contract.visual")
vector_res_path = os.path.join(output_dir, "contract.vector")
if not os.path.exists(visual_predict_res_path):
    visual_predict_res = pipeline.visual_predict(
      input="contract.pdf",
      use_doc_orientation_classify=False,
      use_doc_unwarping=False)

    visual_info_list = []
    for res in visual_predict_res:
        visual_info_list.append(res["visual_info"])

    vector_info = pipeline.build_vector(visual_info_list, flag_save_bytes_vector=True,retriever_config={
        "module_name": "retriever",
        "model_name": "embedding-v1",
        "base_url": "https://qianfan.baidubce.com/v2",
        "api_type": "qianfan",
        "api_key": "api_key" # your api_key
    })

    pipeline.save_visual_info_list(visual_info_list, visual_predict_res_path)
    vector_info = pipeline.build_vector(visual_info_list)
    pipeline.save_vector(vector_info, vector_res_path)
else:
    visual_info_list = pipeline.load_visual_info_list(visual_predict_res_path)
    vector_info = pipeline.load_vector(vector_res_path)

chat_result = pipeline.chat(
    key_list=["甲方"],
    visual_info=visual_info_list,
    vector_info=vector_info,
    chat_bot_config={
      "module_name": "chat_bot",
      "model_name": "ernie-3.5-8k",
      "base_url": "https://qianfan.baidubce.com/v2",
      "api_type": "openai",
      "api_key": "api_key" # your api_key
    },
    retriever_config={
      "module_name": "retriever",
      "model_name": "embedding-v1",
      "base_url": "https://qianfan.baidubce.com/v2",
      "api_type": "qianfan",
      "api_key": "api_key" # your api_key
    }
)
print(chat_result)

支持自定义提示词工程，拓展大语言模型的功能边界

早期进行文本图像信息抽取时，如从身份证、营业执照中提取信息，通常预先设定好关键词（如姓名、公司名称等），再查找与之相关的信息。然而，在实际应用中，关键信息往往以问题的形式呈现，例如在研报中查询《哪吒2》的票房数，要回答此类问题就需要对提示词进行修改。本次升级支持自定义提示词工程，涵盖问题描述、提取规则、结果规范以及少样本学习等方面，可根据实际问题灵活调整，从而拓展大语言模型在信息抽取方面的功能边界。
一个提示词修改示例如下：

text_task_description：对话任务的描述，例如“你现在的任务是根据提供的文本内容回答用户的问题，并给出你回答问题引用的原文片段”。
text_rules_str：用户设置的细节规则，例如“对于问题结果，使用“答案：”标注，对于引用原文片段，使用“引用原文：”标注。问题结果中的日期格式为“YYYY-MM-DD””。
text_few_shot_demo_text_content：用于少样本演示的文本内容，例如“当用户询问关于“时间”的问题时，返回某年某月某日”，通常设置为空即可。
text_few_shot_demo_key_value_list：用于少样本演示的键值对列表，例如[{“该规定是何时公布的？”: “2005年2月4日”}, {“该规定是何时施行的？”: “2005年4月1日”}]，通常设置为空即可

产出结果：

{'chat_res': {'reasoning_content': ['好的，我现在需要处理用户的问题。用户提供的OCR文本是关于《勘察设计注册工程师管理规定》的内容，问题是要找出该规定是何时公布的。首先，我要仔细阅读OCR文本，寻找相关的日期信息。\n\n在OCR文本的开头部分，我看到这样的文字：“勘察设计注册工程师管理规定 （2005年2月4日中华人民共和国建设部令第137号公布自2005年4月1日起施行）”。这里有两个日期，一个是公布的日期，另一个是施行的日期。用户的问题是询问公布的日期，所以需要确认哪个对应公布。\n\n根据中文法律法规的常见格式，通常会在标题后的括号内注明公布日期和施行日期，格式为“（日期公布机关公布，自日期起施行）”。这里的结构是“2005年2月4日中华人民共和国建设部令第137号公布自2005年4月1日起施行”，所以前半部分“2005年2月4日”应该是公布的日期，后半部分“2005年4月1日”是施行日期。\n\n接下来，我需要确认用户的问题是否确实指向公布日期。用户的问题明确问的是“该规定是何时公布的？”，所以答案应该是2005年2月4日。需要将日期格式转换为“YYYY-MM-DD”，即“2005-02-04”。\n\n然后，检查是否有其他可能的日期信息。OCR文本的其他部分主要涉及注册工程师的管理规定，没有提到其他日期。因此，可以确定答案正确。\n\n最后，按照用户的要求，用JSON格式返回结果，问题作为key，答案作为value，并引用对应的原文片段。确保日期格式正确，没有其他多余内容。\n'], '该规定是何时公布的？': {'答案': '2005-02-04', '引用原文': '勘察设计注册工程师管理规定 （2005年2月4日中华人民共和国建设部令第137号公布自2005年4月1日起施行）'}}}

精彩课程预告

为了帮助您全面掌握PP-ChatOCRv3的最新技术进展，百度研发团队精心打造视频精讲课程，深入解读PP-ChatOCRv3的最新技术方案，并分享其在实际产业中的应用技巧。同时，我们还特别设置了产业场景实战营，配套详细教程文档，将手把手指导您针对具体问题灵活优化自定义提示词工程能力。机会难得，立即点击链接报名：https://www.wjx.top/vm/wFZcC0n.aspx?udsid=749768

在这里插入图片描述