Skip to content
简体中文

切换主题

// ⚡ 推理加速与服务

Medusa 多解码头

在模型上加几个额外的解码头,自己给自己做推测解码。

Medusa 2024 高级 扩展

通俗解释

不需要单独的草稿模型:额外的头同时预测后面第 2、3…个 token,用树状注意力一次验证多个候选组合。

用在哪里:推理加速框架

时间轴 · 前置 → 当前 → 后继

2022 推测解码2024 Medusa 多解码头

01原始论文

Medusa: Simple LLM Inference Acceleration Framework with Multiple Decoding Heads

Tianle Cai, Yuhong Li, Zhengyang Geng, et al. · 2024

arXiv ↗ ✓ 已核实 · arXiv API

02知识位置

左边先学;右边是学完它之后可以学的。

先学

Medusa 多解码头

学完可以学

  • 图谱里暂无后继

相关:EAGLE多 token 预测

推测解码EAGLE多 token 预测 Medusa 多解码头

左:前置 · 右:后继 · 虚线:相关。看完整关系图 →

03学习资料

官方文档、图解、课程和参考实现,链接都核对过。

04在课程里哪里学

给 AI 读 / 数据说明

数据更新于:AI 算法图谱与课程体系(北京时间)