RegExplainer: Generating Explanations for Graph Neural Networks in Regression Tasks

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Zhang, Jiaxing, Chen, Zhuomin, Mei, Hao, Da, Longchao, Luo, Dongsheng, Wei, Hua
Natura: Preprint
Pubblicazione: 2023
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866912083700875264
author Zhang, Jiaxing
Chen, Zhuomin
Mei, Hao
Da, Longchao
Luo, Dongsheng
Wei, Hua
author_facet Zhang, Jiaxing
Chen, Zhuomin
Mei, Hao
Da, Longchao
Luo, Dongsheng
Wei, Hua
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2307_07840
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle RegExplainer: Generating Explanations for Graph Neural Networks in Regression Tasks
Zhang, Jiaxing
Chen, Zhuomin
Mei, Hao
Da, Longchao
Luo, Dongsheng
Wei, Hua
Machine Learning
Artificial Intelligence
I.2.0
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.
title RegExplainer: Generating Explanations for Graph Neural Networks in Regression Tasks
topic Machine Learning
Artificial Intelligence
I.2.0
url https://arxiv.org/abs/2307.07840