颠覆!NVIDIA发明新技术KVTC:内存使用量缩减20倍
创始人
2026-03-23 13:44:20
0

快科技3月22日消息,NVIDIA研究人员推出一项全新技术KVTC(KV快取转换编码),能把大型语言模型(LLM)追踪对话历史的内存用量,最高缩减20倍,而且不用修改模型本身。

这一突破有望解决大型语言模型长对话推理时的内存不够用问题,大大降低企业使用AI的硬件成本,同时还能把模型首次生成回应的时间,最高提速8倍。

简单来说,KVTC技术的核心就是压缩大型语言模型背后的KV缓存——它相当于AI模型的“短期记忆”。我们可以把KV缓存理解成学生记笔记:模型处理对话时,会把关键信息(也就是Key和Value)记下来,下次生成回应时,不用从头重新计算整段对话,响应速度就能大幅提升。

但问题是,对话越长,这份“笔记”就越大,甚至会膨胀到几个GB,占用大量GPU内存,反而拖慢模型运行、限制其处理能力。

NVIDIA资深深度学习工程师Adrian Lancucki表示:“大型语言模型进行推论时,性能瓶颈往往不在运算能力,而在GPU内存。”那些暂时不用的KV缓存,会一直占用宝贵的GPU资源,逼得系统只能把它们转移到CPU内存或硬盘里,这样不仅会增加数据传输的负担,还可能出现新的卡顿问题,这些额外成本最终也会体现在企业的使用费用中。

和现有压缩技术相比,KVTC没有那些明显的局限,它借鉴了我们熟悉的JPEG图片压缩思路,通过“主成分分析、自适应量化、熵编码”三个简单步骤,就能实现高效压缩。

更方便的是,这项技术不用改动模型的核心设置和代码,属于“非侵入式”设计,企业拿来就能快速部署。它的核心优势是,能抓住KV缓存“数据高度相关”的特点,在保留关键信息的同时,去掉冗余数据,而且解压时可以分块、逐层进行,不会影响模型实时回应。

多轮测试显示,KVTC的表现远超现有主流方法。在参数量从15亿到700亿的多种模型(包括Llama 3系列、R1-Qwen 2.5等)上,即便将内存压缩20倍,模型准确率也几乎不受影响,损失不到1%,与未压缩时相差无几;而传统压缩方法仅压缩5倍,就会出现明显的准确率下降。

另外,在H100 GPU上处理8000个Token的提示时,不使用KVTC需要3秒才能生成第一个回应,使用后仅需380毫秒,提速整整8倍。

需要注意的是,KVTC更适合长对话、多轮互动场景,比如编程助手、迭代式代理推理等,若对话较短,很难发挥其压缩价值。

目前,NVIDIA正计划将这项技术整合进Dynamo框架的KV块管理器,使其能与vLLM等主流开源推论引擎兼容。

业内人士认为,随着大型语言模型可处理的对话长度不断增加,KVTC这类标准化压缩技术,未来可能会像视频压缩一样普及,助力AI更广泛地落地应用。

相关内容

热门资讯

原创 华... 六年了。 2020年,台积电断供,麒麟芯片一度成了“绝唱”。那一年,多少人断言:华为手机,完了。 可...
苹果保密工作白做了 手机壳厂提... 快科技9月8日消息,随着北京时间9月10日凌晨苹果发布会的临近,关于其首款折叠屏iPhone的命名讨...
重庆获批建设国家人工智能产业创... 央广网重庆9月8日消息(记者陈静 实习记者冯汀汀)9月8日,记者从重庆市经济和信息化委员会获悉,重庆...
风云气象卫星驰援尼泊尔,为跨境... 中国和尼泊尔边境吉隆-热索瓦口岸严重泥石流灾害发生后,中国气象局启动风云气象卫星国际用户防灾减灾应急...
关于二层全网管交换机的带机量误... 在网络设计与设备选型中,“带机量”常被简单理解为端口数量的叠加,这种认知误区可能导致网络规划与实际负...
折叠屏何以成为手机高端化新引擎 证券时报记者 王小伟 本周,华为、小米相继发布折叠屏旗舰新机,苹果秋季发布会预计也将发布首款折叠屏产...
消息称原小米L3负责人王乃岩投... IT之家 9 月 8 日消息,据“21 世纪经济报道”,原小米智能驾驶 L3 技术负责人王乃岩已投身...
机器人进后厨绝非“机器替人” 来源:证券日报 随着智能化设备加速向线下门店渗透,可以炒菜、煮面、饮品制作的餐饮机器人正从营销噱头转...
中国科学家领衔国际团队找到中子... (来源:新华日报) 本报讯 (记者 蔡姝雯) 近日,由中国科学院紫金山天文台、国家天文台、清华大学等...
丽晖智能取得玻璃热切系统专利,... 国家知识产权局信息显示,浙江丽晖智能装备有限公司取得一项名为“玻璃热切系统和热切车间”的专利,授权公...