arrow
Return

CCGL: Contrastive Cascade Graph Learning

delete2022-01-01
delete20
delete
OA
AI
X
Xovee Xu
F
Fan Zhou
K
Kunpeng Zhang
S
Siyuan Liu *
DOI:10.1109/TKDE.2022.3151829delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Supervised learning, while prevalent for information cascade modeling, often requires abundant labeled data in training, and the trained model is not easy to generalize across tasks and datasets. It often learns task-specific representations, which can easily result in overfitting for downstream tasks. Recently, self-supervised learning is designed to alleviate these two fundamental issues in linguistic and visual tasks. However, its direct applicability for information cascade modeling, especially graph cascade related tasks, remains underexplored. In this work, we present Contrastive Cascade Graph Learning (CCGL), a novel framework for information cascade graph learning in a contrastive, self-supervised, and task-agnostic way. In particular, CCGL first designs an effective data augmentation strategy to capture variation and uncertainty by simulating the information diffusion in graphs. Second, it learns a generic model for graph cascade tasks via self-supervised contrastive pre-training using both unlabeled and labeled data. Third, CCGL learns a task-specific cascade model via fine-tuning using labeled data. Finally, to make the model transferable across datasets and cascade applications, CCGL further enhances the model via distillation using a teacher-student architecture. We demonstrate that CCGL significantly outperforms its supervised and semi-supervised counterparts for several downstream tasks.
Keywords:
Training
Visualization
Uncertainty
Supervised learning
Self-supervised learning
Predictive models
Linguistics
Information cascade
contrastive self-supervised learning
graph learning
information diffusion
data augmentation

Journal

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

Organization

University System of Maryland cover
University System of Maryland
Scholars:
6.5W
Papers: 5.6W
Citations: 113
Cited Papers

Cited Papers

Short stature and celiac disease: A relationship to consider even in patients with no gastrointestinal tract symptoms
err1983-11-01
err0
PREAI
errE. Cacciari; S. Salardi; R. Lazzari; A. Cicognani; A. Collina; P. Pirazzoli; P. Tassoni; G. Biasco; G.R. Corazza; A. Cassio
errShare
errSave
err1999-01-01
err0
PREAI
errT. Çaşkurlu; A. I. Tasci; M. Samasti; Z. Bayraktar; M. Çek; G. Sevin
errShare
errSave
Full-Scale Information Diffusion Prediction With Reinforced Recurrent Networks
err2023-05-01
err25
PREAI
errYang, Cheng; Wang, Hao; Tang, Jian; Shi, Chuan; Sun, Maosong; Cui, Ganqu; Liu, Zhiyuan
errShare
errSave
Interpretation of aeromagnetic data in the Franklin area, northern New Zealand
err2017-01-09
err0
PREAI
errD. J. Robertson; P. N. P. Vidanovich; S. K. Zoellner; J. B. Meyers
errShare
errSave
Locomotor deficits in the mutant mouse, Lurcher
err1987-04-01
err0
PREAI
errP.A. Fortier; A.M. Smith; S. Rossignol
errShare
errSave
researcher View more