TASM让多模态模型在有限上下文下更智能地记忆和检索信息。
Task-Aware Structured Memory for Dynamic Multi-modal In-Context Learning

- 用任务向量指导压缩,避免逐样本判断带来的偏差。
- 通过图匹配合并语义相近的标记,保留视觉信息结构。
- 分层内存设计支持动态查询检索,适合长序列场景。
多模态大语言模型依赖上下文学习实现快速任务适配,但受限于有限的上下文窗口和长多模态序列中键值缓存的增长成本。现有记忆压缩方法通常依赖固定的标记删除或样本相关的权重估计,引入偏见,破坏语义结构,尤其影响视觉表征,并生成无法适应新查询的静态记忆。我们提出 TASM(任务感知结构化记忆),一种无需训练的框架,通过任务感知、结构保持且可动态访问的记忆构建解决上述问题。TASM 使用任务向量引导压缩,以任务级方向替代样本特定信号,捕捉演示间的共享相关性;通过二部图匹配实现语义感知的标记聚合,不进行破坏性剪枝;最终将记忆结构化为紧凑的核心记忆与潜在库,支持查询自适应的动态检索。评估表明,TASM 在高强度压缩下仍保持高性能,有效平衡了效率与适应性。
原文摘要 · Abstract (English)
Multi-modal large language models (MLLMs) depend on in-context learning (ICL) for rapid task adaptation, but their scalability is severely limited by finite context windows and the growing cost of key-value (KV) caches in long multi-modal sequences. Existing memory compression approaches typically rely on rigid token removal or sample-dependent importance estimation, which introduces bias, disrupts semantic structure, particularly for visual representations, and yields static memories that cannot adapt to new queries. We introduce TASM (Task-Aware Structured Memory), a training-free framework that addresses these limitations through task-aware, structure-preserving, and dynamically accessible memory construction. TASM employs task-vector guided compression to replace sample-specific signals with a task-level direction that captures shared relevance across demonstrations. To preserve the underlying manifold, it applies semantics-aware token merging via bipartite graph matching, aggregating tokens without destructive pruning. Finally, TASM structures memory into a hierarchy comprising a compact Core Memory and a Latent Bank, facilitating query-adaptive dynamic retrieval. Evaluations confirm TASM maintains high performance under heavy compression, effectively balancing efficiency with adaptability.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。