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

EAGLE 3.1:修复LLM推理中注意力漂移的推测解码算法

EAGLE团队、vLLM团队和TorchSpec团队联合发布了EAGLE 3.1,旨在解决生产环境中推测解码的不稳定性。该算法通过FC归一化和归一化后隐藏状态反馈两大架构改进,有效应对注意力漂移问题。在长上下文任务中,EAGLE 3.1的接受长度比EAGLE 3提升高达2倍;在Kimi K2.6模型上的基准测试显示,并发数为1时每用户输出吞吐量提升2.03倍。EAGLE 3.1完全向后兼容,已合并至vLLM主线,并将随v0.22.0版本发布。

来源MarkTechPost作者: Michal Sutter
EAGLE 3.1:修复LLM推理中注意力漂移的推测解码算法
报告错误

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

查看更正说明

本文仅提供摘要

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

展开要点与分析

文章情报

工程师进阶

要点

  • EAGLE 3.1修复了推测解码中的注意力漂移问题,即起草模型在深度推测时注意力从原始上下文偏移到自身生成内容。
  • 两项架构改进:FC归一化稳定隐藏状态,以及将归一化后的隐藏状态反馈到下一步,使起草模型更稳定。
  • 长上下文任务中接受长度提升2倍,Kimi K2.6模型吞吐量提升2.03倍(并发数1)。
  • 完全向后兼容EAGLE 3检查点,已集成至vLLM,计划在v0.22.0中发布。

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