SheepNav
精选今天0 投票

Akashic:基于MemAttention的低开销LLM推理服务,长上下文效率提升1.88倍

大型语言模型(LLM)驱动的智能体系统正在变得日益复杂,它们需要跨多轮对话、工具调用乃至跨会话工作流持续积累上下文。然而,传统做法——为每次请求重放完整历史——正变得不切实际:长上下文不仅增加了预填充(prefill)成本,可能超出上下文窗口限制,更会将任务相关的关键证据淹没在无关内容中,同时损害服务效率与输出质量。针对这一痛点,来自学术界的研究团队提出了 Akashic,一个围绕 MemAttention 构建的低开销内存系统,旨在以更智能的方式管理长上下文,显著提升推理效率与任务准确性。

MemAttention:分块记忆与语义关联

Akashic 的核心创新在于 MemAttention 机制。它不再将整个对话历史视为一个不可分割的连续序列,而是将上下文组织成有界的分块(chunks),并显式建模跨分块之间的语义关系。这种方法的关键优势在于:它能够保留跨分块的关键证据,而无需每次都重写完整的历史记录。当智能体需要回溯之前某个工具调用的结果或几轮对话前的用户意图时,MemAttention 可以直接定位到相关分块,而不是在海量文本中“大海捞针”。

软硬协同:减少检索碎片化与I/O开销

除了核心算法创新,Akashic 还采用了硬件-软件协同设计的内存放置策略。系统会分析分块之间的共现模式,将那些很可能被同时检索到的分块在物理存储上“就近”放置。这种设计有效减少了检索过程中的碎片化问题,并大幅降低了 I/O 开销。在分布式推理场景中,这意味着更少的数据跨节点搬运,从而节省带宽、降低延迟。

性能数据:准确率提升10.2个百分点

研究团队在四个代表性工作负载和三种不同规模的模型上进行了评估。结果显示,与当前最先进的记忆基线方案相比,Akashic 实现了多项关键指标的显著提升:

  • 任务准确率最高提升 10.2 个百分点
  • 吞吐量提升高达 1.21 倍
  • 可持续请求率(即在保持服务质量的前提下能处理的最大并发请求数)提升高达 1.88 倍

这些数据表明,Akashic 不仅在准确性上优于现有方法,而且在系统效率方面也有质的飞跃,能够支撑更大规模的 LLM 智能体服务。

行业意义与展望

长上下文管理一直是 LLM 落地中的核心挑战之一。现有方案如滑动窗口、稀疏注意力或简单的摘要压缩,往往在信息完整性和计算效率之间难以两全。Akashic 提出的“分块+语义关联+协同放置”的组合思路,为这一难题提供了一个优雅且高效的解决方案。

对于正在构建复杂智能体应用的开发者而言,这项技术意味着:他们可以更自由地设计需要长期记忆和工具协作的工作流,而无需担心上下文爆炸或性能瓶颈。Akashic 的研究成果不仅停留在理论层面,其显著的性能提升数据已经展示了在真实生产环境中部署的巨大潜力。

随着 LLM 应用从简单的问答向复杂的自主代理演进,类似 Akashic 这样的高效记忆系统将成为基础设施的关键一环。未来,我们或许会看到更多融合了语义理解与系统优化思想的创新方案,共同推动大模型服务进入更高效、更智能的新阶段。

延伸阅读

  1. 记忆进入循环:面向语言代理的进程内检索扩展工作记忆研究
  2. FirstResearch:让AI科学家的研究问题变得可审计
  3. 叙事世界模型:为长篇小说创作打造的叙事学记忆系统
查看原文