Graph Neural Networks Gone Hogwild

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Solodova, Olga, Richardson, Nick, Oktay, Deniz, Adams, Ryan P.
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913953499578368
author Solodova, Olga
Richardson, Nick
Oktay, Deniz
Adams, Ryan P.
author_facet Solodova, Olga
Richardson, Nick
Oktay, Deniz
Adams, Ryan P.
contents Graph neural networks (GNNs) appear to be powerful tools to learn state representations for agents in distributed, decentralized multi-agent systems, but generate catastrophically incorrect predictions when nodes update asynchronously during inference. This failure under asynchrony effectively excludes these architectures from many potential applications where synchrony is difficult or impossible to enforce, e.g., robotic swarms or sensor networks. In this work we identify "implicitly-defined" GNNs as a class of architectures which is provably robust to asynchronous "hogwild" inference, adapting convergence guarantees from work in asynchronous and distributed optimization. We then propose a novel implicitly-defined GNN architecture, which we call an 'energy GNN'. We show that this architecture outperforms other GNNs from this class on a variety of synthetic tasks inspired by multi-agent systems.
format Preprint
id arxiv_https___arxiv_org_abs_2407_00494
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Graph Neural Networks Gone Hogwild
Solodova, Olga
Richardson, Nick
Oktay, Deniz
Adams, Ryan P.
Machine Learning
Distributed, Parallel, and Cluster Computing
Graph neural networks (GNNs) appear to be powerful tools to learn state representations for agents in distributed, decentralized multi-agent systems, but generate catastrophically incorrect predictions when nodes update asynchronously during inference. This failure under asynchrony effectively excludes these architectures from many potential applications where synchrony is difficult or impossible to enforce, e.g., robotic swarms or sensor networks. In this work we identify "implicitly-defined" GNNs as a class of architectures which is provably robust to asynchronous "hogwild" inference, adapting convergence guarantees from work in asynchronous and distributed optimization. We then propose a novel implicitly-defined GNN architecture, which we call an 'energy GNN'. We show that this architecture outperforms other GNNs from this class on a variety of synthetic tasks inspired by multi-agent systems.
title Graph Neural Networks Gone Hogwild
topic Machine Learning
Distributed, Parallel, and Cluster Computing
url https://arxiv.org/abs/2407.00494