FlashDecoding

英文:FlashDecoding 分类:training

基本释义

加速长上下文解码(生成)阶段注意力计算的技术(FlashAttention 的推理优化版):将长序列的键值分块并行计算部分注意力后归约,大幅降低解码延迟,配合 FlashAttention 2 支持百万 token 级推理。

技术要点

问题:解码时逐 token 生成,KV 缓存极长导致注意力计算成为瓶颈(无法利用数据并行并行度);方法:按序列分块→各块并行计算注意力(部分 logits/max/sum)→归约合并;效果:长上下文解码吞吐成倍提升、显存友好(不物化完整注意力矩阵);生态:集成于 vLLM/SGLang 等推理引擎;变体:FlashDecoding++(多级归约)。

应用场景

用于长上下文推理(128K-1M token)、大模型服务优化(吞吐提升)、推理引擎部署(vLLM 标配)与生成延迟敏感场景(聊天/Agent 实时响应)。

返回AI百科目录 · AI最前沿首页

相关词条:微调 · 监督微调 · 指令微调 · 余弦衰减 · 张量并行度