月之暗面与清华新论文:LLM预填充可跨数据中心,1T模型吞吐升54%

robot
摘要生成中
ME News 消息,4 月 18 日(UTC+8),据 动察 Beating 监测,月之暗面(Moonshot AI)与清华大学 4 月 16 日在 arXiv 挂出新论文《Prefill-as-a-Service》,提出让大模型推理的预填充阶段(prefill)跨数据中心运行。 大模型推理分两步:prefill 先把输入一次性读进来、生成一份 KV 缓存;decode 再根据这份缓存逐字吐出结果。两步需要的硬件特性完全不同,prefill 吃算力,decode 吃显存带宽。业界主流做法是把两步拆到不同机器上(PD 分离),但这要求两边在同一个数据中心里用 RDMA 互联,因为密集 attention 模型的 KV 缓存每秒几十 Gbps 地吐,一旦传慢 GPU 就空转。 转折来自新一代 hybrid attention 模型。论文实测 Kimi Linear、MiMo-V2-Flash、Ring-2.5-1T 等模型通过少量完整 attention 层加大量线性层的组合,把 KV 缓存吞吐量砍掉了约一个数量级,Ring-2.5-1T 的综合压缩比达到 36 倍。这时 KV 缓存可以从 RDMA 专网搬到普通以太网上传。 PrfaaS 的具体做法:组建独立的「预填充集群」,只把长上下文、未命中前缀缓存的请求路由过去,短请求留在本地 PD 集群;预填充完成后通过以太网把 KV 缓存回传本地集群做 decode。配套引入长度阈值路由、带宽感知调度器和混合前缀缓存池。论文用内部 1T 参数 hybrid 模型(基于 Kimi Linear 架构)做了一组实测,整体服务吞吐比同构 PD 部署高 54%,比朴素异构方案高 32%,每台机器只占适中的跨数据中心带宽。 (来源:BlockBeats)
此页面可能包含第三方内容,仅供参考(非陈述/保证),不应被视为 Gate 认可其观点表述,也不得被视为财务或专业建议。详见声明
  • 赞赏
  • 9
  • 2
  • 分享
评论
请输入评论内容
请输入评论内容
YieldNotYell
· 28 分钟前
长度阈值路由这个设计很细,长短请求分开处理才是正经优化
回复0
Circuit Daydreamer
· 3小时前
混合注意力模型降KV缓存吞吐,技术细节等论文细读
回复0
AirdropCartographer
· 4小时前
54%提升确实香,但跨数据中心走以太网,抖动怎么搞
回复0
深海冷启动
· 4小时前
只路由未命中请求,缓存命中率成了关键瓶颈
回复0
玻璃穹顶下
· 4小时前
同构PD vs 异构PD vs PrfaaS,这对比维度设得挺聪明
回复0
蓝牡丹冷静剂
· 4小时前
1T参数模型测这个,硬件成本不敢想
回复0
GateUser-fb035825
· 4小时前
预填充集群独立部署,运维复杂度又上去了,收益值得吗
回复0
闲鱼DAO成员
· 4小时前
带宽感知调度听着简单,实际落地估计一堆坑
回复0
GateUser-aa277334
· 4小时前
这思路有意思,把预填充扔远端,本地专心解码,延迟能扛住吗
回复0
查看更多