DSpark: Confidence-Scheduled Speculative Decoding with Semi-Autoregressive Generation
DSpark:半自回归生成与置信调度推测解码
🧭 推测解码 · 半自回归 · 置信调度 ★ 深度解读
Peking University · DeepSeek-AI
半自回归架构缓解并行预测因果衰减,置信调度动态剪枝低信心后缀,系统吞吐与交互速度显著提升
📌 工作参考价值(4 点)
- 半自回归草稿架构:在并行骨干后追加轻量马尔可夫/RNN头,仅引入极低延迟开销(<2%),大幅缓解并行预测的多模态冲突和接受率衰减。类似思路可用于电商搜索推荐中的对话助手、自动回复生成等低延迟场景,在保持并行速度的同时提升生成连贯性。
- 置信度驱动的动态调度:通过置信头估计前缀生存概率,结合硬件吞吐曲线动态选择验证长度,让高并发下自动回收低信心token的计算资源。在推荐系统的消息推送、搜索词改写等LLM调用中,可借鉴此机制,根据服务负载动态调整生成预算,避免盲目固定长度造成的容量浪费。
- 异步调度与核函数解耦:生产部署中采用历史预测近似当前容量,延迟调度与CUDA图回放兼容,同时通过展平序列与标记张量实现变长验证的高效执行。工程上为推荐系统的LLM推理服务(如推荐理由生成、用户查询补全)提供了可行的低开销动态批处理方案。
- 顺序温度缩放校准:对累积置信度进行逐位置校准,使生存概率估计更准确,支撑硬件感知调度。该校准方法可直接复用于需要可靠概率输出的在线调度场景,如流量分发、个性化推送的置信过滤。