What Do GNNs Actually Learn? Towards Understanding their Representations

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Nikolentzos, Giannis, Chatzianastasis, Michail, Vazirgiannis, Michalis
Format: Preprint
Publié: 2023
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866916491251679232
author Nikolentzos, Giannis
Chatzianastasis, Michail
Vazirgiannis, Michalis
author_facet Nikolentzos, Giannis
Chatzianastasis, Michail
Vazirgiannis, Michalis
contents In recent years, graph neural networks (GNNs) have achieved great success in the field of graph representation learning. Although prior work has shed light on the expressiveness of those models (\ie whether they can distinguish pairs of non-isomorphic graphs), it is still not clear what structural information is encoded into the node representations that are learned by those models. In this paper, we address this gap by studying the node representations learned by four standard GNN models. We find that some models produce identical representations for all nodes, while the representations learned by other models are linked to some notion of walks of specific length that start from the nodes. We establish Lipschitz bounds for these models with respect to the number of (normalized) walks. Additionally, we investigate the influence of node features on the learned representations. We find that if the initial representations of all nodes point in the same direction, the representations learned at the $k$-th layer of the models are also related to the initial features of nodes that can be reached in exactly $k$ steps. We also apply our findings to understand the phenomenon of oversquashing that occurs in GNNs. Our theoretical analysis is validated through experiments on synthetic and real-world datasets.
format Preprint
id arxiv_https___arxiv_org_abs_2304_10851
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle What Do GNNs Actually Learn? Towards Understanding their Representations
Nikolentzos, Giannis
Chatzianastasis, Michail
Vazirgiannis, Michalis
Machine Learning
In recent years, graph neural networks (GNNs) have achieved great success in the field of graph representation learning. Although prior work has shed light on the expressiveness of those models (\ie whether they can distinguish pairs of non-isomorphic graphs), it is still not clear what structural information is encoded into the node representations that are learned by those models. In this paper, we address this gap by studying the node representations learned by four standard GNN models. We find that some models produce identical representations for all nodes, while the representations learned by other models are linked to some notion of walks of specific length that start from the nodes. We establish Lipschitz bounds for these models with respect to the number of (normalized) walks. Additionally, we investigate the influence of node features on the learned representations. We find that if the initial representations of all nodes point in the same direction, the representations learned at the $k$-th layer of the models are also related to the initial features of nodes that can be reached in exactly $k$ steps. We also apply our findings to understand the phenomenon of oversquashing that occurs in GNNs. Our theoretical analysis is validated through experiments on synthetic and real-world datasets.
title What Do GNNs Actually Learn? Towards Understanding their Representations
topic Machine Learning
url https://arxiv.org/abs/2304.10851