重排序模型解读 mxbai-rerank-base-v2 强大的重排序模型

mxbai-rerank-base-v2 强大的重排序模型

模型介绍

mxbai-rerank-base-v2 是 Mixedbread 提供的一个强大的重排序模型,旨在提高搜索相关性。该模型支持多语言,特别是在英语和中文方面表现出色。它还支持代码和 SQL 排序,能够处理长达 8k 个标记的上下文,并且在技术内容排序方面表现优异。

模型类别(20250517)

  • mxbai-rerank-base-v2
  • mxbai-rerank-large-v2 (🍞)

训练使用了三个步骤进行训练:

  • GRPO (引导式配筋提示优化)
  • 对比学习
  • 偏好学习

主要特点

  • 多语言支持:支持 100 多种语言,特别是在英语和中文方面表现出色。

  • 长上下文支持:能够处理长达 8k 个标记的上下文,适用于复杂的查询和文档。

  • 代码和 SQL 排序:在排序代码片段和技术内容方面表现优异。

  • 快速推理:推理速度比同类模型快 8 倍,适合需要快速响应的应用场景。

  • 开源和易于集成:基于 Apache 2.0 许可证开源,易于定制和集成到现有的搜索系统中。

benchmark

在 A100 GPU (80GB)上测量:

综合评价

1.5B 模型比 bge-reranker-v2-gemma 快 8 倍,同时提供更高的准确性。这种速度优势意味着您可以在不牺牲质量的情况下每秒处理更多查询,使模型成为性能和成本效益都很重要的大批量生产环境的理想选择。

BEIR 是评估英语信息检索模型的行业标准基准。Mixedbread 的 rerank-v2 在 BEIR 排行榜上处于领先地位,其性能优于所有竞争模型,并接近最先进的嵌入模型的有效性,同时使用 BM25 作为第一阶段检索器。

模型通过重新排列候选文档以更好地与用户查询保持一致,显著提高了搜索质量。大型变体(1.5B 参数)的 BEIR 得分为 57.49,是基准中最高的,而我们的基本变体 (0.5B) 也以令人印象深刻的 55.57 分超过了更大的型号。

这代表了对上一代产品的重大改进,与 v1 相比,v2 模型的性能提高了 8 个百分点以上。这些改进使我们的 reranker 对于检索质量至关重要的生产环境非常有价值,而无需大型模型的极端资源需求。组织现在可以有效地部署最先进的搜索功能,确保高准确性和最佳性能。

BEIR Benchmark Performance:

博客地址:https://www.mixedbread.com/blog/mxbai-rerank-v2

huggingface:https://huggingface.co/mixedbread-ai/mxbai-rerank-large-v2

安装

bash 复制代码
pip install mxbai-rerank

推理示例代码

python 复制代码
from mxbai_rerank import MxbaiRerankV2

# 初始化重排序器
model = MxbaiRerankV2("mixedbread-ai/mxbai-rerank-base-v2")

# 示例查询和文档
query = "《杀死一只知更鸟》的作者是谁?"
documents = [
"《杀死一只知更鸟》是哈珀·李于1960年出版的一部小说。",
"《白鲸记》是赫尔曼·梅尔维尔写的小说。",
"哈珀·李于1926年出生在阿拉巴马州的门罗维尔。"
]

# 获取排序结果
results = model.rank(query=query, documents=documents)
print(results)
相关推荐
AI4Sci.11 小时前
在云服务器上基于lora微调Qwen2.5-VL-7b-Instruct模型(下)
人工智能·算法·机器学习·大模型·lora微调·大模型本地部署·qwen2.5-vl-7b
陈敬雷-充电了么-CEO兼CTO12 小时前
字节跳动开源Coze,开启AI Agent开发新时代?
人工智能·gpt·chatgpt·开源·大模型·agi·coze
Eastmount13 小时前
[AI Coding] 一.腾讯CodeBuddy IDE内测、安装及基本用法(国产AI IDE启航)
人工智能·大模型·腾讯·ai coding·codebuddy ide
山顶夕景1 天前
【LLM】Kimi-K2模型架构(MuonClip 优化器等)
大模型·llm·agent·强化学习·智能体
哥本哈士奇(aspnetx)1 天前
Dify快速搭建问答系统
大模型
威化饼的一隅1 天前
【多模态】天池AFAC赛道四-智能体赋能的金融多模态报告自动化生成part2-报告输出
大模型·agent·多模态·智能体
精致先生1 天前
RAG(检索增强生成)
人工智能·大模型·rag
威化饼的一隅1 天前
【多模态】天池AFAC赛道四-智能体赋能的金融多模态报告自动化生成part1-数据获取
大模型·agent·多模态·智能体
美林数据Tempodata2 天前
美林数据用大模型重构电能质量评估,让隐蔽合规问题无所遁形
重构·大模型
胡耀超2 天前
我们如何写好提示词、发挥LLM能力、写作指南:从认知分析到动态构建的思维方法
人工智能·python·学习·大模型·llm·提示词·八要素思维