大语言模型推理工程旨在降低延迟、提升吞吐、减少内存。本文推荐5个实践项目:从零构建迷你vLLM、用Triton编写FlashAttention内核、搭建量化实验、实现投机解码、构建生产级推理服务,帮助深入理解KV缓存、连续批处理、量化等优化技术。
DeepSeek-V4.1 的 DSpark 是在主干模型后挂载的三层草稿模块,一次前向生成 5 个草稿 token:Markov head 补充 token 间依赖,confidence head 估计接受概率,调度器按负载决定送验数量。文章讲解投机解码的接受规则、并行草稿越靠后越不准的原因、阈值随负载变化,以及按置信度截断保持输出分布的条件。
DSpark是一种投机解码技术,通过并行草稿与轻量级顺序组件结合,提升LLM生成速度。在llama.cpp中测试Qwen3-8B,启用DSpark后生成速度从95.0提升至124.9 tokens/s,约31.5%加速。尽管MTP更通用,DSpark在草稿质量上占优,但模型支持有限,适合实验。
mlx-dspack项目在Mac上通过投机解码技术,将Qwen3.8-27B模型推理速度提升至3倍,8-bit量化下性能优于4-bit。项目基于DeepSeek的DSpark和z-lab的DFlash,支持自动调参优化草稿长度。实测显示8-bit加投机解码比4-bit普通解码快39%,但加速效果受硬件、量化方式和MLX版本影响,长上下文和内存限制仍是瓶颈。
文章讨论了PD分离与引擎功能的适配,介绍了投机解码及其在大型自回归模型中的应用。通过引入小模型进行多次自回归,提升推理效率。探讨了Eagle 3的实现细节及其对推理性能的影响,得出礼貌性输入不影响推理性能的结论。
自2023年9月推出以来,Workers AI团队致力于提升平台质量,推出了快速推理的投机解码、异步批处理API和扩展的LoRA支持,显著提升了推理速度和用户体验。
翼支付的研究成果《Falcon》提出了一种增强的半自回归投机解码框架,显著提升了大型语言模型的推理速度,达到了2.91-3.51倍的加速比。该方法通过改进的投机采样和专门设计的解码树,提高了模型的并行性和输出质量,已在多个实际业务中成功应用。
本研究提出了一种新的动态草稿长度策略SVIP,旨在解决投机解码中固定草稿长度的问题。SVIP根据草稿词元分布的熵自适应调整草稿长度,实验结果表明其在主要基准测试中可实现最高20%的加速效果。
本文介绍了一种通过知识蒸馏和投机解码技术(DistillSpec)加速大型语言模型推理的方法。该方法在多项基准测试中实现了10-45%的加速,并有效降低了解码延迟。结合轻量级草稿模型和新采样机制,显著提高了解码效率,实验结果显示在多种模型架构中优于现有方法。
美团技术团队在ACL 2024收录的4篇论文涵盖了训练成本优化、投机解码、代码生成优化、指令微调等技术领域。这些论文是与高校、科研机构合作的成果,介绍了早期退出投机解码、图结构投机解码、DolphCoder和指令微调等方法。这些方法在提高文本生成速度和代码生成性能方面取得了显著成果。
本文提出分阶段投机性解码算法,将投机批量重组为树并加入第二阶段解码,以加速小批量设备端LLM推断。在762M参数GPT-2-L上,解码延迟降低3.16倍且不损输出质量。文章还综述了SpecInfer、Medusa、Speculative Streaming等投机解码进展,这些方法通过预测推断、令牌树验证或多解码头并行预测实现2至3.6倍加速。
该文综述大语言模型并行解码与推理加速方法,包括SPEED、PaLM 2、BiLD、Paraformer、投机解码等,通过并行生成未来令牌、双模型协同或轻量适配器,在保持质量的同时显著降低延迟,最高提速10倍以上。
该文综述多种大语言模型推理加速方法,包括Medusa(通过额外解码头并行预测多标记,提速2.2至3.6倍)、级联投机解码、多候选验证、HydraSum多解码器、分块并行解码、自推测解码及Hydragen共享前缀注意力等,分别从草稿模型、并行验证和注意力计算等角度减少解码步骤,提升吞吐量。
完成下面两步后,将自动完成登录并继续当前操作。