星空体育

网易星空体育 > > 正文

每天500万亿Token背后,KV Cache正把存储墙推到台前

0
分享至


过去二十年,服务器峰值浮点算力每两年增长约3倍,而DRAM带宽每两年只增长1.6倍,互连带宽仅1.4倍,这就形成了行业内称之为存储墙的痛点,而这个问题其实已经存在很久,而如今的AI时代,随着计算能力的增强及其对数据需求量的扩大,据中国信通院统计,中国每天平均的Token消耗量,从2024年的1000亿,跃升到2025年8月的500万亿,一年多时间增长了约5000倍。在此背景下,存储墙问题也被放大了。

推理时代,算力与存储必须加强融合

大模型推理的核心流程之一,是自回归解码,模型每生成一个Token,都要回头看一遍之前所有的上下文。如果每次都重新计算全部历史数据,算力会被海量的重复计算消耗殆尽。于是KV Cache诞生了,它把中间计算结果缓存起来,用“以存换算”的方式跳过重复计算。这个机制极大地减少了计算开销,但它同时埋下了一颗雷:数据越攒越多,而缓存总得有地方放。

这个问题随着Token消耗量的提升在不断被放大,对此,联想集团首席科学家肖利民告诉我们,过去典型的AI应用是一问一答的对话机器人,如今变成为能够感知、会规划、能执行任务的数字员工。实际上,每一次Agent执行任务,背后都是动辄数十轮、上下文动辄数万Token的推理调用。

而推理,恰恰是对存储要求最高的环节。KV Cache的存放问题,本质上是一场空间挤压战,数据如放全部在GPU显存,显存通常不够用,进而人们想到将数据卸载到CPU直连内存,但CPU直连内存容量同样扛不住。因此,数据很自然地再往外卸到外部存储,但是GPU无论走PCIe或网络去访问外部存储,带宽和延迟与访问本地显存相比,存在数量级的差距,“数据在外部存储和GPU之间搬运的开销,往往会拖累推理性能。”肖利民指出。

在此背景下,推理基础设施就出现了两个痛点。一个是KV可用的GPU显存和CPU直连内存容量不足,一个是GPU访问外部存储的开销太大。业界现有的主流方案是KV Cache 卸载到外部共享CXL内存池,但该方案主要解决前一个问题,通过CXL交换机,把内存资源与CPU解耦,实现独立弹性扩容和多主机共享。但对后一个痛点,也就是GPU访问CXL内存池开销大问题,始终力不从心。

实际上,更底层的核心问题是存储硬件与计算硬件的发展速度不匹配。计算性能的提升速度远快于存储,而且这个差距还在逐年拉大。这就是计算机体系结构领域经典的存储墙问题,而这个问题从来没有被完全真正解决过,人们用多级缓存等各种手段不断缓解它,但缓解不等于消除。大模型推理的爆发,不过是把这道旧墙,重新又推到了所有人的面前。

于是,存算融合的思路呼之欲出:既然把数据搬到算力那边成本太高,那就反其道而行之,把计算单元搬到数据旁边去。与其让海量的KV数据经过PCIe链路在GPU和外部存储之间奔波,不如让计算单元驻扎在内存旁边,就地处理数据,只把计算结果传回来。

对此,肖利民给出的路径是,把内存池化技术和近存计算技术结合在一起,构建一套面向大模型推理的存算融合池化系统,用内存池化解决KV Cache的大容量扩容和跨实例共享,用近存计算在内存池内嵌计算单元,把 attention这类访存密集型算子下沉到内存池一侧,从整体上降低端到端的推理时延、提升系统吞吐。

在肖利民看来,这条路线并不算激进,“技术的发展通常是逐步演进的,”他进一步指出,“现阶段算力基础设施出现断崖式重写的可能性不大,但推理负载带来的这些新变化,确实需要更多创新技术被引入到推理算力基础设施里面去,存算融合池化技术就是这个演进过程中一个非常重要的技术点。”

平衡点在哪?

从当前存算融合技术路径上看,主要分为CIM(存算一体)、PIM(存内计算)、以及PNM(近存计算)三条路径。前两者通常把计算单元直接做到了存储芯片内部,带宽极高、延迟极低,这种做法在某种意义上已经算是存算一体了,但这通常需要改变现有存储器本身的结构,属于高度定制化的硬件,与现有的软件生态难以兼容。对任何一个要考虑产品化落地的团队来说,这都是一道需要攻克的坎。

有鉴于此,在肖利民看来,不改动存储器本身结构的PNM是当下较为现实的路径。对此,肖利民表示,把计算单元放置在存储器的边上、内存控制器的位置,既拉近了计算与存储的距离,又不改动存储器原有的结构,从源头上维持了与现有软硬件生态的兼容性。这个选择背后是清醒的商业判断:“我们未来要真正落地做产品化实现,兼容现有生态的路径会有很大的优势。”肖利民解释道。

但在实现近存计算的产品化路上,还有很多挑战。其中肖利民认为最大的挑战有两个。第一个是近存计算单元放置策略选择上的两难。计算单元要既离存储足够近,又不对现有生态产生大的改变,放在哪里本身就是一道难题。选定了内存控制器附近之后,新的问题接踵而至,内存控制器原本就承担着来自主机的访存请求,如今旁边又多了一个近存计算单元的访存请求,两路请求同时涌向存储器,冲突不可避免。团队的解决办法是在内存总线上内置一套 QoS 仲裁机制,对两路不同的访问进行仲裁调度,既避免冲突,又保证两边的服务质量不会互相拖累,还能维持一定的公平性。“存储总线的速率高,如何做冲突检测,如何保证两边的QoS不往下掉,同时还要保证公平性,确实是个较为复杂的问题。”肖利民强调道。

第二个是算子相对固化与模型快速迭代之间的矛盾。大模型几乎每隔几个月就要更新换代,而硬件计算单元相对固化。联想采用的是指令驱动的可重构计算编排思路,让同一套硬件通过指令驱动算子编排适配不同模型的计算需求,缓解了这个矛盾。

除此之外,在软件实现和负载运行层面,还要面临如何用统一编址实现算力统一纳管和高效调用?多机共享内存池时如何保证数据一致性?以及一次推理请求到来时,哪些计算该下沉到内存池侧、哪些留在GPU本地算、KV该不该卸载等问题,这些问题都需要寻找到一个最佳的平衡点。据肖利民介绍,联想研究院智能计算基础设施团队为此建立了一个成本分析模型,根据链路拓扑、上下文长度等输入动态决策,在编排策略上按头切分KV并与GPU张量并行策略对齐,尽量避免设备之间的通信,提高设备并行计算效率。

当所有问题都找到解决的平衡点之后,联想团队进行了测试,肖利民介绍,他们在三个核心模块上进行了分项性能测试。在内存池卸载方面,与NVMe层池化方案对比,在固定输入长度场景下,首Token延迟(TTFT)可降低3.5倍;多轮对话场景下可降低2.4倍,且随着对话轮次增多,优势愈发明显;在KV跨实例共享方面,借助内存池实现KV复用,相比完全重算,首Token延迟最高加速12~28倍;相比RDMA分布式内存架构,也有2~3倍的加速;在存算融合方面,把注意力计算下沉到内存池侧后,解码延迟可降低2.5倍,同时腾出大量显存,用以支持更长上下文、更大规模的模型。为验证对大模型推理的整体加速效果,联想团队还进行了端到端推理性能测试。在推理延迟方面,与RDMA分布式内存池相比,取得了2~3倍的加速;在系统吞吐方面,性能提升更为可观,随着上下文变长,系统吞吐提升6~20倍。

从G2.5到G1.5是接下来要做的事

肖利民团队目前实现的是G2.5层次的基于内存池化和近存计算技术的存算融合池化系统。但肖利民告诉我们,接下来从G2.5向G1.5演进将会是重点研发的方向之一。

具体来说,其思路是借助超级节点里scale-up高速互连技术,把共享存算融合池进一步向GPU靠拢,从G2.5推进到G1.5层次,进一步缩短访问路径。“一旦超节点成为AI系统主流,利用它里面scale-up高速互连,不仅享有G2.5已获得的收益,而且在计算与存储的互连带宽、延迟上,都会比现有互连网络有一个数量级的提升。”肖利民说。在硬件上,3D堆叠封装、TSV垂直互连等技术逐渐成熟后,也有望进一步提升存算之间的带宽密度;软件上则会持续扩展算子库,适配更多主流模型。肖利民同时强调,随着芯片新技术的日益成熟,团队会在“存算融合”理念下会吸纳更多新的芯片技术,使得推理性能在现有水平上再提升一个层次。

在肖利民看来,联想存算融合池化技术的差异化特色,主要在于将内存池化和近存计算相结合,同时解决容量不足和访存开销大这两大难题。“目前国内外相关工作大多聚焦于怎么做出一个大容量内存池,解决KV Cache容量不足的问题;或者,通过存算融合技术,缓解访存开销大的问题”肖利民进一步指出,“目前暂未见到把近存计算引入到大容量内存池里,同时KVcache解决容量不足和访存开销大两个难题的产品。”。

当然,算力基础设施技术的演进并不会一蹴而就,目前阶段出现类似AI技术的“GPT时刻”的可能性不太大。在肖利民看来,在算力基础设施中,计算、存储、通信、调度多种要素交织在一起,相互融合、协同优化,“某一块跑快了,另一块拖后腿,然后被加快,再又换一块成了短板,有点像是按下葫芦浮起瓢。” 肖利民进一步表示,“我们的存算融合池化技术不是要终结这样一种局面,而是在大模型推理负载把存储墙问题推到舞台中央之后,通过近存计算靠近大内存池的方式,来加速大模型推理性能。实际上,更为理想的方式,不仅局限于存储和计算的融合,而是围绕计算、存储、通信、调度多种要素做系统性的协同优化,以实现算力基础设施及其上运行大模型推理应用的更大性能提升。”

特别声明:以上内容(如有图片或视频亦包括在内)为自媒体平台“网易号”用户上传并发布,本平台仅提供信息存储服务。

Notice: The content above (including the pictures and videos if any) is uploaded and posted by a user of NetEase Hao, which is a social media platform and only provides information storage services.

相关推荐
热点推荐
“听到咔嚓一声”,快乐没有5分钟!多位网友晒图:iPhone Duo 刚到手折叠屏就裂开,维修费最高近8000元!

“听到咔嚓一声”,快乐没有5分钟!多位网友晒图:iPhone Duo 刚到手折叠屏就裂开,维修费最高近8000元!

深度知局
2026-10-26 21:51:33
姆巴佩疑发表针对中国的不当言论!国内媒体发声 球迷:又是皇马的

姆巴佩疑发表针对中国的不当言论!国内媒体发声 球迷:又是皇马的

风过乡
2026-10-27 08:05:27
太恐怖了!入职7天被熟人匿名举报劝退,简历核查无问题仍丢offer,网友:出卖你的,都是了解你的人

太恐怖了!入职7天被熟人匿名举报劝退,简历核查无问题仍丢offer,网友:出卖你的,都是了解你的人

火山詩话
2026-10-26 17:27:10
万岁山NPC“胖郎中”宣布离职:4个月把了9000人的脉,拒绝景区涨薪挽留,想当一名好中医,“在这里一辈子也不可能实现”,将考医师资格证

万岁山NPC“胖郎中”宣布离职:4个月把了9000人的脉,拒绝景区涨薪挽留,想当一名好中医,“在这里一辈子也不可能实现”,将考医师资格证

极目星空体育-星空online(中国)
2026-10-26 14:28:19
凌晨,美国飞北京航班起飞后突然返航,机上有262人,航司回应

凌晨,美国飞北京航班起飞后突然返航,机上有262人,航司回应

南方都市报
2026-10-26 08:21:39
网传女孩抑郁症,结三次婚,每次不到半年就离,彩礼不退,已经攒了几十万,光金手镯就有仨

网传女孩抑郁症,结三次婚,每次不到半年就离,彩礼不退,已经攒了几十万,光金手镯就有仨

丫头舫
2026-10-26 23:07:52
国际乒联227家乒协收举报信!核心内容曝光:直指刘国梁+王楠夫妇

国际乒联227家乒协收举报信!核心内容曝光:直指刘国梁+王楠夫妇

念洲
2026-10-27 08:11:32
男子炒股10年倾家荡产,如今妻子病重急需用钱,打开电脑后他傻眼

男子炒股10年倾家荡产,如今妻子病重急需用钱,打开电脑后他傻眼

罪案洞察者
2025-09-02 11:30:12
演员黄磊女儿黄多多红毯首秀引热议!

演员黄磊女儿黄多多红毯首秀引热议!

韩小娱
2026-10-25 13:56:15
马院暴增至1440家!不包分配的毕业生,都去哪了?

马院暴增至1440家!不包分配的毕业生,都去哪了?

户外阿毽
2026-10-26 13:34:03
人民日报痛批集采极端低价!7.4元内镜止血夹,显著低于成本!

人民日报痛批集采极端低价!7.4元内镜止血夹,显著低于成本!

梅斯医学
2026-10-26 20:54:21
原来他已在美国去世,仅55岁,是中国男篮名宿,哥哥是篮球"狂"徒

原来他已在美国去世,仅55岁,是中国男篮名宿,哥哥是篮球"狂"徒

娱圈办事处
2026-10-26 14:42:15
乌情报公布:俄“匕首”导弹工厂使用中国机床

乌情报公布:俄“匕首”导弹工厂使用中国机床

桂系007
2026-10-26 18:18:49
美国入境新规!11月3日起赴美航班旅客都要签声明,绿卡、公民也不例外

美国入境新规!11月3日起赴美航班旅客都要签声明,绿卡、公民也不例外

北美省钱快报
2026-10-27 03:58:37
黄蜂险胜奇才:米勒39分生涯新高迎500三分里程碑 浓眉29+10

黄蜂险胜奇才:米勒39分生涯新高迎500三分里程碑 浓眉29+10

醉卧浮生
2026-10-27 09:26:19
杨议诚恳道歉,演出被取消,还有374万债务没还,文旅局介入核查

杨议诚恳道歉,演出被取消,还有374万债务没还,文旅局介入核查

江山挥笔
2026-10-27 07:16:05
Anthropic科学家宣布:全球博士生职业生涯一夜清零!

Anthropic科学家宣布:全球博士生职业生涯一夜清零!

新智元
2026-10-27 02:27:43
男子救轻生女子反被诬告后拘留,看守所释放当晚,他怒杀女子全家

男子救轻生女子反被诬告后拘留,看守所释放当晚,他怒杀女子全家

罪案洞察者
2025-08-18 11:09:26
杭州一高架上风筝线割坏两辆车,车主称找不到肇事者只能自行出险赔偿

杭州一高架上风筝线割坏两辆车,车主称找不到肇事者只能自行出险赔偿

现代快报
2026-10-26 22:37:20
上海用户买冰柜后选择仅退款,自称冰柜"已销毁"无法退货,商家跨省索要后发现冰柜正插电使用,结局亮了

上海用户买冰柜后选择仅退款,自称冰柜"已销毁"无法退货,商家跨省索要后发现冰柜正插电使用,结局亮了

子非鱼人
2026-10-26 23:41:51
2026-10-27 10:00:49
钛媒体APP incentive-icons
钛媒体APP
独立财经科技媒体
141092文章数 862690关注度
往期回顾 全部

Anthropic最硬气招股书,先把丑话说在前头

披露缅北电诈的自媒体人已遇害 在广州被劫持遗弃山上

披露缅北电诈的自媒体人已遇害 在广州被劫持遗弃山上

13年后重回诺坎普,穆里尼奥又滑跪了

胡歌杨幂刘诗诗袁弘陈龙结伴去埃及

金价跌回8字头

云辇-M上车 第二代海豹07让驾驶者也能爱上的家用车

态度原创

苹果霸占前四,W42单品销量盘点,国产第一是华为

海甸岛26万人,还缺一个年轻人“待得住”的地方

新手爸妈如何提升性生活质量?

七林647、九宁642、四文637:头部高中录取线,家长该看位次不是死盯分

高清大图!歼-15红蓝对抗训练最新画面

无障碍浏览 进入关怀版