当前位置:首页 > 百科 > AI长文本推理省内存神器面世,省53%显存还更准 正文

AI长文本推理省内存神器面世,省53%显存还更准

来源:星锐云联资讯网   作者:休闲   时间:2026-07-17 05:34:14

AI大模型虽已深入生活,长文存还但在处理长文本时存在一个常被忽视的本推痛点:显存爆炸。当用户输入十万字级别的理省文档时,仅为了缓存上下文,内存就需要占用数十GB的神器省显显存。这不仅让普通消费级显卡无法承载,面世即便勉强运行,更准推理速度也会慢到难以接受。长文存还

近日,本推英属哥伦比亚大学与微软研究院联合发布了一种名为 SeKV的理省创新方法。该研究由微软研究院高级研究员、内存牛津大学博士、神器省显武汉大学校友何宇航(音)等人共同完成。面世

核心突破:低资源下的更准高精度

SeKV 方法在大幅降低显存占用的同时,保持了极高的长文存还回答准确率:

  • 显存减半:在 128K 长度的文本处理中,显存占用较完整缓存减少 53%以上。
  • 精度反超:在仅使用 1/10原有显存的条件下,其准确率比现有语义压缩方法平均高出近 6个百分点
  • 零训练成本:无需重新训练大模型,仅需增加一个参数占比低于 万分之五的小模块即可部署。

这意味着,拥有 24GB 显存的消费级显卡,原本仅能处理 100K 左右的长文本,应用 SeKV 后可轻松应对接近 300K的长文本需求。

(来源:相关论文)

痛点解析:传统缓存的线性增长困境

大模型在处理长文本时,通常会将每个词的键值对(Key-Value Pairs)缓存下来,以便生成回答时回溯上下文。这种机制虽然必要,却存在致命缺陷:

  1. 线性增长:缓存大小随文本长度线性增加。从一万字到十万字,显存占用从几GB激增至几十GB。
  2. 硬件瓶颈:显存增长速度远滞后于文本长度增长,导致硬件成本急剧上升。

以往的解决方案多为“做减法”,存在明显短板:
* 丢弃策略:仅保留最近或看似重要的内容,导致关键信息永久丢失,模型只能“瞎猜”。
* 摘要压缩:将缓存压缩为摘要,但压缩决策一旦做出便不可逆,后续无法还原细节。

SeKV 的创新路径:语义分段与动态检索

SeKV 另辟蹊径,采用“摘要驻留显存 + 详情存储CPU”的双层架构:

  1. 语义切分:将长文本按语义边界切分为段落。
  2. 显存优化:每个段落仅保留一个浓缩摘要置于显存中,用于快速检索和锚点定位。
  3. CPU存储:原文详细内容经压缩后存入 CPU 内存。
  4. 按需加载:当模型识别到某段落与当前问题相关时,再从 CPU 调取详细数据进行推理。

类比理解
这就像图书馆的运作模式。显存中存放的是“目录”和“关键锚点”(如书的封面和标题),而详细内容则存储在“仓库”(CPU)中。用户通过目录定位后,再提取具体书籍。这种机制确保了信息不丢失,且检索效率极高。

(来源:相关论文)

技术原理:奇异值分解与路由模块

SeKV 能够以 1/20的体积压缩内容而不丢失核心信息,主要依赖以下两项技术:

1. 奇异值分解(SVD)

对每个段落的键值矩阵进行分解,提取最重要的几十个成分,形成三个小矩阵。当需要恢复段落时,通过矩阵乘法还原出近似原始内容的高保真版本。这一过程类似于将文章生成二维码,通过扫描即可还原信息。

2. 轻量级路由模块

  • 极小参数:仅训练一个参数量为基础模型 万分之五的路由模块。
  • 快速训练:在 8 张 A100 显卡上训练 2-6 小时即可完成。
  • 即插即用:无需对大模型进行微调,可直接部署在现有模型上。

应用前景

SeKV 技术为长文本处理提供了高性价比的中间方案,通过牺牲极少量精度换取显存容量的成倍提升,显著降低了使用门槛:

  • 法律行业:快速审阅数百页的合同文件。
  • 学术研究:高效通读并分析数十篇学术论文。
  • 内容创作:让 AI 总结长篇小说或复杂文档。

这项技术让普通硬件也能胜任重度长文本任务,真正推动了 AI 应用的普及化。

参考资料:

  • 相关论文:https://arxiv.org/pdf/2606.31145
  • 作者 LinkedIn:https://www.linkedin.com/in/yuhang-he-7b20344b/

排版:胡巍巍
注:封面/首图由 AI 辅助生成

标签:

责任编辑:热点