FlexPrefill

长序列推理加速,稀疏注意力省算力不掉精度

FlexPrefill 是一个针对大语言模型长序列推理的稀疏注意力机制实现,源自 ICLR 2025 Oral 论文。它解决的是长上下文场景下标准注意力计算开销随序列长度平方增长的问题,通过上下文感知的方式动态选择重要的键值对,在保持模型精度的同时显著降低预填充阶段的计算量。核心能力包括:基于查询和键值交互的稀疏模式识别、自适应稀疏率调整、以及高效的 CUDA 内核实现,支持无缝集成到现有 Transformer 架构中。项目提供完整的训练和推理代码,以及论文复现脚本,适合研究者和工程师探索长序列模型的高效部署。

开源 free 开源模型
访问官网 ↗ GitHub ↗ 文档 ↗
GitHub 星标 ★ 172
维护状态 低维护
是否开源 是
定价模式 free

项目数据

分类开源模型
开发团队ByteDance-Seed
所属国家
官网地址
定价模式free
价格说明开源项目,Apache-2.0许可证,可自由使用,无付费版本。
访问状态
是否开源是
开源协议Apache-2.0
主要语言Python
技术栈/模型large-language-models,natural-language-processing,research,sparse-attention
GitHub 星标★ 172
30天Star增速
HF 下载量
上线时间2025-02-18 00:00:00
最近更新2026-09-02 00:00:00
维护状态低维护
中文支持
访问方式
移动端支持
综合评分
收录时间2026-08-09
浏览次数2

使用教程

核心亮点

  • 提出上下文感知的稀疏注意力机制,能根据输入动态选择关键token,比静态稀疏更精准
  • 在长序列预填充阶段显著降低计算量,实测加速比明显,且保持模型精度
  • 代码结构清晰,提供论文复现脚本和示例,便于研究者快速上手

不足之处

  • 项目处于早期阶段,文档和社区支持待观察
  • 稀疏注意力对硬件和算子优化依赖较高,通用性可能受限

适用场景

  • 长文档理解与摘要生成
  • 多轮对话中的长上下文建模
  • 代码库级代码生成与理解

替代项目

FlashAttention、Sparse Transformer、Longformer

项目介绍

FlexPrefill 是基础设施领域的开源项目,由 ByteDance-Seed 开发,2025 年首次发布。

它收录于基础设施分类,该分类目前共有 1,130 个项目,GitHub 星标数为 172。

项目已超过三个月没有代码更新,维护节奏明显放缓,最近一次代码更新于 2026-09-02。Apache-2.0许可证,可自由使用,无付费版本。

它主要面向的使用场景是:长文档理解与摘要生成。同类可对比的替代方案包括 FlashAttention、Sparse Transformer、Longformer。

上一篇:scalax

下一篇:dsc

同类项目推荐

freebuff-proxy 开源

聚合多账号,一键接入 OpenAI 兼容 API,轻松管理会话。

Multi-account OpenAI-compatible gateway for coding models. Pooled keys, session life···

★ 213 2026-08-20
microduck 开源

用 Rust 造一只会走路的桌面小鸭,快速上手双足机器人。

A Tiny biped duck robot

★ 8684 2026-09-10
soperator 开源

用 Kubernetes 原生方式运行 Slurm,简化 HPC 集群管理。

Run Slurm in Kubernetes

★ 439 2026-08-10
ollama 开源

一条命令本地跑起大模型,免费、私密、不卡顿

Get up and running with Kimi, GLM, MiniMax, DeepSeek, gpt-oss, Qwen, Gemma and other···

★ 181484 2026-08-09