月之暗面与清华新论文: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 认可其观点表述,也不得被视为财务或专业建议。详见声明
  • 赞赏
  • 8
  • 2
  • 分享
评论
请输入评论内容
请输入评论内容
SlippageSailor
· 8 分钟前
PrfaaS这名字,产品经理是懂起名的
回复0
Exit Liquidity Poet
· 3小时前
长上下文未命中才走远程,短请求本地处理,这个路由策略很细
回复0
金属浮雕机械手
· 4小时前
同构PD vs 异构 vs PrfaaS,这对比实验设计得挺干净
回复0
Gas费别闹
· 4小时前
1T参数模型实测,敢拿这么大的模型跑,底气够足
回复0
GateUser-4590f4c6
· 4小时前
把预填充当服务卖,以后会不会出现预填充即插即用?
回复0
月光断线器
· 4小时前
看完整篇,最想知道的是实际部署时丢包率容忍度多少
回复0
玻璃穹顶漫游
· 4小时前
带宽感知调度,说白了就是穷则思变,网贵就得精打细算
回复0
玻璃鱼缸套利
· 4小时前
以太网传KV缓存,以前觉得疯了,现在居然成了论文
回复0