cs.AIOct 17, 2026

Train Locally, Learn Together: Distributed Multi-Agent Training via ADMM

Authors: Yiwei Chen, Changsheng Wang, Yicheng Lang, Qing Qu, Alfred O. Hero, Sijia Liu

Organizations: Michigan State University · University of Michigan, Ann Arbor

Abstract

Multi-agent systems (MAS) can outperform individual language-model agents by combining specialized roles, diverse reasoning processes, and collaborative decision-making. However, collaborative MAS training is typically centralized, requiring a common trainer to aggregate role-local trajectories and learning signals and maintain model and optimization states. This motivates a distributed multi-agent learning paradigm in which agents improve system-level capabilities through role-local learning and decentralized communication, paving the way toward scalable, resilient, and self-organizing MAS. To this end, we introduce, to the best of our knowledge, the first optimization-grounded framework for distributed MAS training based on consensus alternating direction method of multipliers (ADMM). We formulate collaborative MAS learning as a consensus-constrained finite-sum optimization problem, naturally lending itself to distributed optimization via ADMM. The proposed approach alternates role-local learning with inter-agent consensus using distributed communication protocols such as ring all-reduce, enabling joint optimization while keeping each agent's data and training state local. We evaluate our framework on distributed safety alignment and general reasoning tasks. Under compromised-agent settings, ADMM can reduce HarmBench attack success rate (ASR) by up to 26.4%, while achieving reasoning gains comparable to centralized training. These results show that distributed MAS training retains the benefits of centralized joint optimization without requiring a centralized trainer.

Figures & tables

Explore similar work

CardsList
  1. MAS-OPD: On-Policy Distillation for Multi-agent Systems

    Sep 28, 2026Qiyong Zhong, Mao Zheng, Mingyang Song +5Multi-Agent CoordinationMulti-Agent Collaboration

  2. Learning to Communicate: Toward End-to-End Optimization of Multi-Agent Language Systems

    Apr 23, 2026Ye Yu, Heming Liu, Haibo Jin +3Multi-Agent System OptimizationLLM Agent Training

  3. Learning to Act and Cooperate for Distributed Black-Box Consensus Optimization

    May 1, 2026Zi-Bo Qin, Feng-Feng Wei, Tai-You Chen +1Multi-Agent System OptimizationMulti-Agent Coordination