Performance Prediction of Hub-Based Swarms
Puneet Jain, Chaitanya Dwivedi, Vigynesh Bhatt, Nick Smith, Michael A Goodrich
TL;DR
The paper tackles the challenge of predicting performance in hub-based swarms for the best-of-$N$ problem, where the collective state space grows rapidly with agent count. It introduces a graph-based representation of the ABM, encodes collective-state dynamics as a probabilistic graph, and learns low-dimensional embeddings via a GraphSAGE-based encoder. The authors demonstrate that tensor-derived collective states cluster by their likelihood of success and that 3D embeddings reveal trajectories and basins of attraction predictive of swarm performance. This approach enables scalable analysis, clustering, and potential regulation of swarm behavior across different agent and site configurations, with implications for planning and control in large-scale bio-inspired systems.
Abstract
A hub-based colony consists of multiple agents who share a common nest site called the hub. Agents perform tasks away from the hub like foraging for food or gathering information about future nest sites. Modeling hub-based colonies is challenging because the size of the collective state space grows rapidly as the number of agents grows. This paper presents a graph-based representation of the colony that can be combined with graph-based encoders to create low-dimensional representations of collective state that can scale to many agents for a best-of-N colony problem. We demonstrate how the information in the low-dimensional embedding can be used with two experiments. First, we show how the information in the tensor can be used to cluster collective states by the probability of choosing the best site for a very small problem. Second, we show how structured collective trajectories emerge when a graph encoder is used to learn the low-dimensional embedding, and these trajectories have information that can be used to predict swarm performance.
