百万上下文下的 DeepSeek V4:SGLang 推理优化实战|AICon深圳

点击查看原文>

Agent 时代,哪些方向正在成为行业关键变量?50 + 实战案例揭晓答案!

模型参数规模不断突破,推理成本持续下降,开源生态日益繁荣。当模型能力逐渐成为行业共识,一个新的问题开始浮现:当人人都能获得强大的模型能力之后,真正的竞争力还剩下什么? 答案正在从模型能力本身,转向围绕模型构建可规模化的智能系统;从单点能力提升,转向系统工程与组织级落地能力。

在这一背景下,2026 年 AICon 人工智能开发与应用大会 · 深圳站正式启动。本次大会将于 8 月 21 日—22 日举办,聚焦 AI 基础设施、大模型系统、智能体工程、数据智能、多模态技术与行业落地等关键方向,邀请来自腾讯、阿里、华为、百度、蚂蚁集团等 50 + 头部科技企业技术负责人、科研机构一线专家,系统性分享前沿洞察与实战干货,共同探讨 AI 技术从能力到系统、从实验到生产的真实路径。

SGLang, RadixArk Engineer 杨雨豪已确认出席 “AI Infra、推理工程与异构计算” 专题,并发表题为百万上下文下的 DeepSeek V4:SGLang 推理优化实战的主题分享。DeepSeek V4 模型采用了由 SWA(滑窗注意力)、CSA(压缩稀疏注意力,4:1 压缩+TopK 筛选)和 HCA(高压缩注意力,128:1 压缩)组成的混合注意力架构,相较于 V3 单一的 MLA 结构更为复杂,同时模型支持百万级上下文长度,对推理框架的缓存管理、算子效率和并行策略提出了严苛挑战。本次演讲分享 SGLang 团队如何通过缓存架构、算子融合和并行策略的全栈优化,实现 DeepSeek-V4 百万上下文推理的 Day-0 支持与持续性能提升,并结合 InferenceX 公开基准展示在 GB300 NVL72 上的实测结果。他在本次会议的详细演讲内容如下:

演讲提纲:

一、DeepSeek V4 推理的核心挑战

  1. 混合注意力架构的复杂性:SWA(滑窗 128)、CSA(4:1 压缩 + TopK 稀疏筛选)、HCA(128:1 纯压缩)三种异构 Attention 共存,每种有独立的 KV Cache 生命周期和索引逻辑

  2. 百万级上下文带来的内存与计算压力:KV Cache 显存占用激增,TopK 等索引操作在长序列下成为瓶颈

  3. Compressor 机制引入跨 token 依赖:CSA/HCA 的 KV 计算需复用前序 token 中间状态,打破了传统逐 token 独立计算的假设

二、ShadowRadix:三套 KV Cache 的统一管理

  1. 在 RadixTree 上扩展虚拟地址表,每个 token 分配唯一虚拟位置,通过除法映射到三组 Shadow 页表(SWA 原始位置 / CSA 除以 4 / HCA 除以 128)

  2. SWA 引入 Tombstone 机制,自动释放滑窗外过期 KV,Shadow B/C 无需淘汰

  3. 额外维护两个 Ring Buffer 管理 Compressor 中间状态,保证 CSA/HCA 跨 token 依赖的正确性

  4. 踩坑:三套 Cache 的页大小、淘汰策略、前缀匹配逻辑完全不同,SWA 为支持前缀缓存需保留多于 128 token 的 KV,不能简单按滑窗大小截断

三、算子级优化

  1. FlashCompressor:将 Compressor 流程中 Softmax、bias、scale-dot 等零散操作融合为单一 kernel,HBM 读写从 5 次降至 2 次

  2. Lightning TopK:序列分片到同一 Cluster 内的多个 CTA,各 CTA 用直方图做本地统计,通过 SM 互联(而非 HBM)完成 reduce;百万上下文 BS=1 下从 100μs 降至 15μs

  3. MegaMoE(DeepSeek 官方):将 dispatch / group GEMM / combine 三阶段融合重叠为单一 kernel,直接调用 DeepGEMM 实现;限制:仅支持 FP8×FP4、仅 Blackwell SM100、仅 NVLink 互联

  4. 踩坑:Kernel 语言选型——Triton 开发快但优化上限有限,CUDA 可做极致优化但开发慢,TileLang 居中;FlashCompressor 用 Triton 即可,Lightning TopK 必须用 CUDA

四、并行策略与工程化

  1. DP Attention:V3 延续方案,提升吞吐

  2. CP(Context Parallelism):将长序列均匀分片到多卡,Attention 前 AllGather 所需 KV(因压缩后体积小,通信开销低);超长输入(90 万+)下 CP 优于 TP,短输入 TP 更优(CP 额外通信开销不划算)

  3. EP(Expert Parallelism):GB300 NVL72 上大规模专家并行,AlltoAll 走 NVLink 带宽充足

  4. 多流并行:SWA/CSA/HCA 三种 Attention 无数据依赖,分配到不同 CUDA Stream;解码阶段 SM 利用率未饱和,多流收益明显;用 CUDA Event 控制依赖顺序

  5. 踩坑:FlashMLA 库专为 DP Attention 设计(头数完整),开启 TP 切分头数后只能 padding 到要求的头数,存在浪费,后续仍需优化;CP 的 KV AllGather 目前无法做 overlap(前后有数据依赖),计划用 Ring Attention 方案解决

五、投机采样适配

  1. V4 MTP Layer 仅含 SWA,无 CSA/HCA,结构比主模型轻量

  2. 将 KV 位置等元数据准备移入 CUDA Graph,消除 CPU 开销;兼容 Overlap Scheduling 实现 CPU/GPU 完全并行

  3. 实测 4K 与 900K 上下文解码速度差异极小(TP=8, BS=1, MTP Len=3)

  4. 踩坑:大 Batch(512/1024)下投机采样收益下降,因 GPU 瓶颈从访存转为计算,投机采样本质优化的是访存

六、KV Cache Offloading 与 PD 分离

  1. HiSparse:CSA pool 溢出时卸载至 CPU 内存,SWA/HCA 体积小暂留 GPU

  2. PD 分离结合 ShadowRadix 虚拟索引,KV Cache 按配置从 P 节点传输至 D 节点

七、性能数据

  1. InferenceX 基准(Input 8K / Output 1K):开启 MTP 单用户交互速度 180 token/s;GB300 NVL72 高吞吐场景单 GPU 吞吐 11,500 token/s

  2. 强化学习框架 Miles Day-0 支持:Rollout 阶段用 FP8、Training 用 BF16 混合精度,奖励值与 benchmark 分数随训练步数稳步提升

实践痛点

1. 架构复杂度换性能

ShadowRadix 为三种 Attention 各维护一套独立的页表、淘汰策略和前缀匹配逻辑,系统复杂度相比 V3 单一 MLA 大幅膨胀。例如 SWA 为支持前缀缓存必须保留多于 128 token 的 KV,不能按滑窗大小简单截断,即使是最基础的缓存管理也充满 corner case。三套 Cache 的联调、一致性维护和 bug 排查成本很高,后续每新增一个功能(如 PD 分离、CP)都要在三套系统上分别适配。

2. 并行策略没有通解

CP 在超长输入下优于 TP,但 AllGather 目前无法做 overlap(有数据依赖),且长上下文高并发容易 OOM;TP 在短输入下更好但受限于卡间通信;EP 依赖 NVLink 带宽。本质上不存在自动化策略选择,用户需要自己在 DP/TP/CP/EP 的组合空间里反复试参,调优成本不低。运行时动态切换并行策略也不可行,涉及 CUDA Graph 重建、元数据重建等问题。

3. Kernel 开发的工程代价

实际开发中 Triton、CUDA、TileLang 三种语言混用:Triton 开发快但优化上限有限,CUDA 能做极致优化但开发慢,TileLang 居中。Lightning TopK 必须用 CUDA,FlashCompressor 用 Triton,训练反向算子用 TileLang。三套工具链混合意味着团队需要同时维护三种代码风格和调试流程,且融合 kernel 一旦出现数值精度问题,排查难度远高于朴素 PyTorch 实现。

听众收益

  • 复杂模型架构落地的工程方法论:DeepSeek V4 的三种异构 Attention 共存是业界新趋势,ShadowRadix 的虚拟地址映射 + 分层页表设计提供了一套可复用的思路——当模型架构变复杂时,如何在不重写整个缓存系统的前提下做扩展式适配,而非推倒重来。

  • 性能优化的决策框架:分享覆盖了从算子融合(FlashCompressor 减少 HBM 读写)、到并行策略选型(短输入用 TP、超长输入用 CP、高吞吐用 EP)、再到投机采样适用边界(小 Batch 有效、大 Batch 收益下降)的完整决策链路,帮助参会者在自身业务场景中快速定位优化方向,避免盲目套用单一方案。

  • 多硬件适配的实战经验:Kernel 语言选型(Triton 快速迭代 vs CUDA 极致优化 vs TileLang 折中)、MegaMoE 的硬件约束(仅 Blackwell + NVLink)、FlashMLA 在 TP 下的 padding 浪费等具体案例,为正在做跨平台部署或硬件选型的团队提供真实的 tradeoff 参考,减少试错成本。

除此之外,本次大会还策划了AI Infra、推理工程与异构计算超级个体与蜂群智能的共生进化迈向机器人 AGI 的关键技术与产业实践Agent 安全:从风险到可控端侧智能与 AI 原生终端AI Agent 高价值商业场景实战等 11 个专题论坛,届时将有来自不同行业、不同领域、不同企业的 50+资深专家在现场带来前沿技术洞察和一线实践经验。

大会限时早鸟票享 8 折专属优惠,现在报名立减 1160,更多详情可扫码或联系票务经理 13269078023 进行咨询。


本文来源:InfoQ