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

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.

相关推荐
热点推荐
杭州临平发生刑案致2死,嫌犯畏罪自杀,涉事别墅区物业:推测嫌疑人跟随业主进入小区;周边商户:一名死者30岁左右,平时对人很客气

杭州临平发生刑案致2死,嫌犯畏罪自杀,涉事别墅区物业:推测嫌疑人跟随业主进入小区;周边商户:一名死者30岁左右,平时对人很客气

台州交通广播
2026-11-10 02:09:01
两人已拍拖多年,早被监听发现,事发后黄仁勋很果断,黄乐很聪明

两人已拍拖多年,早被监听发现,事发后黄仁勋很果断,黄乐很聪明

朗威谈星座
2026-11-09 20:50:54
黄仁勋为了业绩都跟客户睡觉了,我们为什么不可以?

黄仁勋为了业绩都跟客户睡觉了,我们为什么不可以?

News脑洞
2026-11-09 15:22:46
蓉城回成都!大巴接机+红毯铺地,球员有说有笑,球迷:欢迎亚军

蓉城回成都!大巴接机+红毯铺地,球员有说有笑,球迷:欢迎亚军

二疯说球
2026-11-09 22:18:35
再见了,陈坤,再见了,黄晓明,中国演艺圈迎“新四大小生”时代

再见了,陈坤,再见了,黄晓明,中国演艺圈迎“新四大小生”时代

皮皮电影
2026-11-09 09:26:00
湖南湘潭一女子买早餐时余额不足,银行卡里近26万不翼而飞,当事人:短信提醒被取消,余额六天内分多笔被转走,银行:疑遭电诈,建议报案

湖南湘潭一女子买早餐时余额不足,银行卡里近26万不翼而飞,当事人:短信提醒被取消,余额六天内分多笔被转走,银行:疑遭电诈,建议报案

台州交通广播
2026-11-10 08:38:36
乒协公布混团世界杯参赛名单

乒协公布混团世界杯参赛名单

界面新闻
2026-11-10 09:38:12
莫斯科“中国城”等地数百年轻人被捕!还收到征兵传票

莫斯科“中国城”等地数百年轻人被捕!还收到征兵传票

项鹏飞
2026-11-09 19:20:37
一通电话叫来黄仁勋,黄乐到底是谁?

一通电话叫来黄仁勋,黄乐到底是谁?

凤凰网财经
2026-11-10 00:03:02
石破天惊的发现,中国汉字已正式推至7000年前,别再纠结夏朝了

石破天惊的发现,中国汉字已正式推至7000年前,别再纠结夏朝了

小豫讲故事
2026-11-09 12:10:25
特朗普质疑中期选举公正性:部分提前投票的批量票源规模大到离谱,无论多少人去现场投票,都根本不可能翻盘,问题全都出在投票机器上

特朗普质疑中期选举公正性:部分提前投票的批量票源规模大到离谱,无论多少人去现场投票,都根本不可能翻盘,问题全都出在投票机器上

极目新闻
2026-11-10 07:38:51
搬家公司9公里收费13900元,不给钱就扣货,威胁“你家就在这跑不掉”……警方打掉两个家族式“套路搬”团伙,近百人落网

搬家公司9公里收费13900元,不给钱就扣货,威胁“你家就在这跑不掉”……警方打掉两个家族式“套路搬”团伙,近百人落网

封面新闻
2026-11-09 16:20:46
黄乐是怎么被盯上的?从一箱没拆封的英伟达芯片说起

黄乐是怎么被盯上的?从一箱没拆封的英伟达芯片说起

迷世书童
2026-11-09 15:19:32
黄克诚带队转移,发现警卫员总走在最前,想起侦察员那句提醒

黄克诚带队转移,发现警卫员总走在最前,想起侦察员那句提醒

纪史行者
2026-11-08 08:10:16
女子和情夫山上发生关系,嫌情夫不肯开房太小气,两人吵起来,女子再次被发现身上盖满石块

女子和情夫山上发生关系,嫌情夫不肯开房太小气,两人吵起来,女子再次被发现身上盖满石块

丫头舫
2026-11-09 22:14:32
全球首个长内容AI流媒体「Glanze」,让一万个平行宇宙找到落点

全球首个长内容AI流媒体「Glanze」,让一万个平行宇宙找到落点

极客电影
2026-09-11 18:02:52
黄仁勋台北爱情故事

黄仁勋台北爱情故事

凤凰网财经
2026-11-09 16:36:34
性生活是不是人的刚需?网友说:男性性生活是刚需,女性不是!

性生活是不是人的刚需?网友说:男性性生活是刚需,女性不是!

黯泉
2026-11-09 17:39:59
妻子被迫后喜欢上偷情,情夫是邻居,2012年丈夫怒杀妻子情人

妻子被迫后喜欢上偷情,情夫是邻居,2012年丈夫怒杀妻子情人

汉史趣闻
2026-11-09 10:00:08
俄方长期将战损列为国家机密,黑客却泄露出记录:近40万俄罗斯人在俄乌冲突中丧生

俄方长期将战损列为国家机密,黑客却泄露出记录:近40万俄罗斯人在俄乌冲突中丧生

14号观星辰
2026-11-09 18:14:49
2026-11-10 11:04:49
将门创投 incentive-icons
将门创投
加速及投资技术驱动型初创企业
2538文章数 597关注度
往期回顾 全部

科技要闻

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

头条要闻

陕西一民办小学卖惨骗捐被查封 女校长等多人被抓

头条要闻

陕西一民办小学卖惨骗捐被查封 女校长等多人被抓

体育要闻

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

娱乐要闻

整个内娱为了她俩“画地为牢”?

财经要闻

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

汽车要闻

雨中的埃安Ray7,有点上头

态度原创

家居
房产
游戏
艺术
公开课

家居要闻

2026建博会(广州) 公装联探展交流活动

房产要闻

突发!陵水杀出重磅宅地!

速通玩家用无限跳突破《塞尔达:时之笛 重制版》

艺术要闻

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

公开课

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

无障碍浏览 进入关怀版