DeepSeek:最新DSpark论文解读

DeepSeek:最新DSpark论文解读

12分钟 ·
播放数240
·
评论数4

6月27日, DeepSeek联合北京大学发表了名为《DSpark: Confidence-Scheduled Speculative Decoding with Semi-Autoregressive Generation》(DSpark:基于置信度调度的推测解码与半自回归生成)的论文,,正式提出了名为 DSpark 的推测解码框架,在业内引起了广泛关注。

当前大语言模型推理的核心瓶颈不在于浮点算力,而在于显存带宽推测解码是业界应对这一瓶颈的主流方案:由一个轻量级草稿模型先行生成候选序列,再由大模型进行一次性验证。然而,该范式长期面临两个结构性困境:其一,并行草稿生成中,越靠后的 token 接受率衰减严重;其二,固定长度的草稿验证在推理负载波动时会造成大量计算资源浪费。

DSpark 针对上述问题提出了两项核心创新。

第一,半自回归架构。该架构在并行骨干网络之上叠加了一个极轻量的马尔可夫头,仅依赖前一个 token 对当前 token 进行概率重标定。通过低秩分解将额外计算开销压缩至 0.2%—1.3%,却使草稿有效接受长度最高提升 30%,在不牺牲并行速度的前提下显著改善了后缀衰减问题。

第二,置信度调度验证机制。草稿模型在生成候选序列的同时输出每个 token 的存活概率估计,系统内置的硬件感知前缀调度器根据置信度与 GPU 吞吐曲线动态裁定验证范围,对低置信度 token 执行早停,避免无效计算。配合在线草稿器校准技术,将预期校准误差从 8% 压缩至约 1%。

在线部署实测中,DSpark 实现了单用户生成速度提升 60%—85%,高并发场景下系统有效吞吐量提升 4 倍,两层 DSpark 即可匹敌此前五层架构性能。目前 DeepSeek 已将包含 Eagle3、DFlash 与 DSpark 在内的 DeepSpec 全栈训练库开源,供学术界与工业界共同研究使用。该项目整体体现了算法设计、调度策略与硬件特性深度融合的工程实践水平。

论文地址:github.com

展开Show Notes
牡龙
牡龙
8天前
04:12 听起来有点像是扩散模型?不管对不对先出一版,再验证再生成?
牡龙:听完感觉有点像 diffusion gemma 的架构?
RayHu
:
投机解码中,系统会先用一个小参数的“草稿模型”快速猜想(并行生成)出一串候选的 Token(先出一版),然后交由大模型一次性进行并行评分(再验证)。大模型会接受预测正确的 Token,一旦发现错误就立即打断,丢弃后面的内容并由大模型自己生成正确的 Token(再生成)。但这种模式还是半自回归的,就是上文影响下文。 DiffusionGemma是真正的纯扩散语言模型。它完全抛弃了从左到右逐字吐出的自回归范式,采用从随机占位符开始、双向注意力多轮去噪的生成方式。 所以说 DSpark是一个为主流大模型外挂的推理加速框架。其本质依然在服务于从左到右的自回归模型,只是通过“并行草稿骨干 + 轻量马尔可夫串行头(半自回归) + 动态置信度调度验证”的机制,让原本的生成速度变得更快。 如果要说 DSpark 和扩散模型有什么渊源,那就是 DSpark 继承了 DeepSeek 之前发表的 DFlash框架的并行骨干,而 DFlash 确实在生成草稿的局部环节借用了“块扩散(Block Diffusion)”技术。但这仅仅是草稿阶段的一个技术借用,DSpark 的整体系统依然是投机解码。
3条回复