ITBEAR科技资讯
网站首页 科技资讯 财经资讯 分享好友

AI处理长文本迎新突破:潜在上下文模型让速度飙升,内存占用锐减

时间:2026-06-15 22:52:28来源:互联网编辑:快讯

当人工智能助手处理一篇几十万字的长文档时,常常会陷入“记忆力危机”——就像一位每次都要从头翻阅厚厚参考书的学生,随着文档长度增加,处理速度越来越慢,甚至因内存不足而无法运行。这一困扰大语言模型(如ChatGPT)的痛点,正被一项名为“潜在上下文语言模型”(LCLM)的新技术打破。该研究由多所顶尖高校及实验室联合完成,通过改变信息处理方式,让AI在长文本任务中实现速度与准确率的双重突破。

传统AI处理长文本时,依赖一种名为“KV缓存”的技术,即临时存储输入文字的“关键信息”和“值信息”以供后续计算。这一过程如同学生在草稿纸上记录中间计算结果,文字越长,草稿纸越大,翻找和占用空间的时间也越长。尽管学界尝试通过“删减缓存”压缩信息,但要么因过度删除导致回答质量下降,要么因保守删减节省空间有限,甚至需要提前预知用户问题才能决定删减内容,实用性大打折扣。

LCLM的研究团队选择了一条截然不同的路径:不直接删减缓存,而是用更聪明的“记录方式”替代。他们设计了一个“速记员”角色——由专门模型将原始文字压缩成简短的“语义摘要符号”,再交给AI理解和回答。这一过程并非简单删减,而是通过重新编码将信息浓缩为更紧凑的符号语言,同时完整保留语义。实验显示,在标准长文理解测试中,LCLM在同等准确率下,处理速度比现有最佳方法快8.8倍;处理6.4万字文档时,速度提升仍达5.2倍。

技术实现的关键在于“编码器-适配器-解码器”的协作架构。编码器将输入文字切成固定窗口(每窗口1024词),通过“平均池化”压缩为少量向量(如16倍压缩率下,1024词压缩为64个向量)。适配器则像“转接头”,将编码器输出的向量维度转换为解码器可接受的格式。解码器基于压缩后的向量生成回答,全程无需直接接触原始长文本。这一设计不仅兼容主流AI推理引擎,还因编码器可并行处理不同文本窗口,大幅提升了压缩效率。

训练这一系统面临的核心挑战是缺乏“正确答案”——无人标注过“某段文字应压缩成哪些向量”。研究团队通过四阶段训练流程逐步攻克难题:第一阶段仅训练适配器,使其学会编码器与解码器间的语言转换;第二阶段解冻编码器,优化压缩方式;第三阶段端到端预训练,让系统协同工作;第四阶段用高质量任务数据微调,提升推理和长文理解能力。这一策略避免了初期编码器输出“噪声”导致训练混乱的问题,确保模型逐步适应协作模式。

数据构建是另一项关键工程。研究团队准备了三大类训练数据:第一类是“交错式预训练数据”,将文字切成交替片段,奇数片段压缩、偶数片段保持原样,迫使模型在任意位置条件化压缩上下文;第二类是“辅助重建数据”,要求模型从压缩向量重建原始文字,保留细节信息;第三类是“监督微调数据”,覆盖推理、长文档问答等任务,部分答案由更强模型重新生成以提高质量。混合训练策略防止了压缩向量“坍塌”为单一用途,确保了通用性。

在基准测试中,LCLM的表现全面领先。以RULER测试集为例,4K长度文本下,16倍压缩率时准确率达75.06%,速度是竞争对手的8.8倍;8倍压缩率时准确率升至85.42%,4倍压缩率时进一步达91.76%,接近不压缩时的94.41%。相比之下,KV缓存压缩方法因需完整处理文档生成缓存,处理时间几乎与压缩率无关,速度劣势明显。内存占用测试中,LCLM在16倍压缩率下处理128K至512K词元时,内存占用几乎不变,仅当文档超过512K词元时,解码器处理压缩序列的内存消耗才逐渐主导。

研究团队还探索了LCLM在智能体应用中的潜力。例如,让AI分析复杂代码库时,LCLM可先压缩整个代码库,提供“鸟瞰视角”,再通过“EXPAND”工具展开特定段落精读。这一设计使AI能先判断问题可能位置,再针对性深入调查,如同侦探先浏览档案再审讯嫌疑人。在RULER测试集的“针在草堆里”任务中,加入智能体功能的LCLM准确率大幅提升:8K长度文本下,平均准确率从72.51%跃升至92.46%;最难“3针”任务中,准确率从55.80%升至97.00%,部分场景下甚至匹配不压缩模型的性能。

尽管成果显著,研究仍存在局限性。例如,编码器与解码器的规模配比尚未完全摸清规律——增大解码器规模对预训练损失下降更明显,但下游任务准确率却因测试集不同而分化;8B解码器表现未达预期,可能与训练数据分布有关。当前测试基于HuggingFace标准实现,未充分利用vLLM等推理框架的高级优化功能,实际部署速度可能更快。

该技术的开源代码已公开,并与主流推理框架兼容,为工业应用奠定了基础。对于普通用户而言,未来使用AI处理长文档时,等待时间可能大幅缩短,成本降低,而回答质量几乎不受影响。对于AI研究界,LCLM证明了端到端训练的通用压缩器在高压缩率和超长文本场景中的优势,为长文本处理提供了新方向。更多技术细节可查阅论文编号arXiv:2606.09659。

更多热门内容