🚀 ColQwen2:基于Qwen2-VL-2B-Instruct与ColBERT策略的视觉检索器
ColQwen是一个基于视觉语言模型(VLMs)的新型模型架构和训练策略的模型,可从文档的视觉特征中高效地对文档进行索引。
它是Qwen2-VL-2B的扩展,能生成ColBERT风格的文本和图像多向量表示。
该模型在论文ColPali: Efficient Document Retrieval with Vision Language Models中被提出,并首次在此仓库中发布。
此版本是未经训练的基础版本,以确保投影层初始化的确定性。

🚀 快速开始
ColQwen2是基于视觉语言模型的创新之作,能利用视觉特征高效索引文档。它扩展自Qwen2-VL-2B,生成的多向量表示为文档检索带来新的可能。
✨ 主要特性
- 动态图像分辨率:该模型接受动态图像分辨率输入,且不改变图像的纵横比,最大分辨率设置为最多创建768个图像块。实验表明,增加图像块数量可显著提升性能,但会增加内存需求。
- 特定版本训练:此版本使用
colpali-engine==0.3.1
进行训练,训练数据与论文中描述的ColPali数据相同。
📦 安装指南
确保安装colpali-engine
的源码版本或版本号高于0.3.1,同时transformers
版本需大于4.45.0。
pip install git+https://github.com/illuin-tech/colpali
💻 使用示例
基础用法
import torch
from PIL import Image
from transformers.utils.import_utils import is_flash_attn_2_available
from colpali_engine.models import ColQwen2, ColQwen2Processor
model = ColQwen2.from_pretrained(
"vidore/colqwen2-v0.1",
torch_dtype=torch.bfloat16,
device_map="cuda:0",
attn_implementation="flash_attention_2" if is_flash_attn_2_available() else None,
).eval()
processor = ColQwen2Processor.from_pretrained("vidore/colqwen2-v0.1")
images = [
Image.new("RGB", (128, 128), color="white"),
Image.new("RGB", (64, 32), color="black"),
]
queries = [
"Is attention really all you need?",
"What is the amount of bananas farmed in Salvador?",
]
batch_images = processor.process_images(images).to(model.device)
batch_queries = processor.process_queries(queries).to(model.device)
with torch.no_grad():
image_embeddings = model(**batch_images)
query_embeddings = model(**batch_queries)
scores = processor.score_multi_vector(query_embeddings, image_embeddings)
📚 详细文档
模型训练
数据集
我们的训练数据集包含127,460个查询 - 页面配对,其中63%来自公开可用的学术数据集的训练集,37%是由网络爬取的PDF文档页面组成的合成数据集,并使用VLM生成(Claude-3 Sonnet)的伪问题进行增强。训练集设计为全英文,以便研究对非英语语言的零样本泛化能力。我们明确验证了没有多页PDF文档同时用于ViDoRe和训练集,以防止评估污染。使用2%的样本创建验证集来调整超参数。
注意:多语言数据存在于语言模型的预训练语料库中,并且很可能存在于多模态训练中。
参数
所有模型在训练集上训练1个epoch。除非另有说明,我们以bfloat16
格式训练模型,在语言模型的变压器层以及最终随机初始化的投影层上使用低秩适配器(LoRA),alpha=32
且 r=32
,并使用paged_adamw_8bit
优化器。我们在8个GPU上进行数据并行训练,学习率为5e-5,采用线性衰减,热身步骤为2.5%,批量大小为32。
局限性
- 聚焦范围:该模型主要专注于PDF类型的文档和资源丰富的语言,可能限制了其对其他文档类型或资源较少语言的泛化能力。
- 支持情况:该模型依赖于从ColBERT后期交互机制派生的多向量检索,可能需要一定的工程工作才能适应缺乏原生多向量支持的广泛使用的向量检索框架。
📄 许可证
ColQwen2的视觉语言主干模型(Qwen2-VL)遵循apache2.0
许可证,附加到模型的适配器遵循MIT许可证。
👥 联系方式
- Manuel Faysse: manuel.faysse@illuin.tech
- Hugues Sibille: hugues.sibille@illuin.tech
- Tony Wu: tony.wu@illuin.tech
📚 引用
如果您在研究中使用了该组织的任何数据集或模型,请按以下方式引用原始数据集:
@misc{faysse2024colpaliefficientdocumentretrieval,
title={ColPali: Efficient Document Retrieval with Vision Language Models},
author={Manuel Faysse and Hugues Sibille and Tony Wu and Bilel Omrani and Gautier Viaud and Céline Hudelot and Pierre Colombo},
year={2024},
eprint={2407.01449},
archivePrefix={arXiv},
primaryClass={cs.IR},
url={https://arxiv.org/abs/2407.01449},
}