Causal integration in graph neural networks toward enhanced classification: benchmarking and advancements for robust performance

  • Simi JOB*
  • , Xiaohui TAO
  • , Taotao CAI
  • , Lin LI
  • , Quan Z. SHENG
  • , Haoran XIE
  • , Jianming YONG
  • *Corresponding author for this work

Research output: Journal PublicationsJournal Article (refereed)peer-review

Abstract

The expansion of Graph Neural Networks (GNNs) has highlighted the importance of evaluating their performance in real-world scenarios. However, existing evaluation frameworks often overlook the integration of causality, a critical component that is essential for more robust evaluation of GNNs. To address this gap, we present a benchmark study that systematically compares standard and causal GNN models with a focus on classification tasks. Our analysis encompasses a careful selection of nine GNN models across seven diverse datasets that span three distinct domains. The results reveal the following: I) Causality-enhanced GNNs consistently outperform their traditional counterparts in graph classification tasks; II) Models integrating causal features exhibit greater generalizability across varied datasets; and III) Incorporation of causal elements significantly improves the predictive accuracy of GNNs. These findings highlight the importance of embedding causality in the evaluation and development of GNNs for improved performance and application.
Original languageEnglish
Article number30
JournalWorld Wide Web
Volume28
Issue number3
Early online date7 Apr 2025
DOIs
Publication statusPublished - 1 May 2025

Bibliographical note

Publisher Copyright:
© The Author(s) 2025.

Funding

This work is partially supported by grants from Australian Research Council (No. DP220101360) and the SAGE Athena Swan Scholarship, UniSQ. Open Access funding enabled and organized by CAUL and its Member Institutions.

UN SDGs

This output contributes to the following UN Sustainable Development Goals (SDGs)

  1. SDG 9 - Industry, Innovation, and Infrastructure
    SDG 9 Industry, Innovation, and Infrastructure
  2. SDG 17 - Partnerships for the Goals
    SDG 17 Partnerships for the Goals

Keywords

  • Causality
  • GAT
  • GCN
  • Graph classification
  • Graph neural networks
  • GraphSAGE

Fingerprint

Dive into the research topics of 'Causal integration in graph neural networks toward enhanced classification: benchmarking and advancements for robust performance'. Together they form a unique fingerprint.

Cite this