Sample data: Performance prediction of hub-based swarms
Data files
Nov 12, 2024 version files 53.34 MB
-
README.md
1.44 KB
-
RSA-Data-20241022T171335Z-001.zip
53.34 MB
Abstract
There are powerful tools for modeling swarms that have strong spatial structures like flocks of birds, schools of fish, and formations of drones, but relatively little work on developing formalisms for other swarm structures like hub-based colonies doing foraging, maintaining a nest, or selecting a new nest site. We present a method for finding low dimensional representations of swarm state for simulated homogeneous hub-based colonies solving the best-of-N problem. The embeddings are obtained from latent representations of convolution-based graph neural network architectures and have the property that swarm states that have similar performance have very similar embeddings. Such embeddings are used to classify swarm states into binned estimates of success probability and time to completion. We demonstrate how embeddings can be obtained in a sequence of experiments that progressively require less information, which suggests that the methods can be extended to larger swarms in more complicated environments.
https://doi.org/10.5061/dryad.hhmgqnkrp
Description of the data and file structure
The data was synthetically generated by running multi-agent simulations and processing the simulations to create graphs.
Description:
Each folder contains sample data along with the code used to generate the predictions. For data from Experiment 2, please email us so we can regenerate it.
There are additional README files in each subfolder, where necessary.
Experiment 1:
CreateGraph.ipynb: runs the simulation and creates a graph of collective states
GraphAnalysis.ipynb: code for converting graph to Markov Process, creating plots, etc.
GraphEmbeddings.ipynb: code for training a Graph CNN to generate embeddings
data:
/graphs_and_figs/10614_trials_2_sites/graph.gexf
/graphs_and_figs/10614_trials_2_sites/node_info.pickle
Experiment 3:
allGraphs_train: sample graphs for training
allGraphs_test: sample graphs for testing
GAT_multi_experiment.ipynb: sample notebook used to train and test the GAT network with confusion matrix and embedding plotting code, along with F1 score
Sharing/Access information
For clarification/access to other data and code, please contact:
Puneet Jain: puneetjain2895@gmail.com
Michael Goodrich: mike@cs.byu.edu
This is a synthetically generated dataset.
