Grounding Memory Summarization in Utility Intent
Organizations: University of Virginia · Northeastern University
Abstract
Existing summarizers for memory systems are typically optimized for human-facing criteria such as faithfulness, which misaligns with their true objective: preserving the evidence needed to support future queries. We show that conditioning summarization on query-answer pairs substantially improves answer quality, and that this utility-aware behavior is transferable across queries. Motivated by these findings, we propose MemSuit, a self-distillation framework in which a teacher summarizer, conditioned on observed query-answer pairs, produces utility-aware memory entries that a student learns to reproduce from the raw conversation alone. To prevent collateral erasure where conditioning on a single query-answer pair discards evidence relevant to other plausible queries, the teacher decomposes each block into multiple self-contained entries that preserve distinct query-relevant facets as independently retrievable units. To align the retriever with the compact, fact-dense style of teacher entries, we further fine-tune the embedding model with a contrastive objective supervised by teacher entries. Across a diverse suite of conversational query types, MemSuit consistently outperforms state-of-the-art baselines, confirming the value of grounding memory in downstream utility.
Figures & tables
| Model | Method | Multi | Temp | Open | Single | Avg | |||||
|---|---|---|---|---|---|---|---|---|---|---|---|
| F1 | BLEU | F1 | BLEU | F1 | BLEU | F1 | BLEU | F1 | BLEU | ||
| Llama-3.1-8B-Instruct | LoCoMo | 21.33 | 16.42 | 17.46 | 14.41 | 10.60 | 6.69 | 30.82 | 23.41 | 20.05 | 15.23 |
| MemoryBank | 24.60 | 19.46 | 12.01 | 9.56 | 10.47 | 8.18 | 38.45 | 34.34 | 21.38 | 17.89 | |
| A-Mem | 17.08 | 13.21 | 24.00 | 18.26 | 11.93 | 10.48 | 29.62 | 21.28 | 20.66 | 15.81 | |
| Mem0 | 28.98 | 23.35 | 19.93 | 15.18 | 19.73 | 16.41 | 45.76 | 39.79 | 28.60 | 23.68 | |
| SimpleMem | 26.92 | 17.66 | 35.11 | 23.58 | 22.41 | 16.89 | 35.92 | 30.83 | 30.09 | 22.24 | |
| Model | Variant | Multi | Temp | Open | Single | Avg | |||||
|---|---|---|---|---|---|---|---|---|---|---|---|
| F1 | BLEU | F1 | BLEU | F1 | BLEU | F1 | BLEU | F1 | BLEU | ||
| Llama-3.1-8B-Instruct | w/o Rtr-Cal | 26.02 | 18.36 | 41.03 | 28.19 | 21.16 | 17.29 | 38.86 | 33.17 | 31.77 | 24.25 |
| w/o Self-Dist | 30.92 | 21.68 | 40.24 | 28.38 | 21.36 | 16.72 | 40.92 | 36.02 | 33.36 | 25.70 | |
| w/o Cal & Dist | 23.16 | 15.51 | 30.88 | 21.30 | 17.53 | 14.50 | 33.38 | 28.62 | 26.24 | 19.98 | |
| MemSuit | 31.18 | 23.55 | 40.77 | 27.94 | 25.42 | 20.83 | 44.89 | 38.61 | 35.56 | 27.73 | |
| Qwen2.5-3B-Instruct | w/o Rtr-Cal | 21.23 | 16.22 | 26.65 | 18.57 | 14.58 | 11.56 | 30.09 | 26.58 | 23.14 | 18.23 |
| Model | Recall@1 | Recall@5 | Recall@10 | |
|---|---|---|---|---|
| Llama | Before | 25.77 | 54.64 | 64.43 |
| After | 35.05 | 63.40 | 69.59 | |
| Qwen | Before | 21.78 | 44.06 | 51.49 |
| After | 26.73 | 48.02 | 55.94 |
| Longmemeval | F1 | Bleu1 |
|---|---|---|
| Mem0 | 8.92 | 7.01 |
| SimpleMem | 28.91 | 21.40 |
| MemSuit | 28.16 | 20.84 |
Appendix figures & tables4 assets
Supplementary material from the paper’s appendix.
Appendix
| Metric | Qwen2.5-3B | Llama-3.1-8B |
|---|---|---|
| Multi-F1 | 22.63 | 22.89 |
| Multi-BLEU | 16.36 | 15.22 |
| Temp-F1 | 28.66 | 27.78 |
| Temp-BLEU | 20.40 | 18.39 |
| Open-F1 | 15.40 | 17.15 |
| Open-BLEU | 13.15 | 13.96 |
| Metric | MemSuit | Adaptive (non-target) |
|---|---|---|
| Multi-F1 | 22.63 | 19.62 |
| Multi-BLEU | 16.36 | 13.82 |
| Temp-F1 | 28.66 | 24.75 |
| Temp-BLEU | 20.40 | 19.57 |
| Open-F1 | 15.40 | 15.85 |
| Open-BLEU | 13.15 | 12.10 |
| LongMemEval | F1 | BLEU-1 |
|---|---|---|
| Mem0 | 8.92 | 7.01 |
| SimpleMem | 28.91 | 21.40 |
| MemSuit (LoCoMo-trained, OOD) | 28.16 | 20.84 |
| MemSuit (LongMemEval-trained, ID) | 30.89 | 24.32 |
| Metric | LoCoMo | MemoryBank | A-mem | Mem0 | SimpleMem | MemSuit |
|---|---|---|---|---|---|---|
| Multi-F1 | 8.51 | 9.91 | 18.22 | 15.44 | 17.56 | 19.88 |
| Multi-BLEU | 6.96 | 7.21 | 14.91 | 11.27 | 12.77 | 15.22 |
| Temp-F1 | 6.34 | 12.07 | 23.86 | 16.08 | 22.39 | 23.87 |
| Temp-BLEU | 5.14 | 9.04 | 18.60 | 12.20 | 15.15 | 15.55 |
| Open-F1 | 9.48 | 6.97 | 15.09 | 15.35 | 16.27 | 22.88 |
| Open-BLEU | 7.36 | 5.24 | 12.83 | 12.25 | 12.70 | 19.52 |