Test-Time Collaborative Classification over Multi-Agent Networks
Organizations: School of Engineering, École Polytechnique Fédérale de Lausanne (EPFL), 1015 Lausanne, Switzerland · DTI, SUPSI, Dalle Molle Institute for Artificial Intelligence (IDSIA USI-SUPSI), 6962 Lugano, Switzerland
Abstract
The increasing heterogeneity of multi-agent systems poses significant challenges for jointly training a global model across agents. At the same time, cooperative inference between agents has long been recognized as a powerful mechanism for distributed decision making over networks. Motivated by these observations, we propose a collaboration framework for distributed binary classification over multi-agent networks, where a set of independently trained agents, potentially differing in architecture, feature space, or modality, coordinate their actions during test time to form collective predictions. This coordination is achieved by exchanging local decision statistics through a distributed learning protocol. We develop a theoretical and experimental study of this independent training and cooperative inference paradigm, and examine its performance under different communication budgets and distributed learning rules. We establish classification error guarantees under sufficient, finite-round, and finite-precision communication, together with PAC-style generalization bounds. These results capture the influence of model heterogeneity, network topology, combination policy, and communication constraints on prediction accuracy. Taken together with the experimental results, they reveal both the price of independent training and the benefit of collective prediction for the proposed distributed decision making framework with models learned from data.
Figures & tables
| Method | Training coordination | Additional fitting | Inference coordination |
|---|---|---|---|
| Non-cooperative | None | None | None |
| Avg-stat / Vote | None | None | One-shot centralized averaging / voting |
| Learned (simplex) fusion | None | Fusion rule fitted on aligned validation outputs | One-shot centralized fusion |
| AdaBoost | Centralized | Boosting-based fitting of local predictors | One-shot weighted fusion |
| Ours / Ours (no centering) | None | None | Iterative peer-to-peer scalar exchange |
| VFL-JT | Joint training on aligned labeled samples | Server-side fusion head learned jointly | One-shot server fusion |
| Ping Hu received her Ph.D. in electrical engineering from EPFL, Switzerland in 2026. She is currently a postdoctoral researcher with the Adaptive Systems Laboratory, EPFL. Her research interests include distributed learning, inference and decision-making over networked systems. |
| Mert Kayaalp received his Ph.D. in computer and communication sciences from EPFL, Switzerland in 2025. He is currently a research scientist at Dalle Molle Institute for Artificial Intelligence (IDSIA USI-SUPSI), Switzerland, where he is affiliated with the UBS-IDSIA AI Lab. His research focuses on machine learning and signal analysis in networked and dynamical systems. |
| Ali H. Sayed is professor and former dean of engineering at EPFL, Switzerland, where he also directs the Adaptive Systems Laboratory. He served before as a distinguished professor and chair of electrical engineering with UCLA. He is a member of the US National Academy of Engineering (NAE) and The World Academy of Sciences (TWAS). He served as a president of the IEEE Signal Processing Society in 2018 and 2019. An author of over 650 scholarly publications and 10 books, his research involves several areas including adaptation and learning theories, statistical inference, and multi-agent systems. His work has been recognized with several awards including the 2022 IEEE Fourier Technical Field Award and the 2020 IEEE Wiener Society Award. He is also a fellow of EURASIP and the American Association for the Advancement of Science (AAAS). |
Appendix figures & tables10 assets
Supplementary material from the paper’s appendix.
Appendix
| Method | Training | Inference per sample |
|---|---|---|
| VFL-JT | ||
| Proposed |
Multi-Agent Collaboration
Multi-agent collaboration studies how autonomous agents coordinate, share information, and combine capabilities to solve tasks together.
See all 312 papers