PRIME: Planning and Retrieval-Integrated Memory for Enhanced Reasoning
Abstract
Inspired by the dual-process theory of human cognition from \textit{Thinking, Fast and Slow}, we introduce \textbf{PRIME} (Planning and Retrieval-Integrated Memory for Enhanced Reasoning), a multi-agent reasoning framework that dynamically integrates \textbf{System 1} (fast, intuitive thinking) and \textbf{System 2} (slow, deliberate thinking). PRIME first employs a Quick Thinking Agent (System 1) to generate a rapid answer; if uncertainty is detected, it then triggers a structured System 2 reasoning pipeline composed of specialized agents for \textit{planning}, \textit{hypothesis generation}, \textit{retrieval}, \textit{information integration}, and \textit{decision-making}. This multi-agent design faithfully mimics human cognitive processes and enhances both efficiency and accuracy. Experimental results with LLaMA 3 models demonstrate that PRIME enables open-source LLMs to perform competitively with state-of-the-art closed-source models like GPT-4 and GPT-4o on benchmarks requiring multi-hop and knowledge-grounded reasoning. This research establishes PRIME as a scalable solution for improving LLMs in domains requiring complex, knowledge-intensive reasoning.
Cite
@article{arxiv.2509.22315,
title = {PRIME: Planning and Retrieval-Integrated Memory for Enhanced Reasoning},
author = {Hieu Tran and Zonghai Yao and Nguyen Luong Tran and Zhichao Yang and Feiyun Ouyang and Shuo Han and Razieh Rahimi and Hong Yu},
journal= {arXiv preprint arXiv:2509.22315},
year = {2025}
}
Comments
Proceedings of AAAI 2026