CaT: Balanced Continual Graph Learning with Graph Condensation

Yilun Liu, Ruihong Qiu, Zi Huang

Main figure for CaT: Balanced Continual Graph Learning with Graph Condensation

Abstract

Continual graph learning (CGL) is purposed to continuously update a graph model with graph data being fed in a streaming manner. Since the model easily forgets previously learned knowledge when training with new-coming data, the catastrophic forgetting problem has been the major focus in CGL. Recent replay-based methods intend to solve this problem by updating the model using both (1) the entire new-coming data and (2) a sampling-based memory bank that stores replayed graphs to approximate the distribution of historical data. After updating the model, a new replayed graph sampled from the incoming graph will be added to the existing memory bank. Despite these methods are intuitive and effective for the CGL, two issues are identified in this paper. Firstly, most sampling-based methods struggle to fully capture the historical distribution when the storage budget is tight. Secondly, a significant data imbalance exists in terms of the scales of the complex new-coming graph data and the lightweight memory bank, resulting in unbalanced training. To solve these issues, a Condense and Train (CaT) framework is proposed in this paper. Prior to each model update, the new-coming graph is condensed to a small yet informative synthesised replayed graph, which is then stored in a Condensed Graph Memory with historical replay graphs. In the continual learning phase, a Training in Memory scheme is used to update the model directly with the Condensed Graph Memory rather than the whole new-coming graph, which alleviates the data imbalance problem. Extensive experiments conducted on four benchmark datasets successfully demonstrate superior performances of the proposed CaT framework in terms of effectiveness and efficiency. The code has been released on.

Proposed method: CaT

Benchmark Results

Tested on 1 dataset Non-standard metric on 1 dataset

Tables show presented results in global context (best 3 papers + GCN + MLP).

OGB

ogbn-products

Global leaderboard →
Non-default metric. This paper reports AP; the dataset default is Accuracy. Values below use AP only.

Sorted by AP. Highlighted rows are from other papers.

#ModelAP ↑Paper
1
MAS
0.9590 ±0.0010
This paper
2
Joint
0.9570 ±0.0070
This paper
3
CGM
0.9560 ±0.0030
This paper
4
SSM
0.9460 ±0.0050
This paper
5
EWC
0.9030 ±0.0180
This paper
6
GEM
0.8690 ±0.0090
This paper
7
ER-GNN
0.8350 ±0.0040
This paper
8
HPNs*
0.8010 ±0.0080
This paper
9
TWP
0.7550 ±0.0440
This paper
10
CaT
0.7100 ±0.0020
This paper
11
Finetuning
0.5640 ±0.0380
This paper
12
LwF
0.5010 ±0.0070
This paper
SOTA performance (not significantly different from the best) Proposed method (this paper) Baseline / comparison (this paper)