Structured Distillation for Personalized Agent Memory: 11x Token Reduction with Retrieval Preservation

TL;DR

通过结构化蒸馏方法,将个性化代理记忆的代币数量减少11倍,同时保留检索能力。

cs.AI 🔴 高级 2026-03-13 3 次浏览
Sydney Lewis
个性化记忆 结构化蒸馏 信息检索 自然语言处理 软件工程

核心发现

方法论

本文提出了一种结构化蒸馏方法,将用户与AI代理的对话历史压缩为一个可检索的层。每次对话被压缩为一个包含四个字段的复合对象:exchange_core、specific_context、thematic room_assignments和regex-extracted files_touched。通过这种方法,平均每次对话的代币数量从371减少到38,实现了11倍的压缩。

关键结果

  • 在4,182次对话(共14,340次交流)中,平均每次交流的长度从371个代币减少到38个,压缩比达到11倍。
  • 在201个以召回为导向的查询中,最佳的纯蒸馏配置达到了96%的最佳逐字MRR(0.717对0.745)。
  • 所有20种向量搜索配置在Bonferroni校正后均无显著性,而所有20种BM25配置显著退化(效应大小|d|=0.031-0.756)。

研究意义

该研究展示了如何在不显著牺牲检索质量的情况下,压缩单用户代理记忆。这种方法允许在单个提示中容纳数千次交流,同时保留逐字来源以供深入分析。对于需要处理大量对话历史的应用场景,如个性化助手和客户服务系统,该方法具有重要意义。

技术贡献

技术上,该研究提出了一种新的结构化蒸馏方法,通过将对话历史压缩为可检索的复合对象,显著降低了记忆存储的代币成本。与现有的摘要方法不同,该方法保留了检索所需的关键信息,并通过多种检索模式验证了其有效性。

新颖性

该方法首次将个性化代理记忆的蒸馏与结构化信息提取相结合,显著提高了记忆压缩效率并保留了检索质量。与传统的对话摘要方法相比,该方法提供了更高效的记忆管理解决方案。

局限性

  • 在BM25配置中,检索质量显著下降,表明该方法在词汇重叠依赖较强的情况下效果不佳。
  • 向量搜索配置在统计上无显著性,可能限制了该方法在某些语义匹配任务中的应用。
  • 该方法主要针对单用户场景,尚未验证其在多用户或跨域应用中的有效性。

未来方向

未来的研究可以探索该方法在多用户环境中的应用,进一步优化蒸馏过程以提高跨域检索的性能。此外,可以研究如何结合其他信息检索技术,如深度学习模型,以进一步提高检索效率和准确性。

AI 总览摘要

长时间与AI代理的对话会产生大量的历史记录,这对于用户来说是有用的,但逐字保留这些记录的代价很高。本文研究了一种个性化代理记忆的方法,将用户的对话历史蒸馏为一个紧凑的检索层,以便于后续搜索。每次交流被压缩为一个复合对象,包含四个字段:exchange_core、specific_context、thematic room_assignments和regex-extracted files_touched。通过这种方法,平均每次交流的代币数量从371减少到38,实现了11倍的压缩。

在4,182次对话(共14,340次交流)中,平均每次交流的长度从371个代币减少到38个,压缩比达到11倍。我们评估了个性化召回是否能在这种压缩下幸存,使用了201个以召回为导向的查询,107种配置跨越5种纯搜索模式和5种跨层搜索模式,以及5个大型语言模型评分器(214,519个共识评分的查询结果对)。最佳的纯蒸馏配置达到了96%的最佳逐字MRR(0.717对0.745)。

结果显示,机制依赖性很强。所有20种向量搜索配置在Bonferroni校正后均无显著性,而所有20种BM25配置显著退化(效应大小|d|=0.031-0.756)。最佳的跨层设置略微超过了最佳的纯逐字基线(MRR 0.759)。结构化蒸馏在不均匀牺牲检索质量的情况下压缩了单用户代理记忆。在1/11的上下文成本下,数千次交流可以在一个提示中适应,而逐字来源仍然可用于深入分析。

我们将实现和分析管道作为开源软件发布。该研究展示了如何在不显著牺牲检索质量的情况下,压缩单用户代理记忆。这种方法允许在单个提示中容纳数千次交流,同时保留逐字来源以供深入分析。对于需要处理大量对话历史的应用场景,如个性化助手和客户服务系统,该方法具有重要意义。未来的研究可以探索该方法在多用户环境中的应用,进一步优化蒸馏过程以提高跨域检索的性能。此外,可以研究如何结合其他信息检索技术,如深度学习模型,以进一步提高检索效率和准确性。

深度分析

研究背景

在人工智能领域,随着对话式AI代理的普及,如何有效管理和检索用户与代理之间的对话历史成为一个重要的研究课题。传统的对话摘要方法通常采用压缩和丢弃原始对话的方式,这种有损的摘要在长时间对话中会导致信息的逐步丢失。近年来,结构化信息提取技术的进步为解决这一问题提供了新的思路。通过将对话历史转化为可检索的结构化数据,可以在保留关键信息的同时大幅减少存储成本。

核心问题

长时间与AI代理的对话会产生大量的历史记录,这对于用户来说是有用的,但逐字保留这些记录的代价很高。传统的摘要方法在压缩过程中会丢失大量关键信息,导致后续检索质量下降。如何在不显著牺牲检索质量的情况下,压缩个性化代理记忆,成为一个亟待解决的问题。

核心创新

本文提出了一种结构化蒸馏方法,将用户与AI代理的对话历史压缩为一个可检索的层。每次对话被压缩为一个包含四个字段的复合对象:exchange_core、specific_context、thematic room_assignments和regex-extracted files_touched。通过这种方法,平均每次对话的代币数量从371减少到38,实现了11倍的压缩。与传统的摘要方法不同,该方法保留了检索所需的关键信息,并通过多种检索模式验证了其有效性。

方法详解

  • �� 采用结构化蒸馏方法,将对话历史压缩为可检索的复合对象。
  • �� 每个对象包含四个字段:exchange_core、specific_context、thematic room_assignments和regex-extracted files_touched。
  • �� 使用多种检索模式评估蒸馏效果,包括向量搜索和BM25配置。
  • �� 通过对比蒸馏文本和逐字文本的检索结果,验证信息保留效果。

实验设计

实验使用了来自6个软件工程项目的4,182次对话,共14,340次交流。评估使用了201个以召回为导向的查询,107种配置跨越5种纯搜索模式和5种跨层搜索模式。实验采用了5个大型语言模型评分器,共评估了214,519个共识评分的查询结果对。关键指标包括MRR、平均评分、P@1和nDCG@10。

结果分析

实验结果显示,最佳的纯蒸馏配置达到了96%的最佳逐字MRR(0.717对0.745)。所有20种向量搜索配置在Bonferroni校正后均无显著性,而所有20种BM25配置显著退化(效应大小|d|=0.031-0.756)。最佳的跨层设置略微超过了最佳的纯逐字基线(MRR 0.759)。

应用场景

该方法适用于需要处理大量对话历史的应用场景,如个性化助手和客户服务系统。通过将对话历史压缩为可检索的结构化数据,可以在保留关键信息的同时大幅减少存储成本,从而提高系统的效率和用户体验。

局限与展望

虽然该方法在向量搜索配置中表现良好,但在BM25配置中检索质量显著下降,表明该方法在词汇重叠依赖较强的情况下效果不佳。此外,该方法主要针对单用户场景,尚未验证其在多用户或跨域应用中的有效性。未来的研究可以探索该方法在多用户环境中的应用,进一步优化蒸馏过程以提高跨域检索的性能。

通俗解读 非专业人士也能看懂

想象你在厨房里做饭。你有很多食材和工具,但你不需要每次做饭都把所有东西都拿出来。相反,你会根据需要选择特定的食材和工具。类似地,AI代理在与用户对话时,也不需要每次都记住所有的对话历史。本文提出了一种方法,将对话历史压缩为一个紧凑的检索层,就像把厨房里的食材和工具整理成一个方便的清单。这样,当你需要某个特定的食材时,你可以很快找到它,而不必翻遍整个厨房。这种方法不仅节省了空间,还提高了效率,让AI代理能够更快地找到用户需要的信息。

简单解释 像给14岁少年讲一样

想象一下你在玩一个大型多人在线游戏。你和你的朋友们在游戏中有很多次对话和冒险,但你不需要每次都记住所有的细节。相反,你会记住那些重要的任务和关键的道具。AI代理也是这样!本文介绍了一种方法,可以帮助AI代理记住重要的对话内容,而不是所有的细节。就像在游戏中,你可以很快找到你需要的任务信息,而不必翻遍整个聊天记录。这种方法让AI代理变得更聪明,更高效!

术语表

结构化蒸馏

一种将对话历史压缩为可检索结构的技术,保留关键信息以供后续检索。

用于压缩用户与AI代理的对话历史。

个性化代理记忆

存储和检索单个用户与AI代理的对话历史的系统。

研究如何有效管理和检索用户的对话历史。

MRR (Mean Reciprocal Rank)

一种评估信息检索系统性能的指标,表示第一个相关结果的平均倒数排名。

用于评估蒸馏文本和逐字文本的检索效果。

BM25

一种基于词频-逆文档频率的检索算法,用于评估文本相似度。

用于评估蒸馏文本的检索效果。

向量搜索

一种基于向量空间模型的检索方法,通过计算向量之间的相似度进行检索。

用于评估蒸馏文本的检索效果。

exchange_core

对话中完成的任务的简要描述,通常为1-2句话。

作为蒸馏对象的一个字段,用于保留关键信息。

specific_context

对话中一个独特的技术细节,如错误信息或参数名称。

作为蒸馏对象的一个字段,用于保留关键信息。

thematic room_assignments

对话中涉及的主题或概念的分类,用于组织和检索信息。

作为蒸馏对象的一个字段,用于组织信息。

regex-extracted files_touched

对话中提到的文件路径,通过正则表达式提取。

作为蒸馏对象的一个字段,用于保留关键信息。

Claude Code

一个用于软件工程项目的AI对话代理,支持用户与AI之间的对话。

研究中使用的对话代理平台。

FAISS

一种用于高效相似性搜索的库,支持大规模向量搜索。

用于存储和检索蒸馏文本的向量。

HNSW

一种基于分层导航小世界图的近似最近邻搜索算法。

用于评估蒸馏文本的检索效果。

Exact

一种精确的向量搜索方法,通过计算向量之间的精确距离进行检索。

用于评估蒸馏文本的检索效果。

Reciprocal Rank Fusion (RRF)

一种融合多种检索结果的方法,通过计算倒数排名的加权和进行融合。

用于多字段模式下的检索结果融合。

CombMNZ

一种融合多种检索结果的方法,通过计算结果的加权和进行融合。

用于跨层模式下的检索结果融合。

开放问题 这项研究留下的未解疑问

  • 1 如何在多用户环境中应用结构化蒸馏方法,以支持多个用户同时与AI代理进行对话?目前的方法主要针对单用户场景,尚未验证其在多用户环境中的有效性。
  • 2 如何进一步优化蒸馏过程以提高跨域检索的性能?现有方法在某些语义匹配任务中的效果有限,可能需要结合其他信息检索技术。
  • 3 如何结合深度学习模型以提高检索效率和准确性?现有方法主要依赖于传统的信息检索技术,可能无法充分利用深度学习的优势。
  • 4 如何在不显著增加计算成本的情况下提高检索质量?现有方法在某些配置下的检索质量有所下降,可能需要进一步优化。
  • 5 如何在不丢失关键信息的情况下进一步压缩对话历史的代币数量?现有方法已经实现了11倍的压缩,但仍有进一步优化的空间。

应用场景

近期应用

个性化助手

通过压缩用户与AI代理的对话历史,提高个性化助手的响应速度和效率。

客户服务系统

在客户服务系统中应用该方法,以便快速检索和处理客户的历史对话记录。

软件工程项目管理

在软件工程项目中应用该方法,以便于团队成员快速访问和检索项目相关的对话历史。

远期愿景

多用户对话管理

开发支持多个用户同时与AI代理进行对话的系统,提高协作效率。

跨域信息检索

结合深度学习技术,开发支持跨域信息检索的系统,提高检索效率和准确性。

原文摘要

Long conversations with an AI agent create a simple problem for one user: the history is useful, but carrying it verbatim is expensive. We study personalized agent memory: one user's conversation history with an agent, distilled into a compact retrieval layer for later search. Each exchange is compressed into a compound object with four fields (exchange_core, specific_context, thematic room_assignments, and regex-extracted files_touched). The searchable distilled text averages 38 tokens per exchange. Applied to 4,182 conversations (14,340 exchanges) from 6 software engineering projects, the method reduces average exchange length from 371 to 38 tokens, yielding 11x compression. We evaluate whether personalized recall survives that compression using 201 recall-oriented queries, 107 configurations spanning 5 pure and 5 cross-layer search modes, and 5 LLM graders (214,519 consensus-graded query-result pairs). The best pure distilled configuration reaches 96% of the best verbatim MRR (0.717 vs 0.745). Results are mechanism-dependent. All 20 vector search configurations remain non-significant after Bonferroni correction, while all 20 BM25 configurations degrade significantly (effect sizes |d|=0.031-0.756). The best cross-layer setup slightly exceeds the best pure verbatim baseline (MRR 0.759). Structured distillation compresses single-user agent memory without uniformly sacrificing retrieval quality. At 1/11 the context cost, thousands of exchanges fit within a single prompt while the verbatim source remains available for drill-down. We release the implementation and analysis pipeline as open-source software.

cs.AI cs.CL cs.IR

参考文献 (20)

Variations in relevance judgments and the measurement of retrieval effectiveness

E. Voorhees

1998 866 引用 ⭐ 高影响力

BEIR: A Heterogenous Benchmark for Zero-shot Evaluation of Information Retrieval Models

Nandan Thakur, Nils Reimers, Andreas Ruckl'e 等

2021 1503 引用 查看解读 →

Mistral 7B

Albert Qiaochu Jiang, Alexandre Sablayrolles, Arthur Mensch 等

2023 3189 引用 查看解读 →

Qwen3 Technical Report

An Yang, Anfeng Li, Baosong Yang 等

2025 3584 引用 查看解读 →

Measuring nominal scale agreement among many raters.

J. Fleiss

1971 9146 引用

TREC: Experiment and evaluation in information retrieval

José Luis Vicedo González, Jaime Gómez

2007 1105 引用

MiniLM: Deep Self-Attention Distillation for Task-Agnostic Compression of Pre-Trained Transformers

Wenhui Wang, Furu Wei, Li Dong 等

2020 1947 引用 查看解读 →

Yi: Open Foundation Models by 01.AI

01.AI Alex Young, Bei Chen, Chao Li 等

2024 807 引用 查看解读 →

Phi-3 Technical Report: A Highly Capable Language Model Locally on Your Phone

Marah Abdin, Sam Ade Jacobs, A. A. Awan 等

2024 2041 引用 查看解读 →

Reciprocal rank fusion outperforms condorcet and individual rank learning methods

G. Cormack, C. Clarke, Stefan Büttcher

2009 784 引用

Sentence-BERT: Sentence Embeddings using Siamese BERT-Networks

Nils Reimers, Iryna Gurevych

2019 16641 引用 查看解读 →

MemGPT: Towards LLMs as Operating Systems

Charles Packer, Vivian Fang, Shishir G. Patil 等

2023 434 引用 查看解读 →

Cumulated gain-based evaluation of IR techniques

K. Järvelin, Jaana Kekäläinen

2002 5335 引用

The measurement of observer agreement for categorical data.

J. Landis, G. Koch

1977 76718 引用

InternLM2 Technical Report

Zheng Cai, Maosong Cao, Haojiong Chen 等

2024 351 引用 查看解读 →

A Survey on Dialogue Summarization: Recent Advances and New Frontiers

Xiachong Feng, Xiaocheng Feng, Bing Qin

2021 118 引用 查看解读 →

The kappa statistic in reliability studies: use, interpretation, and sample size requirements.

J. Sim, C. Wright

2005 4083 引用

Efficient and Robust Approximate Nearest Neighbor Search Using Hierarchical Navigable Small World Graphs

Yury Malkov, Dmitry A. Yashunin

2016 2108 引用 查看解读 →

Retrieval-Augmented Generation for Knowledge-Intensive NLP Tasks

Patrick Lewis, Ethan Perez, Aleksandara Piktus 等

2020 12025 引用 查看解读 →

Statistical Power Analysis for the Behavioral Sciences

Jacob Cohen

1969 61317 引用