跳到正文
Rohan Paul· @rohanpaul_ai · X·· 5 天前AI 评分58
AI 导读

Rohan Paul 分享了 EAGLE-4 投机解码插件的实测表现,其在 27B 模型单次请求下达到 863.8 tokens/s 的解码速度。该工具采用自适应草稿结构和廉价的草稿 LM 头,通过草稿与验证循环实现约 3 倍的加速。

正文 · 原文

Some of the most useful inference work right now is happening in the speculative decoding component.

It has become one of the most dependable ways to get more tokens per second out of a model you’ve already chosen.

EAGLE-4 decodes 863.8 tokens/s on a 27B model on a single request, 1.28X DFlash 2 and 1.40X DSpark.

EAGLE-4 is a speed-up add-on for running AI chatbots: a tiny helper model guesses the next few words, and the big model checks all those guesses at once instead of writing one word at a time. The answers come out identical, just about 3× faster on the tested setup.

EAGLE-4 keeps the draft-and-verify loop and puts its work into the drafting side, with an adaptive draft structure and a cheaper draft LM head.

The loop:

  • A small draft model proposes several tokens ahead
  • The target model checks all of them in one forward pass
  • Matching tokens are kept, and the first mismatch is replaced with the target’s own token
  • Repeat from there

来源:Rohan Paul · x.com