Generalization of Graph Neural Networks through the Lens of Homomorphism

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Li, Shouheng, Kim, Dongwoo, Wang, Qing
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916208021864448
author Li, Shouheng
Kim, Dongwoo
Wang, Qing
author_facet Li, Shouheng
Kim, Dongwoo
Wang, Qing
contents Despite the celebrated popularity of Graph Neural Networks (GNNs) across numerous applications, the ability of GNNs to generalize remains less explored. In this work, we propose to study the generalization of GNNs through a novel perspective - analyzing the entropy of graph homomorphism. By linking graph homomorphism with information-theoretic measures, we derive generalization bounds for both graph and node classifications. These bounds are capable of capturing subtleties inherent in various graph structures, including but not limited to paths, cycles and cliques. This enables a data-dependent generalization analysis with robust theoretical guarantees. To shed light on the generality of of our proposed bounds, we present a unifying framework that can characterize a broad spectrum of GNN models through the lens of graph homomorphism. We validate the practical applicability of our theoretical findings by showing the alignment between the proposed bounds and the empirically observed generalization gaps over both real-world and synthetic datasets.
format Preprint
id arxiv_https___arxiv_org_abs_2403_06079
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Generalization of Graph Neural Networks through the Lens of Homomorphism
Li, Shouheng
Kim, Dongwoo
Wang, Qing
Machine Learning
Despite the celebrated popularity of Graph Neural Networks (GNNs) across numerous applications, the ability of GNNs to generalize remains less explored. In this work, we propose to study the generalization of GNNs through a novel perspective - analyzing the entropy of graph homomorphism. By linking graph homomorphism with information-theoretic measures, we derive generalization bounds for both graph and node classifications. These bounds are capable of capturing subtleties inherent in various graph structures, including but not limited to paths, cycles and cliques. This enables a data-dependent generalization analysis with robust theoretical guarantees. To shed light on the generality of of our proposed bounds, we present a unifying framework that can characterize a broad spectrum of GNN models through the lens of graph homomorphism. We validate the practical applicability of our theoretical findings by showing the alignment between the proposed bounds and the empirically observed generalization gaps over both real-world and synthetic datasets.
title Generalization of Graph Neural Networks through the Lens of Homomorphism
topic Machine Learning
url https://arxiv.org/abs/2403.06079