ColPali 简化并增强了从复杂、视觉丰富的文档中检索信息的能力,从而改变了检索增强型生成。
译自 PDFs Get an Easier Entry to GenAI via New RAG Architecture,作者 Bonnie Chase。
虽然一张图片胜过千言万语,但为 检索增强生成 (RAG) 工作流程准备视觉丰富的多模态文档(如 PDF)既耗时又容易出错。在医疗保健或金融服务等对准确性至关重要的行业,放射学报告或财务报表等文档通常包含提供宝贵上下文信息的图像或图表。虽然这些视觉丰富的元素通常被排除在 RAG 工作流程之外,但一种用于从视觉增强文档中检索信息的新方法将简化多模态文档准备,并改变 RAG 和生成式 AI (GenAI) 的潜力。
大多数检索系统主要关注基于文本的表示,而忽略了文档的视觉元素,例如图像、表格和布局。这种限制会降低检索效率,尤其是在这些视觉特征是理解文档内容的关键的情况下。
针对 PDF(或其他复杂格式)的典型现实世界 RAG 管道涉及以下步骤:
- 提取文本和元数据
- 光学字符识别 (OCR)
- 布局分析:提取表格、图表、饼图等。
在完成获取文本表示的处理步骤后,文本可以作为 检索系统的输入。
这些处理步骤可能很耗时,并会影响检索质量,但 Contextualized Late Interaction over PaliGemma (ColPali) 是一种新的检索模型架构,专注于文档密集型环境中的 RAG,克服了这些挑战。这种新的检索方法直接将整个渲染后的文档(包括其视觉元素)嵌入到适合检索的向量表示中。
ColPali 如何改进文档检索
通过将文档视为视觉实体而不是文本,ColPali 为更准确、更具上下文感知的文档检索开辟了新的可能性,尤其适用于视觉丰富的內容。ColPali 通过以下方式代表了文档检索的进步:
- 消除了对复杂预处理步骤的需求
- 保留文档的视觉上下文
- 能够更全面地理解文档
- 简化 RAG 管道
通过绕过传统的文本提取和 OCR 流程,ColPali 不仅简化了检索过程,而且有可能提高 RAG 系统中检索信息的质量和相关性。
ColPali 的架构建立在两个关键概念之上:来自 视觉语言模型 (VLMs) 的上下文视觉嵌入和后期交互机制。
视觉嵌入
ColPali 使用 PaliGemma 模型,这些模型是 Google 为图像文本压缩等通用任务创建的轻量级 VLM。与传统的纯文本模型相比,VLM 具有独特的优势,因为它集成了 视觉和文本数据,使它们能够处理需要全面理解视觉上下文才能完成的复杂任务,例如解释图像、为视觉输入生成描述性文本以及根据视觉线索回答问题。
使用 PaliGemma,ColPali 可以直接从文档图像创建高质量的上下文嵌入,而无需进行文本提取、OCR 或布局分析等复杂步骤。这种简化的方法使索引更快、更容易,从而提高了文档检索的效率。检索到文档后,RAG 系统中的生成阶段可以专注于使用文本和视觉信息处理和总结最相关的文档。
该模型能够使用视觉元素和文本,从而能够更全面地理解文档内容。这种方法使 ColPali 能够找到传统纯文本方法可能遗漏的相关文档,尤其是在视觉信息至关重要的场合,例如包含图表或包含图表和图形的科学论文的财务报告。
后期交互机制
在检索阶段,交互 描述了通过比较文档的向量表示来评估文档与用户查询的相关性的过程。这种比较有助于系统将文档与查询的意图和内容匹配,从而获得更准确的搜索结果。
ColPali 利用后期交互,即在检索的最后阶段之前分别处理查询和文档。这种后期交互机制可以对图像网格单元向量表示与查询文本标记向量表示进行丰富的比较。通过在查询时执行这些比较,ColPali 避免了对图像和文本一起进行处理的繁重计算负载,从而优化了检索效率。这种方法加快了检索速度并降低了系统的处理要求,使其非常适合处理海量的文档集合。
展望未来
ColPali 架构为文档检索树立了新标准,提供了一个灵活的框架,可以适应新兴的 VLM。基准测试结果表明 ColPali 优于传统方法,标志着该领域范式转变。ColPali 为更加复杂且具备上下文感知功能的系统铺平了道路,这些系统通过将视觉信息有效集成到 RAG 管道中,革新了文档交互和理解。
有了 ColPali 和 Vespa,开发人员仅使用文档页面的可视化表示,即可为 PDF 等复杂文档格式构建一个完整的 RAG 管道。Vespa 的复杂张量框架和计算引擎无缝容纳 ColPali 嵌入,以便通过 Vespa 排名表达式实现后期交互评分。
您可以使用我们的综合笔记本探索 ColPali 的潜力,展示如何在 Vespa 中利用 ColPali 嵌入。深入视觉文档检索的世界,亲自体验 ColPali 的强大功能!