Beyond Weisfeiler-Lehman: A Quantitative Framework for GNN Expressiveness

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Zhang, Bohang, Gai, Jingchu, Du, Yiheng, Ye, Qiwei, He, Di, Wang, Liwei
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866916093459693568
author Zhang, Bohang
Gai, Jingchu
Du, Yiheng
Ye, Qiwei
He, Di
Wang, Liwei
author_facet Zhang, Bohang
Gai, Jingchu
Du, Yiheng
Ye, Qiwei
He, Di
Wang, Liwei
contents Designing expressive Graph Neural Networks (GNNs) is a fundamental topic in the graph learning community. So far, GNN expressiveness has been primarily assessed via the Weisfeiler-Lehman (WL) hierarchy. However, such an expressivity measure has notable limitations: it is inherently coarse, qualitative, and may not well reflect practical requirements (e.g., the ability to encode substructures). In this paper, we introduce a unified framework for quantitatively studying the expressiveness of GNN architectures, addressing all the above limitations. Specifically, we identify a fundamental expressivity measure termed homomorphism expressivity, which quantifies the ability of GNN models to count graphs under homomorphism. Homomorphism expressivity offers a complete and practical assessment tool: the completeness enables direct expressivity comparisons between GNN models, while the practicality allows for understanding concrete GNN abilities such as subgraph counting. By examining four classes of prominent GNNs as case studies, we derive simple, unified, and elegant descriptions of their homomorphism expressivity for both invariant and equivariant settings. Our results provide novel insights into a series of previous work, unify the landscape of different subareas in the community, and settle several open questions. Empirically, extensive experiments on both synthetic and real-world tasks verify our theory, showing that the practical performance of GNN models aligns well with the proposed metric.
format Preprint
id arxiv_https___arxiv_org_abs_2401_08514
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Beyond Weisfeiler-Lehman: A Quantitative Framework for GNN Expressiveness
Zhang, Bohang
Gai, Jingchu
Du, Yiheng
Ye, Qiwei
He, Di
Wang, Liwei
Machine Learning
Discrete Mathematics
Data Structures and Algorithms
Combinatorics
Designing expressive Graph Neural Networks (GNNs) is a fundamental topic in the graph learning community. So far, GNN expressiveness has been primarily assessed via the Weisfeiler-Lehman (WL) hierarchy. However, such an expressivity measure has notable limitations: it is inherently coarse, qualitative, and may not well reflect practical requirements (e.g., the ability to encode substructures). In this paper, we introduce a unified framework for quantitatively studying the expressiveness of GNN architectures, addressing all the above limitations. Specifically, we identify a fundamental expressivity measure termed homomorphism expressivity, which quantifies the ability of GNN models to count graphs under homomorphism. Homomorphism expressivity offers a complete and practical assessment tool: the completeness enables direct expressivity comparisons between GNN models, while the practicality allows for understanding concrete GNN abilities such as subgraph counting. By examining four classes of prominent GNNs as case studies, we derive simple, unified, and elegant descriptions of their homomorphism expressivity for both invariant and equivariant settings. Our results provide novel insights into a series of previous work, unify the landscape of different subareas in the community, and settle several open questions. Empirically, extensive experiments on both synthetic and real-world tasks verify our theory, showing that the practical performance of GNN models aligns well with the proposed metric.
title Beyond Weisfeiler-Lehman: A Quantitative Framework for GNN Expressiveness
topic Machine Learning
Discrete Mathematics
Data Structures and Algorithms
Combinatorics
url https://arxiv.org/abs/2401.08514