arXiv:2605.14177cs.IRcs.AI2026-05

用未来设想引导记忆检索,提升长时对话个性化效果

Thinking Ahead: Prospection-Guided Retrieval of Memory with Language Models

论文配图:Thinking Ahead: Prospection-Guided Retrieval of Memory with Language Models
图 1 · 摘自论文原文
  • 通过模拟未来对话路径生成检索探针,突破纯语义匹配限制
  • 在多轮对话数据集上实现近3倍召回率提升,尤其对低相似度事实有效
  • 适合需要长期用户记忆的智能助手研发,如个性化客服与陪伴型AI

长时程个性化对话需从用户历史交互中检索特定信息。现实中,许多相关事实与查询语义相似度较低,标准检索增强生成(RAG)和GraphRAG系统仍以查询嵌入相似性或固定图遍历为主,常遗漏嵌入空间中远离查询但对用户重要的事实。受人类‘前瞻’能力启发——即利用想象中的未来情景作为回忆线索,本文提出前瞻引导检索(PGR),将检索与记忆存储解耦。给定用户查询,PGR首先构建简短的思维树(ToT)或线性链,模拟可能的下一步行为,并以这些步骤作为检索探针,而非仅依赖原始查询。这些探针检索到的事实被用于下一轮前瞻性模拟,使系统能逐步发现仅在结合用户历史后才相关的记忆。我们还引入MemoryQuest,一个挑战性的多会话基准,每个查询标注3–5个带时间戳的参考事实,且与查询的语义相似度较低。在来自3个公开数据集的185个用户、共1625个查询上,PGR-TOT相比最强基线在MemoryQuest上召回率近乎提升3倍。在成对大模型评分中,PGR生成回复在89%–98%的查询中更受青睐,盲评人工标注也呈现相同趋势。结果表明,显式前瞻模拟显著优于仅依赖相似性的基线,在长时检索与响应质量上均有大幅提升。

原文摘要 · Abstract (English)

Long-horizon personalization requires dialogue assistants to retrieve user-specific facts from extended interaction histories. In practice, many relevant facts often have low semanticsimilarity to the query under dense retrieval. Standard Retrieval-Augmented Generation (RAG) and GraphRAG systems are still largely retrospective: they rely on embedding similarity to the query or on fixed graph traversals, so they often miss facts that matter for the user's needs but lie far from the query in embedding space. Inspired by prospection, the human ability to use imagined futures as cues for recall, we introduce Prospection-Guided Retrieval (PGR), which decouples retrieval from how memories are stored. Given a user query, PGR first expands the goal into a short Tree-of-Thought (ToT) or linear chain of plausible next steps, and uses these steps as retrieval probes rather than relying on the original query alone. The facts retrieved by these probes are then used to personalize the next round of prospection, enabling PGR to uncover additional memories that become relevant only after the simulation is grounded in the user's history. We also introduce MemoryQuest, a challenging multi-session benchmark in which each query is annotated with 3--5 dated reference facts subject to a low query-reference similarity constraint. Across 1,625 queries spanning 185 user profiles from 3 publicly available datasets, PGR-TOT substantially improves retrieval, including nearly 3x recall on MemoryQuest over the strongest baseline. In pairwise LLM-as-judge comparisons against baselines, PGR-generated responses are preferred on 89--98% of queries, with blinded human annotations on held-out subsets showing the same trend. Overall, the results demonstrate that explicit prospection yields large gains in long-horizon retrieval and response quality relative to similarity-only baselines.

对话系统记忆检索前瞻推理个性化

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。