← 返回信息流
技术博客arXiv cs.CL·3 小时前

PraMem利用实践经验记忆实现长时行为预测

原标题:PraMem: Practice-derived Experiential Memory for Long-horizon Behavior Prediction

速览

长时行为预测面临历史序列负担和LLM认知偏差的挑战。现有方法压缩上下文但未解决核心问题。PraMem改变范式,将长序列视为资源,提前构建实践经验记忆作为辅助输入。实验证明其性能优于现有方法,且对经验记忆机制提供了深入分析。

AI 深度解读

背景

长期行为预测(Long-horizon Behavior Prediction)旨在根据用户的长序列历史行为推断其下一步动作,是人工智能领域的关键任务。随着大型语言模型(LLM)的兴起,为序列行为预测提供了新的方向,但LLM在应对长期行为预测时面临两个核心挑战:一是难以从长序列中归纳潜在的 behavioral pattern(行为模式),二是模型自身固有的认知偏差(model-intrinsic cognitive biases)。以往的记忆管理方法遵循“上下文压缩”范式(context-compression paradigm),试图通过减轻历史序列的负担来解决任务,但未能真正攻克上述核心难题。

核心内容

本文主张一种范式转换:将冗长的历史序列从“负担”重新定义为可供挖掘的“宝贵资源”。基于此,作者提出 PraMem(Practice-derived Experiential Memory),即“基于实践的经验记忆”。该方法在预测之前,先对用户的长序列历史行为进行“事先练习”(beforehand practice),从而构建出一份“经验记忆”(experiential memory),并将其作为辅助输入,帮助模型更准确地完成长期行为预测。

具体而言,PraMem 的工作流程如下:

  1. 事先练习阶段:利用 LLM 在长历史序列上进行类似“预演”的操作,提取并结构化其中的行为模式(如重复规则、周期性偏好、情境依赖等),形成紧凑的“经验记忆”表示。
  2. 联合推理阶段:将原始历史序列与经验记忆一同作为输入,供 LLM 进行最终的预测,使模型既能获得完整的上下文信息,又能直接利用通过练习提炼出的关键模式。

该方法的创新在于:不再是压缩或丢弃历史序列,而是主动利用 LLM 的推理能力从中“开采”可复用的模式记忆,从而规避 LLM 对长序列的注意力衰减和认知偏差。

论文在多种不同任务上进行了广泛实验,结果表明 PraMem 显著优于以往的记忆管理方法。进一步的深入分析(in-depth analyses)揭示了经验记忆的内部机制及其随练习轮次演化的规律,为理解 LLM 如何在长序列中学习与泛化提供了有价值的洞察。代码已开源(见原文链接)。

关键要点

  • 问题定义:长期行为预测的核心瓶颈在于 LLM 对长序列的行为模式归纳能力不足,以及模型固有的认知偏差。
  • 范式转变:从“上下文压缩”转向“历史资源开采”,将长序列视为可被利用的经验来源而非负担。
  • 核心方法:PraMem 通过事先在历史序列上执行“练习”步骤,构建出结构化的经验记忆,作为辅助输入。
  • 实验验证:在多个任务上取得比传统压缩式记忆方法更好的性能。
  • 机制分析:论文提供的深度分析揭示了经验记忆的构建原理与演化过程,为后续研究提供参考。
  • 代码开源:作者公开了 PraMem 的实现代码,便于复现与扩展研究。

意义与影响

PraMem 提出的“练习式经验记忆”为 LLM 在长期行为预测方向开辟了一条新路径。其核心贡献在于证明了“将长序列视为资源而非负担”的可行性,这颠覆了此前主流的上下文压缩思路。对于实际应用(如个性化推荐、用户意图预测、自动驾驶中的意图推断等),PraMem 可能带来更准确的长期预测能力,同时无需大幅增加推理成本(因为经验记忆可在离线或低频率下构建)。此外,论文对经验记忆机制的深入剖析,有助于理解 LLM 如何从数据中抽象规律,为可解释性研究提供了新的视角。未来工作可进一步探索经验记忆的跨域迁移、自适应更新以及与其他记忆架构(如神经图灵机、检索增强生成)的融合。

查看原文 →arxiv.org