MemBubble: Benchmarking How Agent Memory Shapes LLM Response Diversity
Organizations: State Key Laboratory of AI Safety, Institute of Computing Technology, CAS · University of Chinese Academy of Sciences · Baidu Inc.
Abstract
Memory serves as the core component of LLM-based agents, enabling personalized responses via extracting and using user preferences from past interactions. However, its use may reduce response diversity, thereby narrowing the range of information available to users. Existing evaluations rely largely on synthetic interactions and cover few memory systems, leaving the effects of memory on response diversity insufficiently evaluated in real-world scenarios. We introduce MemBubble, a benchmark of 1000 instances built on preferences extracted from real user-assistant conversations. To measure the extent to which memory concentrates response distributions, we construct test queries in scenarios distinct from user preferences while preserving topical relevance through shared preference categories. Across multiple LLMs and memory mechanisms, entropy decreases by 34.7% on average compared with directly providing the full interaction history, highlighting the important role of memory processing in diversity loss. We also explore simple user-side interventions and changes in information availability but find that these do not effectively close the diversity gap. Our findings highlight the need to consider preserving diversity and user access to information alongside improvements in personalization when designing and using agent memory systems.
Figures & tables
| No mem. | Full history | NaiveRAG | MemoryBank | A-Mem | mem0 | MemoryOS | |
| Adapted Distinct- | |||||||
| GPT-6-Sol | 3.67 | 3.54 | 2.67 | 2.41 | 2.28 | 2.16 | 2.81 |
| GPT-5.6-Luna | 3.56 | 3.50 | 2.72 | 2.43 | 2.23 | 2.21 | 2.71 |
| Claude Haiku 4.5 | 3.58 | 2.87 | 2.86 | 2.34 | 2.38 | 2.09 | 3.04 |
| Gemini 3.7 Flash | 2.30 | 2.47 | 2.13 | 1.93 | 1.79 | 1.98 | 2.11 |
| GLM-5.3-flash | 3.05 | 3.55 | 2.63 | 2.38 | 2.14 | 2.19 | 2.65 |
| # Dialogues | 0 | 5 | 10 | 20 | 50 | 100 |
|---|---|---|---|---|---|---|
| MemoryBank | -1.31 | -1.00 | -0.99 | -0.90 | -0.83 | -0.67 |
| A-Mem | -1.39 | -1.24 | -1.15 | -0.98 | -0.93 | -0.54 |
| mem0 | -1.24 | -1.21 | -1.18 | -1.27 | -1.15 | -0.93 |
| MemoryOS | -1.27 | -0.82 | -0.88 | -0.80 | -0.49 | -0.40 |
Appendix figures & tables9 assets
Supplementary material from the paper’s appendix.
Appendix
| Task type | Keywords, patterns, or conditions |
|---|---|
| Rewriting, translation, polishing, proofreading, or summarization | Trigger keywords: translate into , translate , rewrite , reword , rephrase , paraphrase , summarize , summarise , proofread , correct the grammar , fix the grammar , check the grammar , make it more , make it less , shorten , expand this ; 改写, 翻译成, 翻译, 润色, 校对, 总结, 概括, 精简. Required long text block: At least 400 characters in the message; three consecutive nonempty lines totaling at least 300 characters; or a quoted span of at least 200 characters enclosed by """ … """ , " … " , or 「…」. Any one condition suffices. |
| Role-playing and jailbreak | Expressions: DAN , AIM , JailBreak , developer mode , do anything now ; you are [now] (a|an|my) , pretend to be , pretend you are , act as (a|an|my) ; role-play , role_play , roleplay ; stay in character , ignore [all] (previous|prior) instructions , without any restrictions ; 你现在是, 扮演, 不受[任何]限制. Persona-field prefixes: name: , age: , gender: , personality: , appearance: , background: , species: . A persona-field line begins with one of these prefixes. |
| Stage | Role | Model | Prompt |
| § 3.2 Preference extraction | Extractor | DeepSeek-V4-Flash ( DeepSeek-AI and others, 2026 ) | Figure 12 |
| § 3.2 Consistency check | Verifier | GPT-5.6-Luna ( OpenAI, 2026b ) | Figure 13 |
| § 3.2 Semantic assessment | Judges | GPT-5.5 ( OpenAI, 2026a ) | Figure 14 |
| Claude Opus 4.8 ( Anthropic, 2026 ) | |||
| § 3.3 Test query generation | Generator | GPT-5.5 ( OpenAI, 2026a ) | Figure 15 |
| § 3.3 Leakage and relevance check | Judge | GPT-5.5 ( OpenAI, 2026a ) | Figure 16 |
| Stage | H1–H2 | LLM–H1 | LLM–H2 |
|---|---|---|---|
| Consistency | 93.5% | 86.0% | 90.5% |
| Semantic assessment | 81.0% | 83.5% | 78.5% |
| Leakage and Relevance | 90.5% | 87.5% | 91.0% |
| Field | Mean | Median | Q1–Q3 | 95th pct. | Min–Max |
|---|---|---|---|---|---|
| Source dialogue (tokens) | 3,118.2 | 2,914 | 1,633–4,009 | 7,236 | 76–12,675 |
| Source dialogue (messages) | 17.6 | 14 | 12–20 | 34 | 10–84 |
| Preference (words) | 16.2 | 16 | 13–19 | 25 | 1–35 |
| Test query (words) | 11.8 | 12 | 11–13 | 15 | 4–22 |
| Model | No mem. | Full history | NaiveRAG | MemoryBank | A-Mem | mem0 | MemoryOS |
|---|---|---|---|---|---|---|---|
| GPT-6-Sol | 4 | 6 | 4 | 1 | 3 | 1 | 2 |
| GPT-5.6-Luna | 4 | 4 | 1 | 1 | 1 | 1 | 2 |
| Claude Haiku 4.5 | 4 | 5 | 3 | 1 | 1 | 1 | 1 |
| Gemini 3.7 Flash | 4 | 4 | 1 | 1 | 1 | 1 | 1 |
| GLM-5.3-flash | 1 | 5 | 1 | 1 | 1 | 1 | 2 |
| Llama3.3-70B | 7 | 5 | 6 | 1 | 1 | 1 | 2 |
LLM Agent Memory
Large language model agent memory lets agents retain, organize, retrieve, and update information across interactions and long-running tasks.
See all 357 papers