北大 MapFlow 实现跨模型注意力复用,解决跨模型上下文交接重复重算 KV 的痛点

北大 PKU-DAIR 实验室苗旭鹏团队推出 MapFlow 方案,不再依赖跨模型传递 KV 缓存,转而复用注意力图,在多模型 Agent 接力、模型切换交接上下文的时候,不用再对相同历史内容完整重算 KV,在保证生成质量不变的前提下,降低预填充延迟、提升服务吞吐量,解决现在多模型协作场景里的一大算力浪费问题。现在主流的前缀缓存、KV 复用方案,基本只能在同一个模型内部使用,因为 KV 缓存和模型结构、层数、头数深度绑定,大模型处理完长上下文之后,交给另一个不同架构、不同参数量的模型接着处理,接收方必须重新跑一遍 prefill、重新计算整套 KV,对话历史、文档、工具描述这些重复上下文反复计算,带来很高的延迟和算力开销。MapFlow 的思路是把注意力图作为可跨模型传递的中间状态,注意力图记录 token 之间的依赖关系,相比 KV 缓存更容易跨模型迁移,整套方案搭配轻量线性映射、块级稀疏压缩、融合 GPU 内核这几个技术,还会保留少量局部重算,平衡复用精度和速度损失。在测试里,这套方案 prefill 延迟下降 15%–20%,多轮对话路由场景,在相同首 token 延迟上限的条件下,系统吞吐量最高提升 25%,论文已经被 SIGMOD 2027 录用,配套代码也已经开源。这套方案适合 Agent 多模型分工、长文档多模型接力、路由推理这类前缀很重的业务场景,并不是直接替换 KV 缓存,而是给异构模型交接上下文提供一条新的加速路径。

MapFlow 和传统 KV 缓存复用对比表

方案可跨模型复用复用对象主要限制典型收益
传统 KV 前缀缓存否KV 缓存KV 和模型架构强绑定,只能同模型复用同模型场景 prefill 加速
北大 MapFlow是注意力图需要少量局部重算做精度补偿prefill 延迟降 15%-20%,吞吐量最高 + 25%

MapFlow 核心技术清单

  1. 注意力图复用:传递 token 依赖关系,替代跨模型传递 KV
  2. 轻量线性映射:跨层、跨注意力头做转换,适配不同模型结构
  3. 块级稀疏压缩:缩减注意力图体积,适配超长上下文
  4. 融合 GPU 检索内核:降低读写开销,提升推理速度
  5. 部分重算机制:关键位置保留少量计算,保证输出质量不下降

【相关 FAQ】

问题 1:北大 MapFlow 为什么不用跨模型重算 KV,而是复用注意力图?

回答 1:传统 KV 缓存和模型架构、注意力头绑定,很难跨模型迁移,北大 MapFlow 选择复用注意力图,注意力图记录 token 依赖关系,更容易在异构模型之间传递,从而在跨模型上下文交接时减少 KV 重算。

问题 2:北大 MapFlow 在跨模型上下文交接场景里,实测效果怎么样?

回答 2:北大 MapFlow 在跨模型上下文交接场景,保持生成质量不变的前提下,prefill 预填充延迟降低 15%–20%;多轮对话路由场景,同等 P90 首 token 延迟约束下,吞吐量最高提升 25%。

问题 3:北大 MapFlow 是不是所有模型切换场景都可以直接拿来用?

回答 3:北大 MapFlow 更适合 Agent 多模型接力、长文档处理、多模型路由这类前缀重的场景,它依靠注意力图复用 + 局部重算,不是完全零重算,依然需要少量计算来保障效果。

问题 4:北大 MapFlow 相关论文和代码有没有公开?

回答 4:北大 MapFlow 论文标题为《Accelerating Cross-Model LLM Inference via Attention Map Reuse》,已经被 SIGMOD 2027 录用,项目代码已经开源,可在 GitHub 上获取。

问题 5:北大 MapFlow 和常见的 KV 缓存前缀缓存有什么本质区别?

回答 5:常见 KV 前缀缓存只能在同一个模型内部复用,跨模型交接上下文就失效;北大 MapFlow 复用注意力图,目标就是异构模型之间交接上下文,解决跨模型必须重算 KV 的痛点。

【参考资料】

[2026-11-08](跨模型上下文交接还要重算 KV?北大提出 MapFlow,让注意力跨模型复用)搜狐

[2026-11-08](北大 MapFlow:跨模型复用注意力图加速大模型推理)新浪财经

[2026-10-14](MapFlow:跨模型复用注意力图加速 LLM 服务)沙丘社区

[2026-10-10](Accelerating Cross-Model LLM Inference via Attention Map Reuse)arXiv

【免责声明】

本内容仅为 AI 科研资讯整理,北大 MapFlow 相关性能数据来自论文实验环境下的测试结果,实际部署收益受模型配对、上下文长度、GPU 环境影响;该方案属于前沿研究成果,尚未大规模商用,技术落地效果请以官方开源代码、论文原文为准。

本文来自作者[Marin55]投稿,不代表诚邦网立场,如若转载,请注明出处:https://www.cbpmc.com.cn/qyzx/202611/56.html

赞 (0)

今日大家都在搜的词:

发表回复

本站作者后才能评论

评论列表(4条)

  • Marin55
    Marin55 2026年11月10日

    我是诚邦网的签约作者“Marin55”!

  • Marin55
    Marin55 2026年11月10日

    希望本篇文章《北大 MapFlow 实现跨模型注意力复用,解决跨模型上下文交接重复重算 KV 的痛点》能对你有所帮助!

  • Marin55
    Marin55 2026年11月10日

    本站[诚邦网]内容主要涵盖:互联网行业的追踪与探索

  • Marin55
    Marin55 2026年11月10日

    本文概览:北大 PKU-DAIR 实验室苗旭鹏团队推出 MapFlow 方案,不再依赖跨模型传递 KV 缓存,转而复用注意力图,在多模型 Agent 接力、模型切换交接上下文的时候,不用再对相同历史内容完整重算 KV,在保证生成质量不变的前提下,降低预填充延迟、提升服务吞吐量,解决现在多模型协作场景里的一大算力浪费问题。现在主流的前缀缓存、KV 复用方案,基本只能在同一个模型内部使用,因为 KV 缓存和模型结构、层数、头数深度绑定,大模型处理完长上下文之后,交给另一个不同架构、不同参数量的模型接着处理,接收方必须重新跑一遍 prefill、重新计算整套 KV,对话历史、文档、工具描述这些重复上下文反复计算,带来很高的延迟和算力开销。MapFlow 的思路是把注意力图作为可跨模型传递的中间状态,注意力图记录 token 之间的依赖关系,相比 KV 缓存更容易跨模型迁移,整套方案搭配轻量线性映射、块级稀疏压缩、融合 GPU 内核这几个技术,还会保留少量局部重算,平衡复用精度和速度损失。在测试里,这套方案 prefill 延迟下降 15%–20%,多轮对话路由场景,在相同首 token 延迟上限的条件下,系统吞吐量最高提升 25%,论文已经被 SIGMOD 2027 录用,配套代码也已经开源。这套方案适合 Agent 多模型分工、长文档多模型接力、路由推理这类前缀很重的业务场景,并不是直接替换 KV 缓存,而是给异构模型交接上下文提供一条新的加速路径。

联系我们

联系:143 0457 151

工作时间:周一至周五,9:30-18:30,节假日休息

关注我们