Saved in:
Bibliographic Details
Main Authors: Chen, Yongqiang, Bian, Yatao, Han, Bo, Cheng, James
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2406.07955
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909222234488832
author Chen, Yongqiang
Bian, Yatao
Han, Bo
Cheng, James
author_facet Chen, Yongqiang
Bian, Yatao
Han, Bo
Cheng, James
contents Interpretable graph neural networks (XGNNs ) are widely adopted in various scientific applications involving graph-structured data. Existing XGNNs predominantly adopt the attention-based mechanism to learn edge or node importance for extracting and making predictions with the interpretable subgraph. However, the representational properties and limitations of these methods remain inadequately explored. In this work, we present a theoretical framework that formulates interpretable subgraph learning with the multilinear extension of the subgraph distribution, coined as subgraph multilinear extension (SubMT). Extracting the desired interpretable subgraph requires an accurate approximation of SubMT, yet we find that the existing XGNNs can have a huge gap in fitting SubMT. Consequently, the SubMT approximation failure will lead to the degenerated interpretability of the extracted subgraphs. To mitigate the issue, we design a new XGNN architecture called Graph Multilinear neT (GMT), which is provably more powerful in approximating SubMT. We empirically validate our theoretical findings on a number of graph classification benchmarks. The results demonstrate that GMT outperforms the state-of-the-art up to 10% in terms of both interpretability and generalizability across 12 regular and geometric graph benchmarks.
format Preprint
id arxiv_https___arxiv_org_abs_2406_07955
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle How Interpretable Are Interpretable Graph Neural Networks?
Chen, Yongqiang
Bian, Yatao
Han, Bo
Cheng, James
Machine Learning
Interpretable graph neural networks (XGNNs ) are widely adopted in various scientific applications involving graph-structured data. Existing XGNNs predominantly adopt the attention-based mechanism to learn edge or node importance for extracting and making predictions with the interpretable subgraph. However, the representational properties and limitations of these methods remain inadequately explored. In this work, we present a theoretical framework that formulates interpretable subgraph learning with the multilinear extension of the subgraph distribution, coined as subgraph multilinear extension (SubMT). Extracting the desired interpretable subgraph requires an accurate approximation of SubMT, yet we find that the existing XGNNs can have a huge gap in fitting SubMT. Consequently, the SubMT approximation failure will lead to the degenerated interpretability of the extracted subgraphs. To mitigate the issue, we design a new XGNN architecture called Graph Multilinear neT (GMT), which is provably more powerful in approximating SubMT. We empirically validate our theoretical findings on a number of graph classification benchmarks. The results demonstrate that GMT outperforms the state-of-the-art up to 10% in terms of both interpretability and generalizability across 12 regular and geometric graph benchmarks.
title How Interpretable Are Interpretable Graph Neural Networks?
topic Machine Learning
url https://arxiv.org/abs/2406.07955