SeKV算法优化AI长文本推理,显存占用降逾五成且精度提升
近日,不列颠哥伦比亚大学与微软研究院联合推出长文本大模型推理优化方案SeKV。该方法由微软研究院高级研究员何宇航等人主导研发,旨在突破大语言模型处理超长上下文时显存消耗呈线性增长的瓶颈。传统键值缓存机制随文本长度倍增,导致普通显卡难以承载数十万字的推理任务。SeKV创新性地引入语义分段与动态路由机制,将核心语义摘要与关键锚点保留于图形处理器显存中,利用奇异值分解技术将段落详细缓存压缩至中央处理器内存,体积缩减至原二十分之一。系统仅需加载一个占比不足基础模型万分之五的轻量级路由模块,即可在生成回复时按需精准调取原文细节。实测数据显示,该方案在仅占用原显存十分之一的条件下,回答准确率较现有语义压缩方法提升近六个百分点,整体显存占用降低超百分之五十。此技术无需更换硬件或重新微调底层模型,使配备24GB显存的消费级显卡即可流畅处理约三十万字的长文本。SeKV的问世将显著降低行业算力门槛,为法律文书审查、学术文献通读及复杂内容生成等场景提供高效且低成本的推理方案。
