vToken:面向可回收 KV 缓存的 Token 级虚拟化方案
评分
77提出 token 级虚拟化层,弥补块管理与 token 级逐出之间的空白,属于有明确增量的系统设计。
实现完整、实验覆盖多策略与多指标,但缺少与更复杂内存管理方案的横向对比。
兼容现有内核并显著降低策略集成成本,便于在 vLLM 生产环境中快速落地。
KV 缓存优化是当前大模型部署热点,论文方案紧贴社区关注的高并发与内存效率问题。
来自 arXiv AI/CL/LG 论文,作者为 vLLM 生态贡献者,但尚未看出处同行评审与复现验证记录。
基础分 85 × T2 信源权重 ×0.9 = 77
精选线 70已入选精选
精选理由
该论文提出系统级的缓存管理创新,实验覆盖多种逐出策略且收益明显,填补了块管理与 token 粒度逐出之间的空白。
AI 点评
「这篇论文最值得关注的是用虚拟化思路化解粒度错配,而不是单纯堆新缓存算法。对做推理服务底层优化的人,它给出了一个低侵入、可复用的抽象层,可能会影响后续 KV 管理器设计。」
速览
大模型推理中 KV 缓存随序列长度和批大小增长,成为关键内存瓶颈。PagedAttention 以固定大小块管理内存,而现有逐出算法按 token 粒度操作,导致块内碎片化,大量已分配内存无法回收。vToken 提出轻量级 token 级虚拟化层,通过 token 表间接寻址解耦逻辑生命周期与物理块放置,并异步重打包实现物理回收,兼容 PagedAttention 内核与 CUDA Graph。在 vLLM 中实现后,相比 Naive-Evict 基线,每请求保留 KV 块减少 27.2% 至 72.3%,SLA 约束吞吐最高提升 1.37 倍,最大并发扩展 2 倍,且集成成本从 500 多行降至 50 行以内。
- vToken 通过 token 表间接寻址,将逻辑 token 存活状态与物理块放置解耦,解决块内碎片导致的内存不可回收问题。
- 异步重打包活动 token,实现物理内存回收,同时保持 PagedAttention 内核与 CUDA Graph 兼容。
- 在 vLLM 中集成 H2O、Random、Scissorhands 等逐出策略,每请求保留 KV 块减少 27.2%–72.3%,SLA 约束吞吐最高提升 1.37 倍。
- 策略集成代码量从 500 多行降至 50 行以内,受限 active-KV 预算下最大并发扩展可达 2 倍。
正文
本文为 AI 生成的中文精读,原文请点「查看原文」
相关条目
85
精选Setoka:面向异构数据个性化Agent的层级用户理解基准
这篇论文非常值得关注,尤其对于从事个性化Agent、对话系统和记忆增强AI的研究者和工程师。它不仅仅是一个基准,而是从心理学出发定义了用户理解的层级,揭示了现有系统在高层理解上的系统性失败。如果你在设计基于记忆的Agent,这篇文章会提醒你记忆系统需要支持跨时间、跨来源的抽象推理,而不仅仅是检索。
85
精选BAND:高维分布估计的贝叶斯网络回归方法——打破维度诅咒
这篇论文值得所有从事高维统计、概率建模和生成式AI的研究者关注。它从理论上给出了一个简单却有力的答案:稀疏结构能打破维度诅咒,而贝叶斯网络恰好提供了自然框架。相比VAE/GAN这类黑箱方法,BAND可解释性强且收敛有保证,尤其适合工业界对可靠置信区间的需求。如果你是研究时间序列预测或风险管理的,这篇论文可能会改变你处理高维密度估计的方式。
84
精选循环一致与不确定性感知的托卡马克边缘等离子体神经代理模型
这篇论文值得关注,因为它展示了如何将先进的AI技术(循环一致性、不确定性量化)直接用于解决核聚变中的核心计算瓶颈。对于聚变领域的研究人员和工程师,这是一个能在不牺牲精度的前提下将模拟速度提升数个数量级的实用工具;对于AI从业者,它提供了一个科学应用中自监督学习和逆问题求解的优秀案例。
83
精选等变学习可迁移的三维经典密度泛函
这篇论文最值得关注的点在于,它把机器学习和统计力学的核心难题直接对接,用等变网络学出了一个可迁移的密度泛函,而且不依赖任何标签。对于做液体物理或软物质的同行,这可能是新的研究范式:以后是不是不用再针对每个体系从头做模拟了?对于AI研究者,这也展示了如何将物理对称性和变分原理嵌入模型,而不是单纯拟合数据。建议重点读一下方法部分,尤其是等变性和变分一致性的实现方式,那才是真正的精髓。
83
精选c-Rectified Flow 的计算与统计保证
这篇论文真正把整流流从“经验好用”推向了“理论可控”,对做生成模型底层算法的人尤其重要。如果你在调 FLUX 或 SD3 这类管线,理解普通整流流在协方差不可交换时可能失效,能帮你避开一些隐蔽的训练陷阱。
83
精选从高保真UMI数据学习可部署操作策略
这篇论文最亮眼的地方是证明了纯无机器人数据也能达到与真实遥操作相当的性能,彻底打破了后训练阶段必须依赖少量真实数据的惯例。对于想降低硬件成本的研究组和工业机器人应用开发者来说,HiFi-UMI的数据生产方案和开源数据集是直接可用的资源。建议关注其头戴式SLAM和微秒同步的工程细节,这些是精度达标的关键。
83
精选资本市场LLM可靠性评分:从“看似合理”到“可银行化”
这篇论文对金融圈和AI圈都很有价值:它给出了一个可操作的、面向实际工作流的评估框架,比泛泛的基准更贴近真实需求。如果你在金融科技公司做LLM落地,或者负责模型的“可解释性”和“可审计性”,这篇值得细读。决策有用性维度的发现尤其值得关注——评审们一致认为这是关键,但不同模型表现差异最大。
83
精选优化器状态应放在哪里?面向内存高效MoE训练的分层状态分配
这篇论文对训练MoE模型的团队非常有价值:它证明了通过精细的优化器状态分配,可以在不牺牲精度的情况下大幅降低显存占用,甚至还能提升性能。尤其适合那些受限于40GB或80GB GPU显存的实验室或企业。值得仔细阅读其分层策略和动量保留的消融分析,可为实际训练框架(如DeepSpeed、Megatron)提供优化方向。