arrow
返回

Rationalizing Graph Neural Networks with Data Augmentation

delete2024-02-13
delete1
PRE
AI
G
Gang Liu *
E
Eric Inae
Tengfei Luo 封面图
Tengfei Luo (Tengfei Luo)
J
Jiang, Meng
DOI:10.1145/3638781delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
Graph rationales are representative subgraph structures that best explain and support the graph neural network (GNN) predictions. Graph rationalization involves the joint identification of these subgraphs during GNN training, resulting in improved interpretability and generalization. GNN is widely used for node-level tasks such as paper classification and graph-level tasks such as molecular property prediction. However, on both levels, little attention has been given to GNN rationalization and the lack of training examples makes it difficult to identify the optimal graph rationales. In this work, we address the problem by proposing a unified data augmentation framework with two novel operations on environment subgraphs to rationalize GNN prediction. We define the environment subgraph as the remaining subgraph after rationale identification and separation. The framework efficiently performs rationale-environment separation in the representation space for a node's neighborhood graph or a graph's complete structure to avoid the high complexity of explicit graph decoding and encoding. We conduct experiments on 17 datasets spanning node classification, graph classification, and graph regression. Results demonstrate that our framework is effective and efficient in rationalizing and enhancing GNNs for different levels of tasks on graphs.
Keyword:
Graph neural network
node classification
graph property prediction
data augmentation
rationalization

期刊

ACM Transactions on Knowledge Discovery from Data 封面图
ACM Transactions on Knowledge Discovery from Data
IF:
4.8
论文数:
1.3K
被引数:
4.4K

机构

U
University of Notre Dame
学者数:
1.2W
论文数: 1.1W
被引数: 1.7W
引用论文

引用论文

Association of resuscitative endovascular balloon occlusion of the aorta (REBOA) and mortality in penetrating trauma patients
err2020-04-16
err0
PREAI
errAlberto F. García; Ramiro Manzano-Nunez; Claudia P. Orlas; Juan Ruiz-Yucuma; Alejandra Londoño; Camilo Salazar; Juan Melendez; Álvaro I. Sánchez; Juan Carlos Puyana; Carlos A. Ordoñez
err分享
err收藏
Biochemical Purification of a Mammalian Slit Protein as a Positive Regulator of Sensory Axon Elongation and Branching
errCell
IF0
err1999-03-01
err0
errOAAI
errKuan Hong Wang; Katja Brose; David Arnott; Thomas Kidd; Corey S Goodman; William Henzel; Marc Tessier-Lavigne
err分享
err收藏
MoleculeNet: a benchmark for molecular machine learning
err2018-01-01
err1.6K
errOAAI
errWu, Zhenqin; Ramsundar, Bharath; Feinberg, Evan N.; Gomes, Joseph; Geniesse, Caleb; Pappu, Aneesh S.; Leswing, Karl; Pande, Vijay
err分享
err收藏
Natural vibrations and waves in pretwisted rods
err1994-01-01
err0
PREAI
errO. Onipede; S.B. Dong; J.B. Kosmatka
err分享
err收藏
Humidity-Independent Methane Gas Detection in Gd0.2La0.2Ce0.2Hf0.2Zr0.2O₂-Based Sensor Using Polynomial Regression Analysis
err2022-12-01
err0
PREAI
errVenkata Ramesh Naganaboina; Satish Bonam; Mariappan Anandkumar; Atul Suresh Deshpande; Shiv Govind Singh
err分享
err收藏
学者 查看更多内容