Cooperative UxS Autonomy with Graph Neural Networks

This paper presents a spatio-temporal graph neural network (STGNN) centric approach to enable heterogeneous agents to collaborate and cooperate for different types of missions. The STGNN-centric approach and corresponding autonomy are encapsulated in the Advanced Graph-enabled Network Technology for Collaborative Autonomous Agents (AGENTCA) technology. Various decentralized and distributed control architectures are reported in the literature, but in some instances these approaches do not leverage the inherent graph network which can increase scalability to larger teams and algorithmic efficiency. Specifically, in this paper advances in artificial intelligence are leveraged to parameterize and encode optimal, or nearly optimal, swarm control techniques. For this work, the team focused on developing a diffusion-based STGNN swarm controller using imitation learning. An expert, centralized swarm control law was used to guide the STGNN during the learning process. The STGNN controller enables the swarm to follow a leader while avoiding static and dynamic obstacles and maintaining a desired separation distance from neighbors and obstacles. The approach is demonstrated in simulation with hundreds of agents and in flight tests with up to thirteen test vehicles.