跳到正文
IFAI

开源项目

NVIDIA 开源 Nemotron 3 Embed:8B 模型登顶 RTEB,专治智能体检索低效

首次发布 传播指数 48

NVIDIA 发布开源商用嵌入模型家族 Nemotron 3 Embed,包含 8B 旗舰与两款 1B 轻量版本。其中 8B 模型登顶 RTEB 榜单,全系支持 32k 上下文与代码检索。它主打为智能体流程减少无效检索和多轮推理开销,1B 版本还针对 Blackwell 架构提供 NVFP4 硬件加速。

NVIDIA 开源 Nemotron 3 Embed:8B 模型登顶 RTEB,专治智能体检索低效
图:huggingface.co

NVIDIA 的 Nemotron 3 Embed 是一套面向生产级 RAG 和智能体检索的开源商用嵌入模型家族。它一共包含三个版本:一个拿下 RTEB 多语言榜单第一的 8B 旗舰,以及两个主打低延迟和高吞吐的 1.14B 轻量版本。

这套模型的核心目的,就是解决多步智能体流程里检索精度差导致的额外开销。一旦检索出无关上下文,智能体就必须反复重新查询、增加多余的思考轮次,不仅浪费 token 预算,还会把噪音带进后续的推理流程。

它能做什么

全系三个模型均原生支持 32k 上下文窗口,能直接处理长篇企业文档、大型代码仓库的跨文件检索以及多轮智能体历史交互,减少信息截断。

模型输入格式非常清晰,使用标准 Mean 池化,只需在输入前加上 query: 或 document: 前缀即可。除了通用文本,它对代码检索和多语言企业数据也做了专门支持。

性能方面,8B 版本的 RTEB 得分为 78.5%,MMTEB Retrieval 得分为 75.5%,直接拿下榜首。即使是 1.14B 的 BF16 轻量版本,RTEB 得分也有 72.4%,相比上一代 1B 模型错误率降低了 27%。

评估显示,更精准的检索能让下游搜索智能体更早命中关键证据。在 ViDoRe V3 和 BRIGHT 等测试中,使用该系列嵌入模型直接降低了下游智能体的多轮推理轮次与 token 成本。

怎么做到的

这套模型并非从小规模从头训练,而是借力成熟底座与多轮精炼。

8B 旗舰版以 Ministral-3-8B-Instruct-2512 为底座,将其因果解码器改造成适合全序列检索的双向编码器。随后经过对比预训练,并在法律、金融、医疗、商业等高门槛领域的精选多语言数据集上完成精调。

1.14B 版本则走了一条精细的剪枝与蒸馏路线。团队先基于 Ministral-3-3B 构建双向底座,用 NVIDIA ModelOpt 的神经架构搜索在隐藏层宽度、注意力头和深度上进行结构化剪枝,压缩至 2B 中间模型。

接着用 8B 教师模型对该中间版本进行蒸馏,校准余弦距离与均方误差;之后再重复一遍剪枝与蒸馏,最终压出 1.14B 规格,并通过两阶段渐进式训练把上下文扩展至长文本。

针对硬件加速,NVIDIA 专门打造了 1B-NVFP4 变体。它针对 Blackwell 架构进行了原生 4 位量化与量化感知蒸馏,在保留 BF16 版本 99% 以上精度的同时,吞吐量提升最高可达 2 倍。

怎么跑起来

目前全部模型的权重、训练数据集和微调配方均已在 Hugging Face 开源,并允许商用。

在部署方面,代码已经适配 vLLM 推理框架,官方还同步提供了基于 Rust 的 NVIDIA NIM 微服务。对于想基于私有行业数据二次优化的团队,NVIDIA NeMo AutoModel 也放出了完整的微调和蒸馏配方。

硬件搭配上,对精度要求极高的企业 RAG 可以直接使用 8B-BF16,跑在常规通用 GPU 上即可。1.14B-BF16 则能在低延迟 CPU 或主流消费级显卡上快速部署。如果你手头有 Blackwell 或 GB200 硬件,切换到 NVFP4 版本能最大限度降低显存占用并拉满吞吐。

为什么重要

如果你正在搭建企业级知识库、代码搜索助手或者多步自主智能体,这套模型非常值得现在接入测试。在整个智能体系统里,提升检索精度往往比换用更贵的生成模型划算得多,因为它能从源头上掐断智能体反复搜索和发散推理带来的额外 token 消耗。

我们认为 NVIDIA 这次的产品划分非常聪明。8B 负责在权威榜单上拿第一树立质量标杆,两款 1B 模型则负责解决高并发生产环境下的成本与延迟痛点。虽然 NVFP4 的极致加速需要绑定 Blackwell 硬件,但对于绝大多数开发者来说,标准的 1.14B BF16 已经在通用硬件上提供了极其扎实的性价比。

来源 (1)

  1. [1]
    huggingface.co(在新标签页打开)原文huggingface.co

每日 AI 内容精选