news 2026/8/21 18:24:47

论文复现:KIVI: 一种无需调优的2位KV缓存量化算法

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
论文复现:KIVI: 一种无需调优的2位KV缓存量化算法

论文复现:KIVI: 一种无需调优的2位KV缓存量化算法

1. 论文标题

KIVI:一种无需调优的2位KV缓存量化算法,用于在LLM推理中实现更大的批处理大小

2. 论文内容:

本论文针对大语言模型(LLM)推理中KV缓存成为内存和速度瓶颈的问题,提出了一种名为KIVI的2位KV缓存量化方法。研究发现,Key缓存中存在少量通道具有极大值,应采用按通道量化;而Value缓存无此模式,应采用按令牌量化。KIVI采用分组策略,将KV缓存分为分组部分(量化存储)和残差部分(全精度存储),在解码时通过分块矩阵乘法结合。实验表明,KIVI可将Llama、Falcon和Mistral等模型的KV缓存压缩至2位,减少2.6倍峰值内存使用,支持最多4倍的批处理大小增加,带来2.35-3.47倍的吞吐量提升,同时在多种生成任务上保持接近全精度的性能。

3. 完整代码实现及解释

importtorchimporttorch.nnasnnimporttorch<
版权声明: 本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!
网站建设 2026/8/21 11:55:21

【同步挤压变换和重新分配方法】【时频分析(TFA)】解决海森堡不确定性原理(HUP)提供新的方法附Matlab代码

✅作者简介&#xff1a;热爱科研的Matlab仿真开发者&#xff0c;擅长数据处理、建模仿真、程序设计、完整代码获取、论文复现及科研仿真。&#x1f34e; 往期回顾关注个人主页&#xff1a;Matlab科研工作室&#x1f34a;个人信条&#xff1a;格物致知,完整Matlab代码及仿真咨询…

作者头像 李华
网站建设 2026/8/21 11:55:13

2024金融AI智能体投资决策的技术趋势:架构师的预判与布局

2024金融AI智能体投资决策技术趋势&#xff1a;架构师的预判与布局 副标题&#xff1a;从技术演进到业务落地的实战指南 摘要/引言 问题陈述 传统金融投资决策系统面临三大核心挑战&#xff1a; 规则僵化&#xff1a;依赖人工制定的规则引擎&#xff0c;无法适应市场环境的…

作者头像 李华
网站建设 2026/8/21 11:55:12

2026养老新政全方位落地,银发生活迎来新机遇!

2026年伊始&#xff0c;养老新政全面落地&#xff0c;从保障体系、服务供给、科技赋能到产业生态实现系统性升级。 既为 3 亿老年人筑牢幸福底线&#xff0c;也为银发经济开辟万亿级新蓝海。以下从政策核心、生活变革、产业机遇三方面展开深度解读。 一、新政核心&#xff1a;…

作者头像 李华
网站建设 2026/8/7 15:43:41

C语言对话-7.方形尖塔

sebastian 翻译 “下周&#xff0c;队伍就要下去了&#xff0c;我想我们会在一周后到那儿。”珍妮总结道&#xff0c;“他们仍在水晶石塔上方挖掘&#xff0c;但没有什么新发现。进展很缓慢&#xff0c;我们不打算进行这样的探查。这超出了我们的探索和采矿的命令。弗兰奈尔仍然…

作者头像 李华
网站建设 2026/8/1 22:18:06

漏洞挖掘实战全指南:从技能构建到落地复盘(CSDN+Word适配版)

漏洞挖掘实战全指南&#xff1a;从技能构建到落地复盘&#xff08;CSDNWord适配版&#xff09; 漏洞挖掘是网络安全攻防体系的核心环节&#xff0c;也是企业提前化解安全威胁、筑牢防护防线的关键手段。它并非单纯的“工具操作”&#xff0c;而是融合技术能力、攻击思维、业务…

作者头像 李华