arrow
Return

Iterative Graph Self-Distillation

delete2024-03-01
delete2
delete
OA
AI
H
Hanlin Zhang *
S
Shuai Lin
W
Weiyang Liu
P
Pan Zhou
汤
汤京永 (Jian Tang)
梁小丹 cover
梁小丹 (Xiaodan Liang)
E
Eric P. Xing
DOI:10.1109/TKDE.2023.3303885delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Recently, there has been increasing interest in the challenge of how to discriminatively vectorize graphs. To address this, we propose a method called Iterative Graph Self-Distillation (IGSD) which learns graph-level representation in an unsupervised manner through instance discrimination using a self-supervised contrastive learning approach. IGSD involves a teacher-student distillation process that uses graph diffusion augmentations and constructs the teacher model using an exponential moving average of the student model. The intuition behind IGSD is to predict the teacher network representation of the graph pairs under different augmented views. As a natural extension, we also apply IGSD to semi-supervised scenarios by jointly regularizing the network with both supervised and self-supervised contrastive loss. Finally, we show that fine-tuning the IGSD-trained models with self-training can further improve graph representation learning. Empirically, we achieve significant and consistent performance gain on various graph datasets in both unsupervised and semi-supervised settings, which well validates the superiority of IGSD.
Keywords:
Task analysis
Representation learning
Kernel
Graph neural networks
Iterative methods
Data augmentation
Training
graph representation learning
self-supervised learning

Journal

IEEE Transactions on Knowledge and Data Engineering cover
IEEE Transactions on Knowledge and Data Engineering
IF:
10.4
Papers:
6.9K
Citations:
3.2W

Organization

H
HEC Montreal
Scholars:
860
Papers: 944
Citations: 6
C
Carnegie Mellon University
Scholars:
1.4W
Papers: 1.4W
Citations: 2.7W
U
universite de montreal
Scholars:
4.6W
Papers: 3.8W
Citations: 46
S
Sun Yat Sen University
Scholars:
9.9W
Papers: 7.2W
Citations: 95
U
University of Cambridge
Scholars:
7.7W
Papers: 7.1W
Citations: 13.7W
researcher View more organizations
Cited Papers

Cited Papers

Artificial intelligence in COVID-19 drug repurposing
err2020-12-01
err280
errOAAI
errZhou, Yadi; Wang, Fei; Tang, Jian; Nussinov, Ruth; Cheng, Feixiong
errShare
errSave
Ultraviolet and optical observations of OB associations in M31
err1995-06-01
err0
PREAI
errJ. K. Hill; J. E. Isensee; R. C. Bohlin; K. P. Cheng; P. M. N. Hintzen; R. W. O'Connell; M. S. Roberts; A. M. Smith; Eric P. Smith; T. P. Stecher
errShare
errSave
errShare
errSave
The cranial endoskeleton of Tiktaalik roseae
err2008-10-01
err0
PREAI
errJason P. Downs; Edward B. Daeschler; Farish A. Jenkins; Neil H. Shubin
errShare
errSave
Inhibition of NAMPT pathway by FK866 activates the function of p53 in HEK293T cells
err2012-08-01
err0
PREAI
errBasant Kumar Thakur; Tino Dittrich; Prakash Chandra; Annette Becker; Yannick Lippka; Divakarvel Selvakumar; Jan-Henning Klusmann; Dirk Reinhardt; Karl Welte
errShare
errSave
researcher View more