网易首页 > 网易号 > 正文 申请入驻

ICDCS 2026 | RcLLM:突破前缀缓存限制,加速生成式推荐

0
分享至

该研究面向大语言模型驱动的生成式推荐,关注长输入带来的在线推理延迟:不同推荐请求往往包含重复的用户历史或候选商品信息,但这些内容在输入中的位置和组合会不断变化,现有 Prefix Caching 难以充分复用已有计算。为此,研究提出分布式推理系统 RcLLM,将 KV Cache 的复用范围从共同前缀扩展到非前缀内容,并结合分层缓存管理、缓存感知的请求调度与选择性重计算,在控制推荐质量变化的同时减少重复计算,缩短用户等待推荐结果的时间,为大模型推荐服务的高效部署提供系统层面的优化方案。


论文题目:
RcLLM: Accelerating Generative Recommendation via Beyond-Prefix KV Caching
论文链接:
https://arxiv.org/abs/2605.07443


摘要

大语言模型能够结合用户评论、历史行为和商品描述生成个性化推荐,但这些信息也使输入越来越长。现有 Prefix Caching 要求请求从开头共享相同内容,难以复用位于不同用户历史之后、顺序又经常变化的商品描述。结果是,同一段商品内容可能被反复计算,成为在线推荐的延迟来源。

RcLLM 通过 Beyond-Prefix KV Caching 将复用范围扩展到非前缀内容,并结合分层缓存、缓存感知调度与选择性重计算,在减少在线计算的同时控制推荐质量变化。在默认 40 实例配置的性能仿真中,RcLLM 相比 Prefix-Cache 在两种模型和三个数据集上的 P99 首 token 延迟获得 1.4–2.8 倍加速,为高效部署大模型推荐提供了新的系统设计思路。

一、背景与动机

在基于大语言模型的推荐场景中,系统会把任务指令、用户历史和候选商品一起组织成 Prompt,再由模型理解用户偏好并输出排序。例如,在图书推荐中,模型可以综合用户过去的评论与候选图书的内容介绍,判断哪些图书更符合当前需求。图 1 展示了这种输入结构。


图 1 : 生成式推荐的输入与输出。任务指令、用户历史和候选商品共同构成 Prompt,模型据此生成推荐排序。

这类任务通常需要读取数千个 token,最后只输出一份较短的商品列表。其主要开销集中在 Prefill 阶段:模型要先处理完整输入,为各个 token 建立 Key 和 Value 状态,才能开始生成结果。后续 Decode 阶段则复用这些 KV Cache,逐个生成输出。因此,首 token 延迟(Time-To-First-Token,TTFT)直接影响用户等待推荐结果的时间。图 2 给出了两个阶段的关系。


图 2 : 大语言模型的自回归推理流程。Prefill 处理输入并建立 KV Cache,Decode 利用已有状态继续生成。

问题在于,不同请求中存在大量重复内容,却未必满足前缀复用的条件。两位用户可能看到同一本书,但各自的历史记录不同,图书描述在 Prompt 中的位置也可能改变。传统 Prefix Caching 只能复用从开头连续匹配的部分,因而难以利用这些分散的重复内容。RcLLM 的出发点,是让复用围绕内容本身展开,减少由输入位置和排列变化引起的重复计算。

二、技术挑战

将这一思路用于推荐服务,需要同时解决缓存规模、执行效率和推荐质量三个相互关联的问题。

首先,缓存需要适应不同类型的数据。用户评论在文字上各不相同,却具有相似的语义表达;商品标题和描述相对稳定,可以提前处理,但整个商品库的 KV 状态可能达到 TB 级。如何同时利用语义相似性和商品重复性,并控制存储成本,是实现跨请求复用的基础。

其次,缓存分布在不同节点后,请求应发往哪里会直接影响收益。选择本地缓存较多的节点能够减少重计算,但热门节点可能出现排队;只选择最空闲的节点,又可能频繁遇到缓存缺失。调度必须同时考虑缓存位置与当前负载。

最后,复用后的状态需要适应当前上下文。一个商品出现在不同用户历史之后,其位置和需要关注的信息都会变化。直接拼接独立计算的 KV 块可能影响商品排序,因此系统还需要识别并更新关键内容,在计算量与推荐质量之间取得平衡。

三、RcLLM 的核心方法

针对上述挑战,RcLLM 将缓存准备放到离线阶段,在请求到达后完成缓存选择、请求调度和必要的状态更新。图 3 展示了整体架构:CPU 内存保存可复用的历史与商品 KV,Global Scheduler 选择执行节点,GPU 上的 Selective Attention 对关键位置进行重计算。


图 3 : RcLLM 整体架构。离线阶段准备两类缓存;在线阶段结合节点选择、缓存读取和 Selective Attention 完成推理。

3.1 分层缓存覆盖用户历史与候选商品

对用户历史,RcLLM 利用评论中反复出现的语义表达。图 4 左侧显示,不同评分的评论 token 呈现明显的语义聚类;右侧进一步表明,在所分析的 Amazon 数据中,超过 93% 的新评论 token 能在历史语义池中找到相似度接近 1 的匹配。


图 4 : 用户历史中的语义复用机会。左图为 1 星与 5 星评论中 token 的语义分布,右图为新评论 token 与历史池的最大相似度累积分布。

基于这一观察,系统将语义和位置信息一起编码,借助 Locality-Sensitive Hashing(LSH,局部敏感哈希)构建紧凑的语义原型库,并预计算对应的 KV 状态。在线请求通过近邻匹配复用这些状态。该缓存池在 Qwen3-8B 下包含约 10⁵ 个原型、占用约 30 GB,复制到各节点的 CPU 内存,使历史缓存可以在本地访问。

候选商品采用另一种组织方式。商品标题、描述等内容通常在请求到达前已知,可以按商品预计算 KV。但以 100 万件商品、每件 100 个 token 为例,Qwen3-8B 的完整商品缓存约需 27.5 TB,难以在每个节点保存一份。图 5 显示,商品热度呈长尾分布,为区别处理热门商品和长尾商品提供了依据。


图 5 : Amazon、Yelp 和 Goodreads 的商品热度分布。横轴为商品评论数,采用对数刻度;纵轴为累积比例。

RcLLM 将最热门的 0.1% 商品复制到所有实例,其余商品按共现关系分片。具体而言,系统构建商品关联图,再通过 METIS 划分,使经常共同出现在请求中的商品尽可能存放在同一节点。这样既降低单节点存储压力,也提高一次请求在本地找到所需缓存的机会。

3.2 缓存感知调度兼顾命中率与负载

请求到达后,Global Scheduler 为各节点计算 affinity score,综合预计商品缓存命中率与节点空闲程度,选择执行位置。缓存命中能够减少计算,负载均衡则避免请求在热门节点长时间排队,两者共同决定实际延迟。

执行节点随后读取匹配的历史和商品 KV,缓存缺失的内容即时计算。任务指令以及时间戳、商品 ID 等请求专属信息仍完整计算,以保留当前推荐任务的要求。系统通过逻辑位置到物理缓存块的映射组织非连续 KV,减少额外拼接拷贝。

3.3 选择性重计算保留关键上下文

缓存复用会引入近似,RcLLM 因此把部分计算预算留给影响排序的关键位置。图 6 中的 Attention 分布显示,商品内部主要关注相邻内容,同时存在少量跨越较长距离的关注位置。这些 heavy hitters 包括部分任务指令和分隔符,承担着连接全局上下文的作用。


图 6 : Qwen3-8B 的平均 Attention 分布。局部结构与少量全局关注位置,为选择性重计算提供依据。

RcLLM 在首个 decoder layer 执行完整 Attention,识别重要 token,并结合缓存偏差确定需要更新的历史状态;随后各层对关键位置和局部窗口精确重计算,其余部分继续复用。位置对齐与重计算共同修正跨上下文复用的误差。系统还让首层计算与 CPU 到 GPU 的缓存传输重叠,降低数据搬运带来的等待。

四、主要实验结果 4.1 实验设置

实验采用 Amazon Reviews、Yelp 和 Goodreads 三个公开数据集。分布式服务性能通过扩展 Vidur 推理仿真器评估,模拟缓存放置、请求排队、PCIe 传输和 100 Gbps 网络;推荐质量则通过基于 Hugging Face Transformers 的 Qwen3-8B 功能原型验证。

默认集群包含 40 个实例。Qwen3-8B 每实例配置一张 80 GB A100,Qwen-72B 每实例使用四张 A100 进行 Tensor Parallelism。延迟对比包括完整重计算 Full-Recompute 与前缀复用 Prefix-Cache;质量评估还比较了 CacheBlend 和 EPIC。

4.2 两种模型均获得更低的首 token 延迟

在默认配置下,相比 Prefix-Cache,Qwen3-8B 的中位数(P50)TTFT 获得 1.5–2.2 倍加速,尾部(P99)获得 1.4–2.4 倍加速。图 7 中,RcLLM 的曲线整体向左移动,说明在三个数据集上,更多请求可以更早开始生成结果。


图 7 : Qwen3-8B 的 TTFT 累积分布,集群为 40 实例。各子图依次为 Amazon、Yelp 和 Goodreads,曲线越靠左表示延迟越低。

对 Qwen-72B,P50 与 P99 的加速比分别为 1.3–2.1 倍和 1.4–2.8 倍,如图 8 所示。这里的加速比为基线延迟除以 RcLLM 延迟。收益来自更广的复用范围:在实验轨迹中,固定前缀仅占输入的约 7%–10%,候选商品与用户历史则占据主要部分,RcLLM 能够利用这两部分中的重复内容。


图 8 : Qwen-72B 在 40 实例配置下的 TTFT 累积分布。各子图依次为 Amazon、Yelp 和 Goodreads。

4.3 通过重计算预算调节推荐质量

更快的响应也需要保留有效的排序信息。实验采用 HR 衡量推荐命中情况,并用 NDCG 同时考虑相关商品的排序位置,两项指标均为数值越高越好。在 Amazon 上,当商品与历史 token 均重计算 30% 时,相比 Full-Recompute,RcLLM 的 HR@10 从 0.700 提升至 0.740,NDCG@10 从 0.403 提升至 0.438。

不同数据集对复用的敏感程度有所差异。在 Goodreads 上,商品与历史 token 分别重计算 30% 和 20% 时,HR@10 与完整计算相同,均为 0.776;NDCG@20 从 0.484 降至 0.468,保留约 96.7%。在 Yelp 上,两类 token 分别重计算 40% 和 50% 时,HR@10 从 0.698 提升至 0.744,但 NDCG@10 从 0.477 降至 0.407,说明命中率和排序质量需要分别观察。

这些结果表明,重计算预算需要结合数据集与目标指标选择。图 9 以 Amazon 为例展示了两类重计算比例的联合影响,为选择质量与效率之间的合适配置提供依据。


图 9 : Amazon 上的重计算预算与 NDCG@20 差异。横轴为商品 token 比例,纵轴为历史 token 比例;格内数值表示 RcLLM 减去 Full-Recompute 的结果,越高越好。

4.4 分布式缓存兼顾扩展性与存储成本

当实例数从 1 增加到 100,RcLLM 仍保持对 Prefix-Cache 的加速优势。图 10 展示了不同集群规模下的结果;Qwen3-8B 与 Qwen-72B 的 P99 最高分别获得约 3.5 倍和 9.5 倍加速,这些最高值对应扩展性实验中的特定配置。


图 10 : 不同集群规模下的 TTFT 加速比。上排为 Qwen3-8B,下排为 Qwen-72B,各列依次为 Amazon、Yelp 和 Goodreads。虚线表示 Prefix-Cache。

分片带来的另一项收益是单节点存储下降。图 11 显示,Amazon 每实例保存的商品 token 数由单实例时约 6600 万降至 40 实例时约 170 万。同时,缓存仍保持较高命中率,说明相似度感知放置能够在减少存储复制的同时保留局部性。


图 11 : 集群扩展时的缓存命中率与存储占用。左图为商品缓存命中率;右图为每实例保存的商品 token 数,以千 token 为单位并采用对数刻度。

4.5 调度与重计算共同影响服务效率

图 12 对比了只关注命中率的 Hit-Only、只关注负载的 Load-Only 和 RcLLM 的联合调度策略。随着请求强度增加,Hit-Only 容易在热门节点产生排队,Load-Only 则可能增加缓存缺失。综合考虑两者的 RcLLM 在不同负载下获得最低或接近最低的平均 TTFT。


图 12 : 不同调度策略的平均 TTFT,集群为 10 实例。三个子图依次为 Amazon、Yelp 和 Goodreads,横轴 QPS 表示每秒请求数。

重计算预算同样影响响应速度。图 13 显示,预算增大时 TTFT 曲线逐步右移。在 Amazon、Qwen3-8B 和 40 实例配置下,商品与历史均重计算 30% 时,P90 TTFT 从 Prefix-Cache 的 0.38 秒降至 0.18 秒。结合质量评估,系统可以按目标指标选择预算,在保留必要上下文计算的同时获得延迟收益。


图 13 : 重计算比例对 TTFT 的影响。Qwen3-8B、40 实例配置,历史与商品采用相同重计算比例;灰色虚线为 Prefix-Cache。

五、总结与展望

RcLLM 将生成式推荐中的重复内容转化为可共享的计算资源:通过语义历史缓存复用相似表达,通过商品缓存分片支撑大规模目录,再以缓存感知调度和选择性重计算兼顾执行效率与推荐质量。实验表明,这一设计能够在不同数据集、模型规模和集群配置下减少冗余 Prefill,降低首 token 延迟。

这项工作的启示是,面向大模型应用的推理优化,可以进一步利用任务输入自身的结构。对于具有大量重复商品与相似历史表达的推荐场景,内容复用、数据放置和请求调度需要协同设计,才能把缓存命中转化为实际的服务效率。

-The End-

扫码观看!

本周上新!

“AI技术流”原创投稿计划

TechBeat是由将门创投建立的AI学习社区(www.techbeat.net)。社区上线700+期talk视频,3000+篇技术干货文章,方向覆盖CV/NLP/ML/Robotis等;每月定期举办顶会及其他线上交流活动,不定期举办技术人线下聚会交流活动。我们正在努力成为AI人才喜爱的高质量、知识型交流平台,希望为AI人才打造更专业的服务和体验,加速并陪伴其成长。

投稿内容

// 最新技术解读/系统性知识分享 //

// 前沿资讯解说/心得经历讲述 //

投稿须知

稿件需要为原创文章,并标明作者信息。

我们会选择部分在深度技术解析及科研心得方向,对用户启发更大的文章,做原创性内容奖励

投稿方式

发送邮件到

yimingzhang@thejiangmen.com

或添加工作人员微信(aceyiming)投稿,沟通投稿详情

关于我“门”

将门是一家以专注于数智核心科技领域的新型创投机构,也是北京市标杆型孵化器。 公司致力于通过连接技术与商业,发掘和培育具有全球影响力的科技创新企业,推动企业创新发展与产业升级。

将门成立于2015年底,创始团队由微软创投在中国的创始团队原班人马构建而成,曾为微软优选和深度孵化了126家创新的技术型创业公司。

如果您是技术领域的初创企业,不仅想获得投资,还希望获得一系列持续性、有价值的投后服务,欢迎发送或者推荐项目给我“门”:

bp@thejiangmen.com


点击右上角,把文章分享到朋友圈

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

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.

相关推荐
热点推荐
哈维力挺!巴萨低调王牌被严重低估!弗里克变阵挖出瑰宝

哈维力挺!巴萨低调王牌被严重低估!弗里克变阵挖出瑰宝

澜归序
2026-11-10 10:11:57
中方尚未批准约60万枚英伟达先进芯片进口,外交部回应

中方尚未批准约60万枚英伟达先进芯片进口,外交部回应

澎湃新闻
2026-11-09 16:58:04
赴泰失联女子已被转卖!两国警方立案,头目撂狠话,给钱也不放人

赴泰失联女子已被转卖!两国警方立案,头目撂狠话,给钱也不放人

月光作笺a
2026-11-10 04:55:20
辣眼睛!多名女生举证河南财大教师涉嫌性骚扰,聊天记录曝光,网友:看样子赵教授得手很多次,不然也不敢这样明目张胆的给漂亮女生发信息

辣眼睛!多名女生举证河南财大教师涉嫌性骚扰,聊天记录曝光,网友:看样子赵教授得手很多次,不然也不敢这样明目张胆的给漂亮女生发信息

火山詩话
2026-11-10 09:12:09
瞎子阿炳:抽大烟,逛青楼,败光家产,凭啥被称为中国的贝多芬?

瞎子阿炳:抽大烟,逛青楼,败光家产,凭啥被称为中国的贝多芬?

月下守候
2026-11-09 12:38:27
黄仁勋离婚,损失811亿美元,女方成为全球第四大女富豪

黄仁勋离婚,损失811亿美元,女方成为全球第四大女富豪

深度人物说
2026-11-09 22:50:34
中国航天憋了个大招!长征九号高达110米,2030年将迎关键一飞!

中国航天憋了个大招!长征九号高达110米,2030年将迎关键一飞!

像梦一场a
2026-11-09 22:09:36
大快人心!成都大清算:主帅下课,董事长总经理离职,球迷公开信要求彻查内耗

大快人心!成都大清算:主帅下课,董事长总经理离职,球迷公开信要求彻查内耗

锐评利物浦
2026-11-10 08:46:46
剖析项英牺牲真相:为何多次拒绝毛泽东北上命令?因自己两个私心

剖析项英牺牲真相:为何多次拒绝毛泽东北上命令?因自己两个私心

阿胡
2025-04-20 11:05:46
27岁女孩网恋奔现遭转卖:花了一二十万买了她,不会放她回去

27岁女孩网恋奔现遭转卖:花了一二十万买了她,不会放她回去

江山挥笔
2026-11-10 09:34:51
前列腺与花生的关系,建议:想保护前列腺健康,最好少吃4种坚果

前列腺与花生的关系,建议:想保护前列腺健康,最好少吃4种坚果

医学科普汇
2026-11-09 20:45:11
台北、新北输,嘉义、彰化输,新竹县也输了,国民党准备找战犯?

台北、新北输,嘉义、彰化输,新竹县也输了,国民党准备找战犯?

今夜繁星坠落
2026-11-10 08:25:36
努力在趋势面前一文不值,40岁女老板做了十几年服装生意,十几块打底衫往年每天卖千件,今年怎么努力也卖不动,太心累

努力在趋势面前一文不值,40岁女老板做了十几年服装生意,十几块打底衫往年每天卖千件,今年怎么努力也卖不动,太心累

捣蛋窝
2026-11-09 00:16:31
广重:一句"宁入广重不考清华",胜过所有排名

广重:一句"宁入广重不考清华",胜过所有排名

朗威谈星座
2026-11-10 05:23:34
隐退后的费玉清和张菲

隐退后的费玉清和张菲

手工制作阿歼
2026-10-13 04:03:54
刘欢在2008年北京奥运会上的着装至今仍有争议,因为那是一个举世瞩目的盛会,刘欢却只是穿了一件普通T恤就开始表演了!

刘欢在2008年北京奥运会上的着装至今仍有争议,因为那是一个举世瞩目的盛会,刘欢却只是穿了一件普通T恤就开始表演了!

财经要参
2026-11-01 16:00:05
炸裂!日本妻子拍AV挽救婚姻,瞒夫出道老公崩溃,如今丈夫成粉丝

炸裂!日本妻子拍AV挽救婚姻,瞒夫出道老公崩溃,如今丈夫成粉丝

悠悠说世界
2026-08-17 13:48:27
16GB+512GB,目前“接近完美”的5款手机,公认几乎“零差评”

16GB+512GB,目前“接近完美”的5款手机,公认几乎“零差评”

叮当当科技
2026-11-09 01:08:19
白宫幕僚长卸任内幕曝光,与特朗普爆发激烈争执,当众大发雷霆

白宫幕僚长卸任内幕曝光,与特朗普爆发激烈争执,当众大发雷霆

金戈远望
2026-11-09 17:14:35
中大本科生反AI逆向项目火了!让黑客Token账单原地暴涨300倍

中大本科生反AI逆向项目火了!让黑客Token账单原地暴涨300倍

量子位
2026-11-10 06:06:17
2026-11-10 10:28:49
将门创投 incentive-icons
将门创投
加速及投资技术驱动型初创企业
2538文章数 597关注度
往期回顾 全部

科技要闻

最强Claude来了!8天从零写出电脑系统

头条要闻

牛弹琴:果然是一场闹剧 这个世界就没有特朗普做不到

头条要闻

牛弹琴:果然是一场闹剧 这个世界就没有特朗普做不到

体育要闻

大连足球这些年的起落,都从他身边经过

娱乐要闻

黄仁勋与黄乐合照曝光,吃饭氛围暧昧

财经要闻

72岁王健林重回“一号位”

汽车要闻

雨中的埃安Ray7,有点上头

态度原创

健康
亲子
游戏
艺术
公开课

痘痘长到25岁还没完?查查内分泌

亲子要闻

幼儿园3年减少5.73万所

《GTA66》联名定制跑车标价百万英镑开售 独一无二

艺术要闻

点开前以为只是玫瑰,点开后:这画得也太美艳了!

公开课

李玫瑾:为什么性格比能力更重要?

无障碍浏览 进入关怀版