颠覆!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更广泛地落地应用。

相关内容

热门资讯

锦评天下丨学AI,光靠报班研学... 2026年暑期,AI研学成为新潮流。市场覆盖范围从一线城市到县城,不少项目要价不菲,仍吸引大量家长报...
风雨砥初心 洪流显担当丨钦州全... 7月3日以来 今年首个登陆我国的第10号台风“美莎克” 给钦州带来明显风雨影响 两县两区和各单位、部...
OpenAI首款硬件:Code... IT之家 6 月 30 日消息,OpenAI 开发者官方账号今天(6 月 30 日)在 X 平台发布...
广西各地各部门闻“汛”而动——... 01台风致广西多地受灾,各部门全力抢险救援,转移群众,抢修电力。 以上内容由传播大模型和DeepSe...
隆之源取得帽盖成型模具专利,一... 国家知识产权局信息显示,厦门隆之源塑胶有限公司取得一项名为“一种帽盖的成型模具”的专利,授权公告号C...
2026年世界互联网大会数字丝... 7月6日,世界互联网大会国际组织在中国北京举行新闻发布会,介绍2026年世界互联网大会数字丝路发展论...
享界G9测试车完成2000km... 【CNMO科技消息】7月7日,华为智选车产品总监(享界系列)@penglay_磊 发文,分享了享界G...
“作为负责任的航天大国,我国正... 1908年6月30日,一颗直径约50米的天体在俄罗斯西伯利亚通古斯地区上空爆炸,瞬间摧毁了超2000...
深圳机器人公司,来到联合国 近日,联合国开源周(UN Open Source Week)在纽约联合国总部举行,来自120多个国家...
“健康九江”便民服务平台升级上... 大江新闻讯 记者 程曼华摄影报道:5月26日,记者从“健康九江”便民服务平台升级上线发布仪式上获悉,...