AI圈报
论文研究精选

Stochastic KV Routing: 实现自适应深度方向的缓存共享

信息来源:Apple Machine Learning Research(RSS)·
原始标题:Stochastic KV Routing: Enabling Adaptive Depth-Wise Cache Sharing

内容摘要

为降低大语言模型推理时KV缓存的高昂内存开销,研究提出了一种沿模型深度维度优化的新方法。该方法通过随机KV路由,在Transformer模型的各层之间动态共享KV缓存,而非每层保留完整独立缓存。实验表明,在保持模型质量基本不变的前提下,该方法能将KV缓存的内存占用减少高达50%,为降低大模型服务成本提供了与现有时间轴压缩、淘汰技术正交的新优化路径。
内容分类AI 论文与研究
内容层级精选情报
发布时间(北京时间)
本站收录时间(北京时间)
信息来源Apple Machine Learning Research(RSS)
站内情报编号intel-961d8c42b4d486a2f33d6b97