RAGO 深度分析报告

RAGO: Systematic Performance Optimization for Retrieval-Augmented Generation Serving
谷歌 DeepMind × UCSD ACM SIGCOMM 2025 RAG 服务端全链路优化 开源 Artifact

一、论文基本信息

正式标题RAGO: Systematic Performance Optimization for Retrieval-Augmented Generation Serving
发布时间2025 年
发布平台arXiv、ACM SIGCOMM 2025(核心计算机网络 / 系统顶会)
作者机构谷歌 DeepMind、加州大学圣地亚哥分校(UCSD)
开源支撑完整的 Artifact 开源项目,可复现实验,集成主流 RAG 流水线框架
核心定位业界首篇针对 RAG 服务端开展全链路系统化异构资源优化的顶级学术研究
注:论文从工作负载抽象、瓶颈建模、异构资源调度三层维度,打通了 RAG 算法设计与底层系统硬件的协同优化壁垒。

二、研究背景与核心动机

2.1 RAG 技术的产业价值

纯大模型架构(LLM-only)的三大固有缺陷:

  • 知识时效性缺陷 — 参数固化的训练数据无法实时同步最新知识,全量微调成本极高
  • 事实幻觉风险 — 统计关联生成缺乏事实依据,金融/医疗/政务场景存在合规风险
  • 知识边界刚性 — 参数规模决定知识上限,扩展参数会指数级推高部署成本

RAG 的三大落地优势:

  • 知识更新成本极低 — 仅需增量更新向量索引,无需重新训练
  • 事实幻觉风险显著降低 — 生成被限定在可溯源的权威知识范围内
  • 部署成本更可控 — 10B 级小模型即可达到 70B 级模型的生成质量

2.2 现有技术的研究缺口

1. 多组件异构性缺口 — 串联向量检索(CPU)、查询重写(小 GPU)、重排(轻量 GPU)、推理(大 GPU)等多阶段异构负载,缺乏统一调度机制。
2. 工作负载可变性缺口 — 从百万级到千亿级向量库,配置组合空间极广,缺乏标准化负载抽象。
3. 流水线阻塞性缺口 — 迭代检索导致推理流水线频繁切换,现有拆分部署未考虑跨阶段依赖,易出现资源空闲。

2.3 核心设计原则

抽象
工作负载标准化可量化
解码
实证拆解瓶颈边界
系统化
自动挖掘最优调度策略

三、RAGSchema 标准化负载抽象

业界首个面向服务端性能优化的 RAG 工作负载结构化抽象,解决性能不可比、优化无基准的难题。

维度核心属性对性能的影响
检索侧属性向量库规模、topK、文档长度、索引类型、混合检索比例、重排规则决定 CPU 计算量、存储 IOPS、内存带宽占用
流水线侧属性查询重写、重排启用、重写模型规模、迭代检索频率决定阶段串联逻辑、端到端延迟传导
资源侧属性CPU 规模、GPU/TPU 数量、资源分配比例、存储介质、网络延迟决定整体资源上限与资源竞争程度

四类典型 RAG 范式

  1. 超大规模检索 (Hyperscale Retrieval)
  2. 长上下文排序 (Long-Context Sequence Processing)
  3. 解码中迭代检索 (Iterative Retrievals + Prefix)
  4. 带查询重写+重排序的完整流水线

四、实证性能瓶颈分析

场景一:超大规模检索
亿级至千亿级向量库,高并发,topK>100
• 检索阶段开销占比 >60%
• 瓶颈在 CPU 侧:计算吞吐、随机读 IOPS、内存带宽
• 批处理规模失衡导致推理资源空闲
核心结论:检索效率直接决定系统吞吐量上限
场景二:长上下文排序
单文档>1000 token,总长度>4096 token
• 前缀阶段开销占比 >50%
• 1:1 资源分配完全不匹配
• KV-cache 传输放大内存带宽瓶颈
核心结论:前缀阶段 GPU 资源配比对性能影响最大
场景三:解码中迭代检索
解码阶段多轮检索,单轮对话检索>2次
• 流水线阻塞空闲是核心瓶颈
• 批处理比例失衡延迟高 2.77 倍(比例相差>4倍)
• 网络传输延迟被多轮迭代放大,涨幅超 300%
核心结论:批处理比例失衡比检索延迟更严重
场景四:完整流水线(查询重写+重排)
8B 查询重写模型 + 120M 重排模型
• 查询重写是 TTFT 延迟核心来源,TTFT 升至 2.4 倍
• 重排模块影响可忽略 (<5%)
• 小模型部署位置放大跨阶段网络依赖
核心结论:非核心模型组件的部署策略对延迟影响远大于其计算开销

五、RAGO 系统化优化框架

业界首个覆盖 RAG 全链路的异构资源优化调度框架,核心是 "混合协同-拆分调度策略"。

5.1 任务放置(Task Placement)

策略适用场景说明
强拆分主 LLM 前缀/解码前缀计算密集、解码内存密集,拆分避免资源竞争
协同部署数据库编码、重排、前缀计算强度相近,时间分片复用加速器
弱拆分查询重写与解码强依赖阶段独立部署,避免自回归拖慢解码

5.2 资源分配(Resource Allocation)

  • 计算密集型(前缀、重排、编码)→ 高算力 GPU/TPU
  • 内存密集型(解码、向量检索)→ 高带宽内存
  • 检索 → 多核 CPU 资源提升并行能力
  • 迭代检索场景:解码 ~50% 资源、前缀 ~30%、检索/重排余下

5.3 批处理策略(Batching Policy)

  • 延迟敏感(重排、前缀)→ 小批量,优先 TTFT
  • 吞吐敏感(解码)→ 大批量,最大化资源利用
  • 检索 → 动态调整,与解码严格匹配(关键)

5.4 Pareto 最优搜索机制

  1. 性能画像 — 基于 RAGSchema 参数量化各阶段开销、延迟、吞吐上限
  2. 策略空间遍历 — 遍历任务放置/资源分配/批处理组合,筛选满足 SLA 的候选
  3. 帕累托最优筛选 — 找到吞吐最大化+延迟最小化的最优边界,生成可部署配置
核心优势:将原本耗时数天的人工调优压缩到分钟级,适配精度远高于人工配置。

六、实验设计与结果

6.1 实验环境

组件配置
向量检索AMD EPYC 9534 64核 / Intel Xeon Gold 5218 CPU
LLM 推理NVIDIA H100 80GB GPU
测试基准HotpotQA(多跳问答)、MS MARCO(必应真实数据)
对比基线LLM-only(70B 级)、传统 RAG(纯拆分 + 1:1 分配)

6.2 核心实验结果

2x
单芯片 QPS 提升
-55%
TTFT 延迟降低
-60%
硬件成本降低
-70%
流水线等待消除
  • 8B 级模型 RAG 系统单芯片 QPS 是 70B 级 LLM-only 方案的 1.5 倍
  • 迭代检索场景通过批处理对齐消除近 70% 流水线等待延迟
  • 将推理开销转移到低成本 CPU,硬件成本较 LLM-only 降低 60%+

6.3 敏感性分析

  • 任务放置:对长上下文/迭代检索影响权重最大,优化空间超 40%
  • 批处理比例:迭代检索核心因素,比例相差>4倍延迟高 2.77 倍
  • 资源分配:检索 CPU 资源 ×2 → 吞吐量提升 ~60%
  • 模型组件:查询重写模型规模对 TTFT 影响最大,重排规模可忽略

6.4 与同类优化技术差异

对比对象RAGO 差异
PipeRAG / RaLMSpec不仅优化解码阶段,覆盖检索→重写→重排→前缀→解码全链路
Chameleon通用型框架,适配所有四类 RAG 场景,非单一检索加速
LangChain / LlamaIndex严格数学成本模型 + 实证数据,无需人工试错
RAGCache / CacheBlend同时覆盖资源调度、批处理策略、组件部署

七、研究价值与落地意义

RAGSchema
首个标准化负载抽象
系统层
首次从系统层切入 RAG 优化
可复用
调度逻辑可扩展到 Agentic AI
  • 成本与体验平衡:降硬件成本同时 TTFT 降低一半以上
  • 部署优先级:任务放置 → 资源分配 → 批处理校准
  • 无侵入集成:Artifact 可无缝集成 LangChain/LlamaIndex
  • "小模型+检索"范式:8B+检索达到 70B 质量,吞吐更高、延迟更低

八、研究局限与后续方向

局限说明
场景覆盖未覆盖复杂 Agentic RAG(多工具编排、长时自治)
部署环境仅单集群调度,未覆盖多集群/跨可用区
优化维度未覆盖检索算法本身与存储层硬件优化
调度目标未纳入能耗、成本、SLA 优先级
网络假设未考虑网络延迟随机波动

后续研究方向

  1. 场景扩展 — 覆盖 Agentic AI:工具调用、多步骤编排、长时状态管理
  2. 算法-系统协同 — 向量检索算法优化 + 系统调度协同设计
  3. 调度目标多元化 — 纳入能耗、成本、SLA、网络波动
  4. 迭代检索优化 — 异步预取、KV-cache 跨阶段复用、结果缓存
  5. 硬件架构定制 — 存算一体、近数据计算等新硬件适配

九、核心结论

里程碑价值:建立了一套从工作负载抽象、瓶颈建模、到异构资源协同调度的完整系统化优化方法论,填补了 RAG 技术在算法层与系统层之间的优化缺口。

关键技术结论:

  • RAG 优化不能照搬 LLM-only 的"前缀-拆分"惯例,需采用混合协同-拆分调度策略
  • 不同场景核心瓶颈差异极大,需通过标准化负载抽象精准定位
  • 批处理比例对齐是优化迭代检索的关键,影响权重高于单纯加硬件
  • 系统化调度可使"小模型+RAG"性能超越超大参数纯 LLM,成本显著更低

本页面由 AI 辅助整理,基于豆包深度分析报告。参考资料:论文 arXiv 完整版、ACM SIGCOMM 2025 正式版、开源 Artifact(google/rago)、HippoRAG/GraphRAG/RAGCache 等行业资料。

⚠️ 免责声明

本站内容均由 AI 基于公开知识辅助生成,仅供学习参考,请勿直接引用作为依据。作者不对信息的准确性、完整性及适用性作保证,亦不对因使用本站内容产生的任何损失承担责任。