论文

MASCAL:多智能体协作中的置信度校准训练方法

Learning Calibrated Confidence for Multi-Agent Collaboration

精选理由

多智能体交接老是靠猜答案对不对?这篇论文给每个 agent 加了正确概率输出,下游能据此决定信不信,还能支持选择性作答。

一篇 arXiv 论文提出 MASCAL,让多智能体系统中每个智能体在强化学习时联合优化任务表现和置信度校准,输出答案正确概率。上游智能体在交接时附带该概率,下游智能体可据此决定信任还是验证收到的内容。论文还引入置信度感知加权,让下游智能体更多学习主动使用该信号的决策。在覆盖 Web 检索与推理任务的 3 种协作架构上,MASCAL 同时提升了任务准确率和校准质量,并能推广到更深的协作链和不同模型规模。

原文 · arXiv cs.LG

Learning Calibrated Confidence for Multi-Agent Collaboration

Complex information seeking on the Web often requires collecting evidence from multiple documents and reasoning across them, as in multi-hop question answering. Multi-agent systems address such tasks by dividing work among specialized agents, where upstream agents produce intermediate answers and supporting evidence that downstream agents build upon. However, upstream answers vary in correctness, and most systems provide no quantitative signal about it, leaving the downstream agent to judge whether to trust or verify what it receives based solely on the textual evidence. In this paper, we propose MASCAL. Each agent learns to produce a correctness probability for its answer by jointly optimizing task performance and confidence calibration in reinforcement learning; the upstream agent passes this probability at handoff, giving the downstream agent a quantitative basis for its decisions, and the final answer also carries a calibrated probability, quantifying the reliability of the system output. To guide the downstream agent in using this signal during training, we introduce confidence-aware weighting, which increases the learning weight at positions responsive to the upstream probability, so that the agent learns more from decisions where it actively uses the signal. Across three collaboration architectures spanning Web retrieval and reasoning tasks, MASCAL improves task accuracy and calibration quality for both agents, and generalizes to deeper collaboration chains and different model scales. Intervention analysis confirms that upstream confidence affects downstream decision behavior, and the learned final confidence supports selective answering and targeted rechecking.