首页 >> 综合 > 快讯 >

NVIDIA 推出 Nemotron Retriever 优化 AI Agent 检索 token 消耗

2026-10-30 09:55:36 来源:网易 编辑:欧阳飘学 

NVIDIA 正式发布 Nemotron Retriever 检索工具,这套工具主打优化 AI Agent 运行过程中的 token 消耗问题,依靠提升检索精准度来减少无效上下文传入大模型,直接降低整体 token 开销,同时还能同步提升智能体回答内容的准确性,是面向企业级 Agent 应用的轻量化检索方案。AI Agent 在执行多轮任务时,往往会把大量无关文档内容送入 LLM 上下文,造成 token 用量居高不下,拉高推理成本;Nemotron Retriever 通过嵌入模型 + 重排模型的组合,精准筛选匹配内容,过滤掉冗余段落,只把高相关性文本交给大模型处理,从源头压缩上下文长度,实现 token 消耗稳步下降。整套工具包含开源权重模型、GPU 加速检索库与 NIM 微服务,开发者可以直接部署到本地 GPU 环境,不用依赖云端接口,兼顾低成本与运行效率。在实测场景下,替换原有检索组件后,Agent 单次查询的 token 成本明显下降,检索准确率还同步提升,兼顾降本与效果两大目标。开发者只需要在检索链路替换嵌入与重排模型,开启智能体检索模式,就能快速完成部署,轻松控制 AI Agent 长期运行的 token 开销。

一、Nemotron Retriever 核心组件清单

  1. 嵌入模型(控制向量检索精度)
  • nemotron-3.5-embed-1b:轻量版本,适合小规模 Agent 业务
  • nemotron-3.5-embed-8b:高精度版本,适配复杂知识库检索
  1. 重排模型(过滤无关内容,缩减 token)
  • nemotron-3.5-rerank-8b:对初步检索结果做二次打分,剔除低相关性文档,大幅减少进入 LLM 的上下文 token 数量
  1. 配套运行组件
  • NeMo Retriever 开源检索库:GPU 加速文档分块、多模态内容提取
  • NIM 微服务:一键容器化部署,支持本地私有化部署,无需公网调用

二、降低 AI Agent 检索 token 消耗的实现步骤

  1. 文档预处理阶段:使用检索库对 PDF、表格、图文资料做结构化分块,控制单块文本长度,避免大块冗余文本。
  2. 向量检索阶段:调用 Nemotron 嵌入模型生成向量,从向量库召回少量候选文档,缩小初选范围。
  3. 重排过滤阶段:开启 rerank 重排模型,对候选文档做相关性排序,只保留 Top3~Top5 高匹配段落,砍掉无效内容。
  4. Agent 调用阶段:把精简后的上下文送入大模型,减少传入 token 总量,完成问答任务。

三、实测 token 优化数据对照表

表格

检索方案单次查询平均 token 消耗量检索准确率
传统无重排检索基准用量基准水平
Nemotron Retriever(1B 嵌入 + 8B 重排)降低约 13%小幅提升
Nemotron Retriever(8B 嵌入 + 8B 重排)降低约 14%明显提升

四、适配 AI Agent 的核心能力

  1. 智能体检索模式:支持 ReAct 多轮检索,自动拆分复杂问题,多次检索再融合结果,同时严格控制每一轮检索带出的文本量。
  2. 多模态文档支持:可解析表格、图表内容,避免图文类文档检索遗漏,不用额外增加大量文本 token。
  3. 私有化部署:全部模型支持本地 GPU 运行,没有云端调用的 token 计费,长期运行成本更可控。
  4. 接口兼容:提供 REST、Python SDK、CLI 三种调用方式,可直接接入现有 Agent 项目,改动量很小。

【相关 FAQ】

问题 1:NVIDIA 推出 Nemotron Retriever 优化 AI Agent 检索 token 消耗,这款工具减少 token 消耗的核心原理是什么?

回答 1:NVIDIA 推出 Nemotron Retriever 优化 AI Agent 检索 token 消耗,核心原理是依靠重排模型过滤掉低相关性文档,只把高度匹配的文本送入大模型上下文,减少无效文本带来的额外 token 开销,同时依靠高精度嵌入模型缩小初选文档范围,从检索全链路压缩上下文长度。

问题 2:NVIDIA 推出 Nemotron Retriever 优化 AI Agent 检索 token 消耗,部署 Nemotron Retriever 需要什么样的硬件环境?

回答 2:NVIDIA 推出 Nemotron Retriever 优化 AI Agent 检索 token 消耗,核心模型可在单张 A10G 及以上 GPU 运行,轻量 1B 嵌入模型甚至可以在中端消费级 GPU 部署,整套检索链路全程 GPU 加速,不会额外增加算力负担,适合中小规模 AI Agent 项目落地。

问题 3:NVIDIA 推出 Nemotron Retriever 优化 AI Agent 检索 token 消耗,Nemotron Retriever 可以直接对接现有 AI Agent 项目吗?

回答 3:NVIDIA 推出 Nemotron Retriever 优化 AI Agent 检索 token 消耗,Nemotron Retriever 提供标准化 API 接口,仅需要替换原有检索环节的嵌入与排序组件,不用改写 Agent 主体逻辑,短时间内就能完成接入,快速实现 token 用量优化。

【数据来源】

[2026-10-29](Nemotron Retriever 优化 AI Agent 检索 token 开销)CSDN 技术博客

[2026-10-28](NVIDIA 发布 Nemotron Retriever 智能检索套件)NVIDIA 开发者官网

[2026-10-27](Nemotron 系列检索模型开放权重 降低 Agent 推理成本)英伟达中文技术博客

[2026-10-20](Nemotron 3.5 检索模型正式上线)NVIDIA 模型开放平台

免责声明:本文所有实测 token 降幅数据均来自 NVIDIA 官方公开测试环境;实际 token 优化效果会受知识库内容、文档分块策略、模型版本影响;模型部署条件、接口参数请以 NVIDIA NeMo Retriever 官方最新文档为准。

 
最新文章