arXiv · 2206.11010
Agent-based Graph Neural Networks
Abstract
We present a novel graph neural network we call AgentNet, which is designed specifically for graph-level tasks. AgentNet is inspired by sublinear algorithms, featuring a computational complexity that is independent of the graph size. The architecture of AgentNet differs fundamentally from the architectures of traditional graph neural networks. In AgentNet, some trained \textit{neural agents} intelligently walk the graph, and then collectively decide on the output. We provide an extensive theoretical analysis of AgentNet: We show that the agents can learn to systematically explore their neighborhood and that AgentNet can distinguish some structures that are even indistinguishable by 2-WL. Moreover, AgentNet is able to separate any two graphs which are sufficiently different in terms of subgraphs. We confirm these theoretical results with synthetic experiments on hard-to-distinguish graphs and real-world graph classification tasks. In both cases, we compare favorably not only to standard GNNs but also to computationally more expensive GNN extensions.
Explore related subjects
Keep this discovery
Karolis Martinkus, Pál András Papp, Benedikt Schesch, Roger Wattenhofer. 2022-06-22. Agent-based Graph Neural Networks. https://arxiv.org/abs/2206.11010
Cite the original work for its findings. Save a collection to share your selection of sources.