LightOn-rerank

开源权重 · LightOn ·

Apache 2.0文本和页面图片需要 GPU

LightOn-rerank 是一组六个重排序模型,用同一个模型给文本段落和文档页面图片打分,所以一个第二阶段就能同时服务文本索引和 PDF 页面索引。它有三个尺寸(0.8B、2B、4B),每个尺寸都有逐点和列表式两个版本,基于 Qwen3.5,Apache 2.0 许可。在视觉文档上,列表式 4B 是 LightOn 自己对比里最强的;在文本上它落后于 jina-reranker-m0。

逐点还是列表式

每个尺寸都有两个版本,而选哪个版本比选尺寸更重要:

训练数据只有英文。借助多语言底座,法语不需要额外训练也能用,但比英文略差。

分数,以及是谁测的

模型打分方式ViDoRe V3(页面图片)BEIR(文本)
只用第一阶段(ColQwen2.5)—55.60*—
LightOn-rerank-PW-0.8B逐点48.20*—
LightOn-rerank-LW-0.8B列表式58.25*—
LightOn-rerank-PW-2B逐点59.87*49.13*
LightOn-rerank-LW-2B列表式62.66*48.12*
LightOn-rerank-PW-4B逐点59.80*—
LightOn-rerank-LW-4B列表式64.69*48.33*
Qwen3-VL-Reranker-2B逐点59.18*49.08
Qwen3-VL-Reranker-8B逐点64.23*—
jina-reranker-m0逐点59.40*51.97

LightOn 自己的测试,数据来自模型卡,所有模型都在同一套流程下重新评测。ViDoRe V3:对 ColQwen2.5 召回的前 100 张页面图片重排,指标 NDCG@10。BEIR:对 BM25 召回的前 100 个结果重排,指标 NDCG@10,取 11 个没进过 LightOn 训练数据的数据集平均。两者都和厂商自己公布的数字对不上:比如 Qwen 的模型卡按自己的设置给 8B 的 ViDoRe V3 分数更高(见 Qwen3-VL-Reranker 页面)。

这张表说明三件事。列表式模型越大越强,LW-4B 以大约一半的参数略微超过 Qwen3-VL-Reranker-8B;逐点 4B 并不比逐点 2B 好。在文本上,三个公布了分数的 LightOn 模型比 jina-reranker-m0 低三到四分,而 m0 的权重只能非商用。另外,PW-0.8B 的分数比它重排的第一阶段还低,所以不要拿它处理页面图片。

快速上手

逐点模型,用 sentence-transformers

pip install "sentence-transformers[image]"   # 5.4 or later

from sentence_transformers import CrossEncoder

model = CrossEncoder("lightonai/LightOn-rerank-PW-2B")

query = "What is late interaction in neural information retrieval?"
documents = [
    "ColBERT computes token-level query-document interactions at search time...",
    "https://example.com/page_2.png",   # a page image: URL, path or PIL.Image
]
rankings = model.rank(query, documents)

文本候选和图片候选可以放在同一次调用里。模型训练时用的对话模板已经打包在检查点里,不需要额外代码就能正确格式化输入。

列表式模型

LW 模型不能用 CrossEncoder 加载:它们生成的是顺序而不是分数。模型卡里附了一段简短的 transformers 实现,包括四候选窗口和在长列表上的滑动。用 vLLM 部署任何一种时,都要关掉思考模式(enable_thinking: False;Qwen3.5 的模板默认开启,会扭曲分数),给整页图片排序时上下文至少留 16,384 token。

什么时候选 LightOn-rerank

优缺点

优点

  • 每个尺寸都是 Apache 2.0
  • 文本和页面图片一个模型搞定
  • LW-4B 在 LightOn 的视觉文档对比里领先
  • 所有基线都在同一流程下重测,并标出了训练数据污染

缺点

  • 只用英文训练
  • 列表式模型需要自定义代码,延迟约为三倍
  • 文本上落后于 jina-reranker-m0
  • 需要 GPU;太大,放不进我们的浏览器内 Demo

要给页面图片选模型?

按场景选型指南对比了 LightOn-rerank、Qwen3-VL-Reranker 以及处理视觉文档的托管选项。

按场景选型 → 全部模型 →

其他模型