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

相关内容

热门资讯

国产超节点商业化大幕拉开 记者 郑晨烨 在7月17日至20日举行的2026世界人工智能大会(WAIC)上,超节点成为全场焦点。...
原创 v... 能够首发搭载最新的旗舰处理器,会自带较高的热度,9月份首发第五代骁龙8至尊版的小米17系列是这样,1...
灿勤科技获得发明专利授权:“一... 证券之星消息,根据天眼查APP数据显示灿勤科技(688182)新获得一项发明专利授权,专利名为“一种...
原创 怎... 今天刚看到的最新消息:高通已正式向客户发送通知函,宣布将对包括骁龙8 Elite Gen 6与骁龙8...
俄罗斯“联盟MS-28”飞船搭... 央视新闻消息,央视记者当地时间7月26日从美国方面获悉,一支由美国和俄罗斯航天员组成的联合乘组在完成...
原创 用... 全球跨文化实验做过一件挺有意思的事——让不同肤色、不同国家、完全没有文化交集的人群,给同一批面孔打分...
国内首款:TP200大型多功能... IT之家 7 月 26 日消息,国内首款按适航要求正向研制的吨级多功能长航时民用无人机 —— TP2...
原创 单... 最新2026年第29周国内单款手机销量排行出炉了,这次榜单前面几位是有所变化的。 特别是华为畅享90...
罗永浩吐槽电视长辈模式不好用:... IT之家 7 月 26 日消息,罗永浩昨天在微博发文,控诉电视机厂商的产品经理。 罗永浩表示,他最近...
SK集团会长崔泰源:Anthr... IT之家 7 月 26 日消息,据彭博社当地时间 24 日报道,SK 集团会长崔泰源近日在美国加利福...