Saved in:
Bibliographic Details
Main Authors: Joly, Antonin, Keriven, Nicolas
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2405.18127
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866929362202263552
author Joly, Antonin
Keriven, Nicolas
author_facet Joly, Antonin
Keriven, Nicolas
contents Graph coarsening aims to reduce the size of a large graph while preserving some of its key properties, which has been used in many applications to reduce computational load and memory footprint. For instance, in graph machine learning, training Graph Neural Networks (GNNs) on coarsened graphs leads to drastic savings in time and memory. However, GNNs rely on the Message-Passing (MP) paradigm, and classical spectral preservation guarantees for graph coarsening do not directly lead to theoretical guarantees when performing naive message-passing on the coarsened graph. In this work, we propose a new message-passing operation specific to coarsened graphs, which exhibit theoretical guarantees on the preservation of the propagated signal. Interestingly, and in a sharp departure from previous proposals, this operation on coarsened graphs is oriented, even when the original graph is undirected. We conduct node classification tasks on synthetic and real data and observe improved results compared to performing naive message-passing on the coarsened graph.
format Preprint
id arxiv_https___arxiv_org_abs_2405_18127
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Graph Coarsening with Message-Passing Guarantees
Joly, Antonin
Keriven, Nicolas
Machine Learning
Graph coarsening aims to reduce the size of a large graph while preserving some of its key properties, which has been used in many applications to reduce computational load and memory footprint. For instance, in graph machine learning, training Graph Neural Networks (GNNs) on coarsened graphs leads to drastic savings in time and memory. However, GNNs rely on the Message-Passing (MP) paradigm, and classical spectral preservation guarantees for graph coarsening do not directly lead to theoretical guarantees when performing naive message-passing on the coarsened graph. In this work, we propose a new message-passing operation specific to coarsened graphs, which exhibit theoretical guarantees on the preservation of the propagated signal. Interestingly, and in a sharp departure from previous proposals, this operation on coarsened graphs is oriented, even when the original graph is undirected. We conduct node classification tasks on synthetic and real data and observe improved results compared to performing naive message-passing on the coarsened graph.
title Graph Coarsening with Message-Passing Guarantees
topic Machine Learning
url https://arxiv.org/abs/2405.18127