Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

RegExplainer: Generating Explanations for Graph Neural Networks in Regression Tasks

About

Graph regression is a fundamental task that has gained significant attention in various graph learning tasks. However, the inference process is often not easily interpretable. Current explanation techniques are limited to understanding Graph Neural Network (GNN) behaviors in classification tasks, leaving an explanation gap for graph regression models. In this work, we propose a novel explanation method to interpret the graph regression models (XAIG-R). Our method addresses the distribution shifting problem and continuously ordered decision boundary issues that hinder existing methods away from being applied in regression tasks. We introduce a novel objective based on the graph information bottleneck theory (GIB) and a new mix-up framework, which can support various GNNs and explainers in a model-agnostic manner. Additionally, we present a self-supervised learning strategy to tackle the continuously ordered labels in regression tasks. We evaluate our proposed method on three benchmark datasets and a real-life dataset introduced by us, and extensive experiments demonstrate its effectiveness in interpreting GNN models in regression tasks.

Jiaxing Zhang, Zhuomin Chen, Hao Mei, Longchao Da, Dongsheng Luo, Hua Wei• 2023

Related benchmarks

TaskDatasetResultRank
Distribution-shift evaluationTRIANGLES
Cosine Similarity0.953
8
Graph RegressionBA-Motif-Counting MultipleMotif (test)
AUC-ROC0.946
6
Graph RegressionBA-Motif-Volume SingleMotif (test)
AUC-ROC76.6
6
Graph RegressionHouse-Grid-Volume SingleMotif (test)
AUC-ROC85.8
6
Graph RegressionHouse-OrGrid-Volume MultipleMotif (test)
AUC-ROC74.1
6
Graph RegressionTriangles MultipleMotif (test)
AUC-ROC0.56
6
Graph RegressionCrippen MultipleMotif (test)
AUC-ROC49.7
6
Distribution-shift evaluationBA-Motif-Counting
Cosine Similarity0.95
5
Distribution-shift evaluationBA-Motif-Volume
Cosine Similarity0.876
5
Distribution-shift evaluationCrippen
Cosine Similarity0.847
5
Showing 10 of 14 rows

Other info

Follow for update