arXiv发布DScout:基于因式分解指导的扩散草稿推测解码技术
摘要:arXiv发布了一种名为DScout的新方法,通过因式分解指导的扩散草稿推测解码技术,显著提升大语言模型(LLM)的推理速度。DScout通过将位置损失分解为包含、权重和预测损失,优化训练过程,并利用吞吐量与每轮延迟的比率指导推理配置选择。在Qwen3-4B和Qwen3-8B模型上的实验表明,DScout相较于自回归解码分别实现了3.84倍和4.14倍的速度提升,超越了现有最先进基线方法。
技术机制剖析
DScout的核心创新在于其因式分解指导的训练与推理方法:
-
训练阶段:
- 位置损失分解:将传统的逐位置损失分解为包含损失、权重损失和预测损失。这种分解方式能够更精细地指导模型学习,提升对前缀接受度的对齐效果,从而提高平均接受token数(MAT)。
- 混合视野训练与区块内掩码:结合混合视野训练和区块内掩码技术,进一步优化模型对长程依赖关系的捕捉能力。
- 接受权重损失:通过接受权重损失调整训练目标,使其与实际推理中的前缀接受机制更加一致。
-
推理阶段:
- 吞吐量优化:将吞吐量定义为MAT与每轮延迟的比率,指导目标条件深度、草稿深度和推理区块大小的选择,以找到针对不同工作负载的高吞吐量配置。
- 配置策略:DScout的推理配置策略在多个模型和并发设置中提供了额外的平均速度提升,最高可达9.75%。
工程权衡与实测表现
-
优势:
- DScout在Qwen3-4B和Qwen3-8B模型上实现了显著的速度提升,超越了现有的推测解码方法。
- 该方法在处理长序列时表现出色,能够有效减少推理延迟。
-
劣势:
- 训练过程中引入的因式分解和复杂损失函数可能增加训练时间。
- 推理配置策略的优化需要针对不同工作负载进行微调,增加了部署复杂性。
-
实测表现:
- 在六个基准测试中,DScout在Qwen3-4B和Qwen3-8B模型上分别实现了3.84倍和4.14倍的速度提升。
- 与现有最先进基线方法相比,DScout的平均速度提升提高了9.1%和7.1%。
开发者落地与部署建议
- 硬件需求:DScout对硬件的要求较高,建议在配备高性能GPU的环境中部署。
- 模型适配:开发者需要根据具体应用场景对DScout的推理配置进行优化,以获得最佳性能。
- 训练成本:由于训练过程的复杂性,建议在资源充足的情况下进行模型训练。
- 应用场景:DScout适用于对推理速度要求较高的应用场景,如实时对话系统、在线翻译和大规模数据分析等。
结论
DScout通过创新的因式分解指导方法,显著提升了LLM的推理速度,为大语言模型的实际应用提供了新的可能性。
—— 完 ——消息来源:ArXiv Machine Learning (cs.LG) (2026-10-21)
主题标签: #DScout #扩散模型 #推测解码 #大语言模型 #推理加速
编辑部与事实核查说明:本篇专刊由智客前沿资讯管线根据官方技术公告、学术论文及开源工程文档整理编译,经算法实体核验与人工编辑审校后发布。若发现技术事实纰漏,欢迎依据勘误方针或一键向编辑部发送勘误反馈。
社区整体评论区