MASPRM: Multi-Agent System Process Reward Model

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Yazdani, Milad, Mostajabdaveh, Mahdi, Zhou, Zirui, Xiong, Ying
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866917271573626880
author Yazdani, Milad
Mostajabdaveh, Mahdi
Zhou, Zirui
Xiong, Ying
author_facet Yazdani, Milad
Mostajabdaveh, Mahdi
Zhou, Zirui
Xiong, Ying
contents Practical deployment of multi-agent systems (MAS) demands strong performance at test time, motivating methods that guide search during inference and selectively spend compute to improve quality. We present the Multi-Agent System Process Reward Model (MASPRM). It assigns values to partial inter-agent transcripts for each action and each agent, and acts as a controller during inference. MASPRM is trained from multi-agent Monte Carlo Tree Search (MCTS) rollouts labeled only with terminal outcome rewards, without requiring human step-level annotations, by propagating returns to local targets. During inference, MASPRM guides step-level beam search (SBS) and MCTS, focusing computation on promising branches and pruning unpromising ones. We train and test MASPRM across different tasks and domains, using GSM8K, MATH, MMLU, and LogiQA as benchmarks. Averaged across these benchmarks, MASPRM improves Hit@1 over policy likelihood by up to $+13.4$ points and improves ranking quality, reducing Hit@1$->$Hit@5 gaps by up to $10.3$ points. MASPRM complements inference-time search by scoring intermediate routed transcripts to guide rollouts in MAS with fixed schedules. Code: https://github.com/milad1378yz/MASPRM
format Preprint
id arxiv_https___arxiv_org_abs_2510_24803
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle MASPRM: Multi-Agent System Process Reward Model
Yazdani, Milad
Mostajabdaveh, Mahdi
Zhou, Zirui
Xiong, Ying
Multiagent Systems
Artificial Intelligence
Practical deployment of multi-agent systems (MAS) demands strong performance at test time, motivating methods that guide search during inference and selectively spend compute to improve quality. We present the Multi-Agent System Process Reward Model (MASPRM). It assigns values to partial inter-agent transcripts for each action and each agent, and acts as a controller during inference. MASPRM is trained from multi-agent Monte Carlo Tree Search (MCTS) rollouts labeled only with terminal outcome rewards, without requiring human step-level annotations, by propagating returns to local targets. During inference, MASPRM guides step-level beam search (SBS) and MCTS, focusing computation on promising branches and pruning unpromising ones. We train and test MASPRM across different tasks and domains, using GSM8K, MATH, MMLU, and LogiQA as benchmarks. Averaged across these benchmarks, MASPRM improves Hit@1 over policy likelihood by up to $+13.4$ points and improves ranking quality, reducing Hit@1$->$Hit@5 gaps by up to $10.3$ points. MASPRM complements inference-time search by scoring intermediate routed transcripts to guide rollouts in MAS with fixed schedules. Code: https://github.com/milad1378yz/MASPRM
title MASPRM: Multi-Agent System Process Reward Model
topic Multiagent Systems
Artificial Intelligence
url https://arxiv.org/abs/2510.24803