免费POC, 零成本试错
FDE知识库

FDE知识库

学习大模型的前沿技术与行业落地应用


收藏

首 token 延迟降 3.25 倍!小红书联合北大、上交提出 HYPIC,让混合注意力大模型用上「位置无关缓存」

发布日期:2026-07-16 18:06:28 浏览次数: 1775
作者:小红书技术REDtech

微信搜一搜,关注“小红书技术REDtech”

推荐语

小红书联合北大、上交提出HYPIC,首次在混合注意力大模型上实现位置无关缓存,将首token延迟平均降低3.25倍,显著提升大模型服务效率。

核心内容:
1. RAG与Agent应用中长prompt带来的成本与延迟挑战
2. 位置无关缓存与混合注意力模型难以兼容的技术瓶颈
3. HYPIC系统如何实现兼容并带来显著性能提升

杨芳贤
53AI创始人/腾讯云(TVP)最具价值专家


引言:随着 RAG(检索增强生成)和 Agent 成为大模型服务的主流形态,一条请求的 prompt 往往由几十到上百个独立文本片段拼接而成,长度动辄十万 token,prefill(预填充)阶段的算力开销成为服务成本与延迟的最大来源。业界为此发展出两条路线:一边是「位置无关缓存」(Position-Independent Caching, PIC),让共享片段的 KV 缓存可以在任意位置复用;另一边是「混合注意力」模型,用线性注意力linear attention替换大部分全注意力(full attention)层来压低算力复杂度。问题在于——两者无法共存:PIC 依赖的是逐 token 的 KV 缓存原语,而线性注意力层只暴露一个「整段压缩后的循环状态」,没有任何逐 token 的抓手。为此,小红书大模型推理团队联合北京大学、上海交通大学提出 HYPIC,这是首个在混合注意力大模型上实现位置无关缓存的服务系统。在 4 个生产级混合注意力模型、5 个工作负载上,HYPIC 将首 token 延迟(TTFT)平均降低 3.25 倍、同 SLO 下可持续 QPS 提升 1.66 倍,而任务质量与「完全重算」仅相差1.71 分

论文标题: 

HYPIC: Accelerating Hybrid-Attention LLM Serving with Position-Independent Caching

论文链接: 

https://arxiv.org/abs/2607.01299

代码链接:

https://github.com/redai-infra/HYPIC

关键词:

大模型推理,位置无关缓存,混合注意力,线性注意力,KV Cache

大模型服务正从单轮聊天,转向检索增强问答、多文档摘要、长程 Agent 等新形态。这类负载有一个共同特征:prompt 由若干语义独立的片段拼装而成(检索到的文档、技能说明、记忆文件、历史轮次等),最终组装成数万乃至数十万 token 的超长上下文。

在这样的长度下,prefill 主导了单请求的算力开销,成为服务商最显著的成本来源之一;更糟的是,一旦发生缓存未命中(cache miss),尾部 TTFT 可高达数十秒,直接损害交互体验。



为压低这项成本,许多工作提出了位置无关缓存(PIC):放开严格前缀约束,让每个语义独立的片段只缓存一次、可拼接到任意前缀之后,正好匹配 RAG / Agent 的 prompt 组装方式。现有 PIC 方法尽管策略各异,本质都归结为两个原语——沿 token 轴直接拼接的splice,以及重算少量 token 以恢复跨段上下文的 correction

与此同时,模型架构也在变。线性注意力把注意力的二次复杂度降到线性,并将无界的 KV 历史压缩成一个固定大小的循环状态。近来的生产模型(如 MiniMax-M1、Ring-2.5、Qwen3.5、Kimi-Linear)并不完全抛弃全注意力,而是把 75% 以上的层线性化、保留少量全注意力层,形成混合注意力(Hybrid-Attention)这一主流设计。以 Qwen3.5-35B-A3B 为例,40 层里有 30 层是线性层。

冲突就此产生:现有 PIC 操作的是逐 token 的 KV 缓存,而线性注意力层只暴露一个 per-request 的循环状态——没有逐 token 的抓手可供 splice 或 correction。结果是,混合模型里的大部分层都落在了现有 PIC 的能力范围之外。在此之前,没有任何系统能为混合注意力大模型提供 PIC。

HYPIC 的核心思想,是针对混合注意力模型里两类层「分而治之」,再用一层系统级并行把冷请求也一并加速。它由三个环环相扣的机制组成。


2.1 线性层:缓存「转移算子」,实现常数时间的状态组合


对线性注意力层,最直接的 PIC 稻草人方案是把两段各自的「零初值末状态」直接相加。在朴素线性注意力下这恰好成立,但在带衰减、门控、delta 擦除的进阶线性注意力(RetNet、Mamba2、GLA、DeltaNet、GDN、KDA 等)下就会失效。

我们发现,真正被漏掉的是一个关键量——段累积转移算子(segment-cumulative transition operator)T_C,即一段内所有 token 转移矩阵的连乘。真实的末状态应为 S(C1·C2) = T(C2)·S(C1) + S(C2),朴素相加恰恰丢掉了 T(C2) 这一项,导致结构性误差(实测 RetNet 慢衰减头在 256 token 时误差已达状态范数的 22%)。

关键洞察是:T_C 和零初值末状态 S(C|0) 都只由片段内部的 token 决定,与前缀无关。因此 HYPIC 在片段首次 prefill 时,把二元组 (T_C, S(C|0)) 一并缓存下来;复用时按组合律左乘 T_C、再相加,即可在常数时间内近乎精确地还原任意前缀下的末状态,且天然覆盖全部线性注意力家族。实测在 Qwen3.5-35B-A3B 上,组合后的状态在第 0 层与完全重算相差仅 6×10⁻⁵——落在 FP16 噪声内。

对于带因果卷积、带 RoPE 的变体,HYPIC 分别用「卷积状态热身」和「状态重旋转」两个补丁做了严格对齐。


2.2 全注意力层:用「缝合窗口」修复跨段注意力



混合注意力模型中少数的全注意力层仍需要 PIC,但以往工作的选择性重算无法直接迁移——因为下方的线性层只保留末状态,非末尾 token 无法向上穿过全注意力层。两种退路(逐 token 存循环状态、或从零重新递推)在存储或算力上都不可接受。

我们的观察是:KV 拼接后的偏差,高度集中在每个复用片段的开头,其余部分几乎不受影响(这与全注意力模型里的 attention sink 现象一致,并在混合注意力模型里同样成立)。

据此,HYPIC 为每个内部片段的开头 w 个 token构造一个缝合窗口(seam window):缓存时这几个 token 不入缓存,复用时在完整前缀下重算它们,从而修复跨段注意力。默认 w=8——由于实际片段长度通常大于 512 token,缝合窗口只占极小一部分,重算开销可控。为支持段首 KV 重算,linear-attention 层在复用时需即时算出缝合窗口自身的 T 和 S,一边推进 running state,一边把每个 seam token 的逐层输出前传给上层的全注意力层。


2.3 段并行:把「长冷请求」变成可加速的负载



缓存未命中不可避免(语料持续更新、低频文档被淘汰),而长冷请求正是尾延迟的主因。现有系统仍把整条 prompt 当作一个整体、在单实例上串行 prefill 所有冷片段,TTFT 随 O(n·|C|) 增长;张量并行等实例内并行虽能加速单次前向,但扩展性有限(100k token 请求在 TP-8 下仍需 17.7 秒)。

但 PIC 已经让每个片段「自包含」——其 prefill 结果只由内部 token 决定。HYPIC 顺势提出段并行(Segment Parallelism)这一实例间并行方案:Router 探测每段命中状态,把冷片段并行分发给多个 scatter worker,再由一个 combine worker 把各段结果组装成完整运行状态。它把冷 prefill 从 O(n·|C|) 降到 O(⌈n/m⌉·|C|+c)。

为把这条路径跑满,HYPIC 还做了两件事:用LPT(最长处理时间优先)贪心策略均衡各 worker 负载;把每段的计算与传输流水线重叠,避免 combine worker 被同步的传输突发拖住。段并行与实例内并行(TP/DP/SP)作用在正交的轴上,可无缝叠加。

我们在 SGLang 上实现了 HYPIC(约 14k 行 Python / Triton),在 8×H20 节点上,选取 4 个生产级混合注意力模型(Ring-mini/flash-linear-2.0、Qwen3.5-35B-A3B / 122B-A10B)、4 个公开数据集(HotpotQA、TriviaQA、MultiNews、GovReport)与 1 条生产 RAG trace 进行评测。


端到端延迟与质量:全量预热后,HYPIC 相比 Prefix Cache 将 p50 TTFT 平均降低3.25 倍(各模型 2.77×–4.05×);而任务质量几乎无损——16 个「模型×数据集」单元平均只落后完全重算 1.71 分,在 Qwen3.5-35B 上甚至反超 0.47 分。作为对照,不缓存转移算子的「朴素相加」直接损失了 66.9% 的分数,印证了 T_C 的必要性。


吞吐与 QPS:在生产 RAG trace 上,同 1 秒 TTFT SLO 下,HYPIC 把可持续 QPS 相比 Prefix Cache 提升 1.66 倍(1.49×–1.85×),峰值单卡吞吐提升约 1.3–1.5 倍。


段并行加速冷请求: 在纯冷未命中路径上,一条 32k token 请求的 TTFT 从单 worker 的 2.83 秒,降到 8 worker 的 0.49 秒——5.7 倍加速,且几乎全部收益来自可近线性扩展的 scatter 阶段;面对不均匀分片,LPT 相比 Round-Robin 把 TTFT 从 1.26 秒压到 0.84 秒(3.6× vs 2.4×)。

开销可控:构造 (T_C, S(C|0)) 的一次性开销,在最重的稠密转移模型上也仅为主前向的 5.2%–6.7%,且一次构造、多次复用摊薄。

HYPIC 首次把「位置无关缓存」带到了混合注意力大模型上:用缓存的段累积转移算子实现线性层的常数时间状态组合,用小小的缝合窗口修复全注意力层的跨段对齐,再用段并行把长冷请求也变成可加速的负载。它让 RAG / Agent 这类长上下文服务,在拥抱高效混合架构的同时,也能享受到片段级缓存复用的红利。

未来,我们将进一步探索分布式的位置无关缓存管理与调度、多级缓存分层管理,推动大模型推理向「更快、更省、更可扩展」持续演进。

53AI,企业落地大模型首选服务商

产品:场景落地咨询+大模型应用平台+行业解决方案

承诺:免费POC验证,效果达标后再合作。零风险落地应用大模型,已交付160+中大型企业

联系我们

售前咨询
186 6662 7370
预约演示
185 8882 0121

微信扫码

添加专属顾问

回到顶部

加载中...

扫码咨询

扫码登录
登录即表示您同意《53AI网站服务协议》
服务协议

欢迎您使用【53AI 官方网站】(以下简称“本网站”或“我们”)。本《会员服务协议》(以下简称“本协议”)是您(以下简称“会员”或“用户”)与【深圳市博思协创网络科技有限公司】之间关于注册、登录及使用本网站会员服务所订立的法律协议。

在您注册或登录前,请务必审慎阅读、充分理解各条款内容,特别是免除或限制责任的条款、知识产权条款、争议解决条款等。此类条款将以加粗形式提示您注意。 当您通过微信公众号授权、手机验证码验证或其他方式成功登录本网站时,即视为您已完全理解并同意接受本协议的全部内容。

一、 定义

本网站:指由【深圳市博思协创网络科技有限公司】运营的,域名为【53ai.com】的网站及相关移动端页面。

会员服务:指本网站向注册会员提供的知识库文章查阅、内容检索及其他相关增值服务。

知识库内容:指本网站发布的包括但不限于文字、图表、数据、研究报告、行业分析等数字化内容资源。

二、 账号注册与登录

登录方式:本网站支持以下登录方式,您可根据实际情况选择:

微信公众号授权登录:您同意将您的微信OpenID信息授权给本网站,用于创建或关联会员账号。

手机验证码登录:您需提供真实有效的手机号码,并通过短信验证码完成身份验证与登录/注册。

账号安全:您的账号仅限您本人使用,禁止赠与、借用、租用、转让或售卖。因您保管不善导致的账号被盗、密码泄露等损失,由您自行承担。

实名认证:根据相关法律法规要求,我们可能要求您在特定功能下完成实名认证。如您拒绝提供,可能无法使用部分或全部服务。

未成年人保护:若您未满18周岁,请在法定监护人的陪同下阅读本协议,并在征得监护人同意后使用本服务。

三、 服务内容与规范

知识库查阅权限:会员登录后,有权按照其会员等级对应的权限范围,在线浏览、检索本网站知识库中的相关文章及内容。

服务变更:我们有权根据业务发展需要,调整、变更或终止部分服务内容,并将以网站公告、公众号消息等方式提前通知。

禁止行为:您在使用服务时不得实施以下行为:

利用技术手段批量爬取、下载、转存知识库内容;

将知识库内容用于商业目的或未经授权地向第三方传播;

干扰本网站正常运行或侵犯其他用户合法权益;

发布违法违规信息或从事违反公序良俗的活动。

四、 知识产权声明

权利归属:本网站知识库中的排版设计、软件代码等内容的知识产权均归【公司全称】或原权利人所有,受《中华人民共和国著作权法》等法律保护。

有限许可:本网站授予会员一项非独占、不可转让、不可转授权的普通许可,仅限于个人学习、研究之目的在线查阅知识库内容。

侵权追责:未经书面许可,任何单位或个人不得以任何形式复制、转载、摘编、镜像、汇编或以其他方式使用上述内容。一经发现,我们保留追究其法律责任的权利。

五、 个人信息保护

我们重视对您个人信息的保护。关于我们如何收集、使用、存储和保护您的个人信息,请单独阅读 《隐私政策》。

您通过微信公众号授权或手机号验证所提供的信息,我们将严格按照《个人信息保护法》的规定处理,仅用于身份识别、服务提供及安全验证等必要用途。

您可以随时通过网站设置或联系客服行使查阅、更正、删除个人信息及撤回授权同意的权利。

六、 免责声明

内容准确性:知识库内容仅供参考,不构成专业建议。我们不对其完整性、准确性、时效性作任何明示或暗示的保证,您应自行判断并承担使用风险。

不可抗力:因自然灾害、政策法规变化、网络故障、第三方平台接口异常(如微信接口维护、运营商短信通道故障)等不可抗力导致的服务中断或延迟,我们不承担违约责任。

第三方链接:本网站可能包含指向第三方网站的链接,该等网站的内容和服务不受我们控制,请您自行甄别风险。

七、 违约责任

如您违反本协议约定,我们有权视情节采取警告、限制功能、暂停服务、注销账号等措施,并保留要求赔偿损失的权利。

如因您的违约行为导致我们遭受行政处罚、第三方索赔或商誉损失,您应承担全部赔偿责任(包括但不限于罚款、赔偿金、律师费、公证费等)。

八、 法律适用与争议解决

本协议的订立、执行和解释均适用中华人民共和国大陆地区法律。

因本协议产生的或与本协议有关的任何争议,双方应友好协商解决;协商不成的,任何一方均可向【公司所在地】有管辖权的人民法院提起诉讼。

九、 其他

本协议构成双方就本服务达成的完整协议,取代此前任何口头或书面约定。

本协议任一条款被认定为无效或不可执行的,不影响其他条款的效力。

我们对本协议享有最终解释权,并在法律允许的范围内保留随时修改的权利。修改后的协议一经公布即生效,继续使用服务即视为同意修订内容。


已查阅