跳到主要内容
AI News HubLIVE
更多
仅摘要

硬件感知的动态投机解码(DSD)| Cohere

Cohere提出了动态投机解码(DSD),根据硬件约束动态调整草稿令牌数量,以提升LLM推理速度。该方法解决了固定K投机解码在高批次大小时性能下降的问题,并在密集和MoE模型上验证了有效性,同时兼容vLLM的异步调度和CUDA图优化。

硬件感知的动态投机解码(DSD)| Cohere
报告错误

纠错通道尚未开通,可先复制下方文章信息留存。

查看更正说明

本文仅提供摘要

根据来源内容权限,本站仅提供编辑摘要、要点和原文链接。

展开要点与分析

文章情报

工程师进阶

要点

  • 动态投机解码(DSD)根据批处理大小动态调整草稿令牌数,优化计算与内存带宽的利用。
  • DSD在密集模型上单调递减K值,在MoE模型上非单调,中批时K值最高。
  • DSD在批大小128和256时比固定K投机解码快约23%,且不退化。
  • 已贡献至vLLM,兼容异步调度和全CUDA图。

要点与分析由自动化流程生成,可能有误,请结合原始来源核实。