This study presents MeCog (Memory-Enhanced Cognitive planning), a framework addressing long-term memory challenges in LLM-based decision-making agents. We use a smaller language model (BERT) as long-term memory within an Actor-Critic reinforcement learning architecture, while short-term memory stores game-state information. Experiments on four TextWorld game types show MeCog outperforms existing methods, achieving 12% higher scores and 16% fewer steps to completion. Ablation studies confirm both memory components’ contributions, with long-term memory particularly impacting tasks requiring strategic planning. Our method enhances LLM capabilities without compromising general abilities, maintaining efficiency through lightweight memory architecture.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Memory-Enhanced Cognitive Planning: A Framework for Improving Long-Term Planning in LLMs

  • Yiqun Niu,
  • Zhongheng Wu,
  • Chongjun Wang

摘要

This study presents MeCog (Memory-Enhanced Cognitive planning), a framework addressing long-term memory challenges in LLM-based decision-making agents. We use a smaller language model (BERT) as long-term memory within an Actor-Critic reinforcement learning architecture, while short-term memory stores game-state information. Experiments on four TextWorld game types show MeCog outperforms existing methods, achieving 12% higher scores and 16% fewer steps to completion. Ablation studies confirm both memory components’ contributions, with long-term memory particularly impacting tasks requiring strategic planning. Our method enhances LLM capabilities without compromising general abilities, maintaining efficiency through lightweight memory architecture.