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