首页 / 文章 / DSpark:置信度调度投机解码——半自回归生成实现推理加速60-85%
← 返回
AI技术

DSpark:置信度调度投机解码——半自回归生成实现推理加速60-85%

✍️ zhirenhun 📅 2026/6/27 👁 170 阅读 ⏱ 8 分钟
DSpark:置信度调度投机解码——半自回归生成实现推理加速60-85%

作者:Xin Cheng, Xingkai Yu et al.(DeepSeek-AI + 北京大学)
原文:DSpark 论文 (PDF) | DeepSpec 开源仓库

DeepSeek 近日开源了 DSpark——一种基于置信度调度(Confidence-Scheduled)的投机解码(Speculative Decoding)推理优化框架。该框架能在保持生成质量的前提下,将单用户生成速度提升 60-85%,目前已部署于 DeepSeek-V4 的生产推理系统。

背景:投机解码的原理与瓶颈

大语言模型采用自回归方式逐 token 生成,每个新 token 都需要一次完整的前向传播,导致推理延迟与输出长度成正比。投机解码(Speculative Decoding)的核心思想是:用一个轻量级的草稿模型快速生成候选 token 序列,再由目标模型并行验证整个块。如果验证通过,全部接受;如果失败,则回退到第一个失败位置。

关键性能公式:每个 token 的平均延迟 L = (T_draft + T_verify) / τ,其中 τ 是每次循环被接受的 token 数量。提升推理速度的路径有三个:

  • 减少草稿时间 (T_draft)——更快的草案器
  • 提高接受率 (τ)——更精准的候选序列
  • 减少无效验证时间 (T_verify)——避免浪费计算在后缀 token 上

DSpark 在这三条路径上同时取得了突破。

创新一:半自回归架构(Semi-Autoregressive Generation)

现有的草稿模型分为两类,各有局限性:

自回归草案器(如 Eagle3):token 逐一生成,依赖关系建模好,接受率(τ)高。但草稿时间与块大小成正比(T_draft ∝ γ),因此只能使用小尺寸的候选块,限制了吞吐量。

并行草案器(如 DFlash):一次前向传播生成所有候选 token,草稿时间几乎与块大小无关,生成速度快。但 token 之间缺乏依赖关系,后缀位置的接受率随长度增加而快速衰减。

DSpark 的解决方案:保持并行骨干网络的高效计算,仅附加一个轻量级的序列化输出头(sequential block)来注入局部过渡信息。这一设计同时获得了两个世界的优势——既保留了并行模型的高生成速度,又通过序列化头显著缓解了后缀衰减问题。

创新二:置信度调度验证(Confidence-Scheduled Verification)

传统投机解码方案对所有候选 token 进行无差别验证。这意味着大量批量计算资源被浪费在那些几乎肯定会被拒绝的后缀 token 上——尤其是在高负载场景下,这种浪费尤为致命。

DSpark 设计了一套智能验证机制:

  • 置信度头(confidence head):估算每个位置候选 token 的存活概率
  • 硬件感知调度器(hardware-aware scheduler):根据实时引擎吞吐量画像,将稀缺的验证预算只分配给预期回报最高的 token
  • 自适应行为:低负载时近乎免费地验证长块;高负载时主动截断低置信度的后缀 token,避免资源浪费

这一机制使得 DSpark 在服务系统负载波动时依然能够维持稳健的吞吐量。

实验效果

离线基准测试(Qwen3-4B/8B/14B):

  • 相比 SOTA 自回归草案器 Eagle3:接受长度提升 26.7-30.9%
  • 相比 SOTA 并行草案器 DFlash:接受长度提升 16.3-18.4%

在线生产部署(DeepSeek-V4 服务系统,真实用户流量):

  • V4-Flash:每用户生成速度提升 60-85%
  • V4-Pro:每用户生成速度提升 57-78%
  • 在严格的交互延迟约束下(Flash 120 TPS,Pro 50 TPS),基线容量出现严重退化,而 DSpark 维持了稳健的吞吐量
  • 实现了此前不可达的性能层级,推动了服务系统的 Pareto 前沿

这些结果的意义在于:DSpark 不仅在离线基准上超越了所有现有方案,更重要的是在实际生产环境中经受住了考验。

开源信息

DeepSeek 已全面开源 DSpark 的相关资源:

  • DSpark 检查点:适用于 DeepSeek-V4-Flash 和 V4-Pro 预览版本
  • DeepSpec 仓库:一个算法驱动的投机解码训练框架,集成了 Eagle3、DFlash 和 DSpark 三种方案,便于研究社区复现和进一步探索

项目地址:https://github.com/deepseek-ai/DeepSpec

原文出处:DSpark: Confidence-Scheduled Speculative Decoding with Semi-Autoregressive Generation,Xin Cheng, Xingkai Yu et al.(DeepSeek-AI + 北京大学)

🧑‍💻

zhirenhun

一个热爱技术的程序员,喜欢分享前沿AI知识和开发经验。

DeepSeek speculative decoding LLM推理优化 投机解码
← 上一篇
Zig 的新 bitCast 语义与 LLVM 后端改进
下一篇 →
使用完全生命周期控制运行隔离沙箱:AWS Lambda 推出 MicroVMs

📌 相关推荐

结构化输出 vs 工具使用 vs 预填充:2026年从Claude获取JSON
2026/7/11
在AWS Bedrock上构建生产级AI代理——值得关注的架构与代码决策
2026/7/11
如何为你的AI代理构建评估框架(确保其在生产环境中不崩溃)
2026/7/11
← 返回文章列表