跳到正文
arXiv · 架构与算子· William L. Tong; Aryo Lotfi; Emmanuel Abbe; Josh Susskind; Cengiz Pehlevan; Eran Malach·· 1 天前精选AI 评分76

用于高效长上下文检索的加权状态融合(Weighted State Souping)

Weighted State Souping for Efficient Long-Context Retrieval

AI 导读

基于摘要分析。该研究面向语言模型在长且信息密集的上下文中检索能力下降的问题,提出 Weighted State Souping:为独立编码并缓存的 SSM 状态学习查询相关权重,以选择性组合上下文表示。其主要研究对象是文档、工具定义和记忆记录等语言或结构化上下文,而非 EEG 或 BCI。 已有 state souping 将 token 分块独立编码,再重组状态供下游查询使用;作者指出,这种方法与普通 SSM 在长且拥挤的上下文中均会退化。新方法的核心变化是在缓存状态的组合阶段引入查询依赖权重,并进一步构建稀疏变体。摘要未说明权重生成方式、训练目标、状态组合公式或稀疏选择机制。 作者报告,该方法在单跳与多跳文档问答、needle-in-a-haystack 检索、变量跟踪和工具使用任务中改善了长上下文表现;稀疏变体的编码成本随上下文长度近乎保持常数,同时保持、部分情况下改善性能。摘要未提供具体数据集、数据划分、指标、数值及缓存构建成本是否计入,因此这些结果只能按所述任务范围理解,不能据此认定端到端总成本恒定。作者还给出解释 state souping 有效性及失败情形的简单理论论证,并对加权方案的优势作非形式化讨论;论证条件、实验协议、消融及统计信息尚未验证。 平台推测(待验证):若将独立片段编码与查询相关状态组合用于 EEG,可能对应长时记录中任务相关片段的选择问题,但这是假设,不是本文已验证的贡献。可复用思路是片段缓存与选择性聚合;需要改造信号编码、查询定义及时间位置处理,并明确监督来源。EEG 的低信噪比、跨被试差异、通道变化和小数据条件,可能使权重偏向伪迹或使状态压缩丢失短暂事件。一个可证伪的小实验是在固定 EEG 编码器和相同数据划分下,对比等权与查询加权聚合,逐步加入无关片段,检查任务指标及缓存、查询成本是否同步获益。相关 EEG 工作尚未检索确认。

阅读价值

值得关注其以查询相关权重组合缓存 SSM 状态的机制,以及稀疏化后上下文长度与编码成本的关系,但摘要不足以验证成本口径、性能增益和 EEG 迁移价值。

来源:arXiv · 架构与算子 · arxiv.org