arrow
Return

DisenSemi: Semi-Supervised Graph Classification via Disentangled Representation Learning

delete2024-01-01
delete0
delete
OA
AI
Y
Yifan Wang
X
Xiao Luo
C
Chong Chen
X
Xian‐Sheng Hua
M
Ming Zhang
W
Wei Ju *
DOI:10.1109/TNNLS.2024.3431871delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Graph classification is a critical task in numerous multimedia applications, where graphs are employed to represent diverse types of multimedia data, including images, videos, and social networks. Nevertheless, in the real world, labeled graph data are always limited or scarce. To address this issue, we focus on the semi-supervised graph classification task, which involves both supervised and unsupervised models learning from labeled and unlabeled data. In contrast to recent approaches that transfer the entire knowledge from the unsupervised model to the supervised one, we argue that an effective transfer should only retain the relevant semantics that align well with the supervised task. We introduce a novel framework termed in this article, which learns disentangled representation for semi-supervised graph classification. Specifically, a disentangled graph encoder is proposed to generate factorwise graph representations for both supervised and unsupervised models. Then, we train two models via supervised objective and mutual information (MI)-based constraints, respectively. To ensure the meaningful transfer of knowledge from the unsupervised encoder to the supervised one, we further define an MI-based disentangled consistency regularization between two models and identify the corresponding rationale that aligns well with the current graph classification task. Experiments conducted on various publicly available datasets demonstrate the effectiveness of our DisenSemi.
Keywords:
Disentangled representation learning
graph neural networks (GNNs)
semi-supervised graph classification
graph neural networks (GNNs)
semi-supervised graph classification

Journal

IEEE Transactions on Neural Networks and Learning Systems cover
IEEE Transactions on Neural Networks and Learning Systems
IF:
8.9
Papers:
7.5K
Citations:
7.2W

Organization

U
university of international business & economics
Scholars:
1.6K
Papers: 2.1K
Citations: 5
S
sichuan university
Scholars:
12.0W
Papers: 7.8W
Citations: 100
P
peking university
Scholars:
11.8W
Papers: 8.7W
Citations: 146
researcher View more organizations