AI Infra 推理优化主题

主题定义

AI Infra 推理优化涵盖大模型推理服务的完整技术栈——从底层的数学原理到工程实现,再到上层的 RAG 检索增强生成和 Agent Memory 系统。

核心要点

  1. Infra 优化的本质是数学等价变换:RMSNorm 砍掉均值计算将两次全局规约简化为一次,是数学简化而非工程 trick
  2. Paged Attention 是 OS 虚拟内存在 GPU 上的重演:借鉴分页思想把 KV Cache 切成固定大小 Block,显存利用率提升 2-4 倍
  3. Continuous Batching 消除短请求被长请求拖累:每个 step 结束后动态调度
  4. vLLM 已成为 LLM Serving 的事实标准:Flattened 布局 + slot_mapping + cu_seqlens
  5. RAG 全链路已形成完整工程方法论:Meta-Chunking、HyDE、Graph RAG、Ragas 评估
  6. Agent Memory 的三种哲学:渐进式压缩(腾讯云)、结构化组织(Memoir)、双层分离(Spring AI)
  7. 记忆即版本控制:Memoir 把 Git 的 branch/commit/merge/rollback 搬进记忆层
  8. 数学等价变换是推理优化主线:RMSNorm 去均值、Safe Softmax 减最大值、Online Softmax 动态修正、Gumbel-Max 采样,都在用等价变换或受控近似换取更少 HBM 访存、更高并行度和更稳定的数值范围
  9. KV-Cache 是 Serving 的核心资源:Prompt Cache 的成本收益、vLLM 的 PagedAttention、Flash-Decoding 的 Split-K 与长上下文延迟,都围绕历史 K/V 的存储、读取和合并展开
  10. 推理优化正在上移到 gateway 层OmniRoute 把 provider health、quota、fallback、routing strategy、compression 和成本 header 放到本地代理层,让多个 coding tools 共享同一套模型使用策略。
  11. 本地模型选型也是推理工程的一部分whichllm 在 serving 之前先估算硬件、VRAM、KV cache、量化、benchmark evidence 和速度,避免只按参数量选择一个“能 fit 但不好用”的模型。
  12. 离线 AI Infra 必须把内容、模型和服务一起治理Project-NOMAD 把 Kiwix、Kolibri、ProtoMaps、Ollama、Qdrant、Docker 编排和本地 Knowledge Base 放在同一 Command Center 中,说明离线 RAG 的瓶颈不只在检索算法,也在内容预下载、存储、GPU、更新和访问控制。
  13. Agent 会话天然利于前缀缓存,但高命中率不是终点:多轮 Agent 调用通常只在历史末尾追加工具输出和新回复,前面的大段系统提示、工具定义和历史对话可以反复命中 Prompt-Cache;但 90% 命中率也意味着每轮都在重发大上下文,仍要同时看 cache read、cache write 和未缓存输入的绝对量。

涉及实体

  • vLLM —— LLM 推理框架的事实标准
  • KV-Cache —— 自回归推理中缓存历史 Key/Value 张量的机制
  • Prompt-Cache —— 跨请求复用稳定前缀的产品层缓存抽象
  • RAG —— 检索增强生成
  • Agent-Memory —— Agent 记忆系统
  • OmniRoute —— 多 provider gateway 与 token/compression/cost 治理层
  • whichllm —— 本地 LLM 硬件适配、模型排序与采购规划 CLI
  • Project-NOMAD —— 离线优先的知识与教育服务器,整合本地 AI Chat、Qdrant RAG 和容器化内容服务

对比矩阵

维度vLLMTensorRT-LLMSGLang
开源部分
核心技术Paged AttentionTRT 优化RadixAttention
维度传统 RAGLLM WikiGraph RAG
知识更新实时需重新编译实时
查询延迟

关键来源