Perfect Recovery for Random Geometric Graph Matching with Shallow Graph Neural Networks

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Liu, Suqi, Austern, Morgane
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915190254075904
author Liu, Suqi
Austern, Morgane
author_facet Liu, Suqi
Austern, Morgane
contents We study the graph matching problem in the presence of vertex feature information using shallow graph neural networks. Specifically, given two graphs that are independent perturbations of a single random geometric graph with sparse binary features, the task is to recover an unknown one-to-one mapping between the vertices of the two graphs. We show under certain conditions on the sparsity and noise level of the feature vectors, a carefully designed two-layer graph neural network can, with high probability, recover the correct mapping between the vertices with the help of the graph structure. Additionally, we prove that our condition on the noise parameter is tight up to logarithmic factors. Finally, we compare the performance of the graph neural network to directly solving an assignment problem using the noisy vertex features and demonstrate that when the noise level is at least constant, this direct matching fails to achieve perfect recovery, whereas the graph neural network can tolerate noise levels growing as fast as a power of the size of the graph. Our theoretical findings are further supported by numerical studies as well as real-world data experiments.
format Preprint
id arxiv_https___arxiv_org_abs_2402_07340
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Perfect Recovery for Random Geometric Graph Matching with Shallow Graph Neural Networks
Liu, Suqi
Austern, Morgane
Machine Learning
Information Theory
Social and Information Networks
Probability
Statistics Theory
We study the graph matching problem in the presence of vertex feature information using shallow graph neural networks. Specifically, given two graphs that are independent perturbations of a single random geometric graph with sparse binary features, the task is to recover an unknown one-to-one mapping between the vertices of the two graphs. We show under certain conditions on the sparsity and noise level of the feature vectors, a carefully designed two-layer graph neural network can, with high probability, recover the correct mapping between the vertices with the help of the graph structure. Additionally, we prove that our condition on the noise parameter is tight up to logarithmic factors. Finally, we compare the performance of the graph neural network to directly solving an assignment problem using the noisy vertex features and demonstrate that when the noise level is at least constant, this direct matching fails to achieve perfect recovery, whereas the graph neural network can tolerate noise levels growing as fast as a power of the size of the graph. Our theoretical findings are further supported by numerical studies as well as real-world data experiments.
title Perfect Recovery for Random Geometric Graph Matching with Shallow Graph Neural Networks
topic Machine Learning
Information Theory
Social and Information Networks
Probability
Statistics Theory
url https://arxiv.org/abs/2402.07340