Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Adaptive Robust Estimator for Multi-Agent Reinforcement Learning

About

Multi-agent collaboration has emerged as a powerful paradigm for enhancing the reasoning capabilities of large language models, yet it suffers from interaction-level ambiguity that blurs generation, critique, and revision, making credit assignment across agents difficult. Moreover, policy optimization in this setting is vulnerable to heavy-tailed and noisy rewards, which can bias advantage estimation and trigger unstable or even divergent training. To address both issues, we propose a robust multi-agent reinforcement learning framework for collaborative reasoning, consisting of two components: Dual-Agent Answer-Critique-Rewrite (DACR) and an Adaptive Robust Estimator (ARE). DACR decomposes reasoning into a structured three-stage pipeline: answer, critique, and rewrite, while enabling explicit attribution of each agent's marginal contribution to its partner's performance. ARE provides robust estimation of batch experience means during multi-agent policy optimization. Across mathematical reasoning and embodied intelligence benchmarks, even under noisy rewards, our method consistently outperforms the baseline in both homogeneous and heterogeneous settings. These results indicate stronger robustness to reward noise and more stable training dynamics, effectively preventing optimization failures caused by noisy reward signals.

Zhongyi Li, Wan Tian, Jingyu Chen, Kangyao Huang, Huiming Zhang, Hui Yang, Tao Ren, Jinyang Jiang, Yijie Peng, Yikun Ban, Fuzhen Zhuang• 2026

Related benchmarks

TaskDatasetResultRank
Mathematical ReasoningMATH 500
Accuracy77.6
442
Mathematical ReasoningMinerva Math
Accuracy30.5
186
Math ReasoningGaoKao En 2023
Accuracy61.8
91
Mathematical ReasoningAMC 23
Accuracy60
81
Aerial Vision-and-Language NavigationAerial VLN seen (val)
Navigation Error (NE)139.5
3
Aerial Vision-and-Language NavigationAerial VLN unseen (val)
Navigation Error (NE)124.4
3
Aerial Vision-and-Language NavigationAerial VLN (test_unseen)
Navigation Error (NE)129.7
3
Showing 7 of 7 rows

Other info

Follow for update