SparseEngine:稀疏优先推理引擎
摘要
SparseEngine 是一个从零构建的稀疏优先推理引擎,其共享的生命周期契约使每种稀疏注意力方法都能自主控制其 KV 表示与计算。与 vLLM 相比,结合 KV 淘汰可实现超过 10 倍的吞吐量提升,并在 Agent 基准测试中带来超过 2 倍的端到端加速。
查看缓存全文
缓存时间: 2026/10/09 06:24
论文页面 - SparseEngine:稀疏优先推理引擎
来源:https://huggingface.co/papers/2609.39068
摘要
长上下文 LLM agent 会积累大量交互历史,导致 KV 缓存内存和注意力计算压力剧增。虽然稀疏注意力能够降低这些开销,但异构的缓存表示和工作流阻碍了其与现有推理引擎的集成,而先前的稀疏服务抽象仅支持特定的布局或工作流。我们提出了 SparseEngine,一个从零构建的稀疏优先推理引擎,其共享的生命周期契约让每种方法都可以控制自身的 KV 表示与计算方式,同时与通用的服务基础设施协调状态转换。SparseEngine 支持四大类别的 15 种方法,并通过 ChainCache 实现跨请求状态管理——该机制可以从保留的历史中恢复 KV 驱逐方法,同时提供可控的前缀缓存裁剪(Prefix Cache Pruning),在从选定历史区域中移除 KV 的同时保持逻辑前缀匹配。在保持方法质量的前提下,SparseEngine 使用 KV 驱逐时吞吐量提升超过 10 倍,在相同并发下解码速度比 vLLM 快 2.5 倍以上,在 agent 基准测试中端到端加速超过 2 倍。代码已开源:https://github.com/CURRENTF/SparseEngine。
查看 arXiv 页面 (https://arxiv.org/abs/2609.39068) 查看 PDF (https://arxiv.org/pdf/2609.39068) GitHub 79 (https://github.com/CURRENTF/SparseEngine) 添加到收藏集 (https://huggingface.co/login?next=%2Fpapers%2F2609.39068)
引用此论文的模型:0
没有模型关联此论文
在模型 README.md 中引用 arxiv.org/abs/2609.39068,即可从此页面链接过去。
引用此论文的数据集:0
没有数据集关联此论文
在数据集 README.md 中引用 arxiv.org/abs/2609.39068,即可从此页面链接过去。
引用此论文的 Spaces:0
没有 Space 关联此论文
在 Space README.md 中引用 arxiv.org/abs/2609.39068,即可从此页面链接过去。
包含此论文的收藏集:0
没有收藏集包含此论文
将此论文添加到收藏集 (https://huggingface.co/new-collection) 以从此页面链接过去。
相似文章
@samsja19:我们非常看好稀疏注意力 + HiSparse,并且一直与 vLLM 团队合作
HiSparse 和稀疏注意力技术正被集成到 vLLM 中,以减少内存使用并提高并发性,从而提升强化学习的推理吞吐量。
SparDA:用于高效长上下文 LLM 推理的稀疏解耦注意力
SparDA 提出了一种解耦稀疏注意力架构,通过添加轻量级"Forecast"投影来预测未来的 KV 缓存需求,从而实现从 CPU 到 GPU 的预取(lookahead prefetching),并降低选择开销。在基于稀疏预训练的 8B 模型上,其 prefill 速度最高可提升 1.25×,decode 速度最高可提升 1.7×,相比非 offload 基线,decode 吞吐量最高可提升 5.3×。
@akshay_pachaar:每个推理引擎都犯了同样的错误。像 vLLM 或 SGLang 这样的推理引擎,是位于你的请求与模型权重之间的软件……
LMCache 是一个开源的 KV 缓存管理层,它将缓存 I/O 与计算分离,可插入 vLLM、SGLang 和 TensorRT-LLM,通过并行化缓存查找和共享 GPU 内存,实现高达 14 倍的首 token 延迟降低和 4 倍的解码加速。
OasisKV:利用前瞻稀疏预取将解码中KV缓存扩展到HBM之外
OasisKV是一个以内存为中心的LLM推理系统,通过使用来自推测解码的前瞻令牌预取稀疏且重要的KV块,将完整KV缓存存储与HBM解耦,与密集vLLM相比实现了高达2.1倍的吞吐量提升,且精度损失极小。
利用指数衰减记忆增强注意力提升查询感知的KV稀疏性
本文探讨了RAT+中的指数衰减记忆模块如何提升长上下文语言模型的查询感知稀疏推理方法,在针尖干草垛任务中,跨多种稀疏预算展示了一致的准确率提升。