Skip to main navigation Skip to search Skip to main content

MARS: Multi-Agent Adaptive Reasoning with Socratic Guidance for Automated Prompt Optimization

  • Jian Zhang
  • , Zhangqi Wang
  • , Haiping Zhu
  • , Kangda Cheng
  • , Kai He
  • , Bo Li
  • , Qika Lin
  • , Jun Liu
  • , Erik Cambria
  • Xi'an Jiaotong University
  • Harbin Institute of Technology
  • National University of Singapore
  • Nanyang Technological University

Research output: Contribution to journalConference articlepeer-review

5 Scopus citations

Abstract

Large language models typically operate in a question-answering paradigm, where the quality of the input prompt critically affects the response. Automated Prompt Optimization (APO) aims to overcome the cognitive biases of manually crafted prompts and explore a broader prompt design space. However, existing APO methods often suffer from rigid template structures and inefficient exploration in the prompt space. To this end, we propose a Multi-Agent Adaptive Reasoning with Socratic guidance framework (MARS). It consists of five complementary agents and formulates the optimization process as a Partially Observable Markov Decision Process, enabling adaptive prompt refinement through explicit state modeling and interactive feedback. Specifically, a Planner agent generates flexible optimization trajectories, a Teacher-Critic-Student triad engages in Socratic-style dialogue to iteratively optimize the prompt based on pseudo-gradient signals in the text space, and a Target agent executes the prompt in downstream tasks to provide performance feedback. MARS integrates reasoning, feedback, and state transition into a unified hidden-state evolution process, improving both the effectiveness and interpretability of optimization. Extensive experiments across multiple datasets show that MARS outperforms existing APO methods in optimization, efficiency, and interpretability.

Original languageEnglish
Pages (from-to)16307-16315
Number of pages9
JournalProceedings of the AAAI Conference on Artificial Intelligence
Volume40
Issue number19
DOIs
StatePublished - 2026
Event40th AAAI Conference on Artificial Intelligence, AAAI 2026 - Singapore, Singapore
Duration: 20 Jan 202627 Jan 2026

Fingerprint

Dive into the research topics of 'MARS: Multi-Agent Adaptive Reasoning with Socratic Guidance for Automated Prompt Optimization'. Together they form a unique fingerprint.

Cite this