MemGuide: Intent-Driven Memory Selection for Goal-Oriented Multi-Session LLM Agents

Authors

  • Yiming Du The Chinese University of Hong Kong, Hong Kong, China MoE Key Laboratory of High Confidence Software Technologies, China
  • Bingbing Wang Harbin Institute of Technology, Shenzhen, China
  • Yang He Hong Kong University of Science and Technology, Hong Kong, China
  • Bin Liang The Chinese University of Hong Kong, Hong Kong, China MoE Key Laboratory of High Confidence Software Technologies, China
  • Baojun Wang Huawei Technologies Ltd., Shenzhen, China
  • Zhongyang Li Microsoft AI
  • Lin Gui King's College London, UK
  • Jeff Z. Pan The University of Edinburgh, UK
  • Ruifeng Xu Harbin Institute of Technology, Shenzhen, China
  • Kam-Fai Wong The Chinese University of Hong Kong, Hong Kong, China MoE Key Laboratory of High Confidence Software Technologies, China

DOI:

https://doi.org/10.1609/aaai.v40i36.40313

Abstract

Modern task-oriented dialogue (TOD) systems increasingly rely on large language model (LLM) agents, leveraging Retrieval-Augmented Generation (RAG) and long-context capabilities for long-term memory utilization. However, these methods prioritise semantic similarity over task intent, degrading multi-session coherence. We propose MemGuide, a two-stage intent-driven memory selection framework: (1) Intent‑Aligned Retrieval retrieves goal-consistent QA‑formatted memory units; (2) Missing‑Slot Guided Filtering reranks units by slot-completion gain via a chain‑of‑thought reasoner and fine‑tuned LLaMA‑8B filter. We also introduce the MS-TOD, the first multi-session TOD benchmark with 132 diverse personas, 956 task goals, and annotated intent-aligned memory targets. Evaluations on MS-TOD show that MemGuide boosts task success rate by 11% (88%→99%) and reduces dialogue length by 2.84 turns, and matches single‑session performance.

Published

2026-03-14

How to Cite

Du, Y., Wang, B., He, Y., Liang, B., Wang, B., Li, Z., Gui, L., Pan, J. Z., Xu, R., & Wong, K.-F. (2026). MemGuide: Intent-Driven Memory Selection for Goal-Oriented Multi-Session LLM Agents. Proceedings of the AAAI Conference on Artificial Intelligence, 40(36), 30584-30592. https://doi.org/10.1609/aaai.v40i36.40313

Issue

Section

AAAI Technical Track on Natural Language Processing I