Streaming Graph Neural Networks via Continual Learning

Code for Streaming Graph Neural Networks via Continual Learning（CIKM 2020). ContinualGNN is a streaming graph neural network based on continual learning so that the model is trained incrementally and up-to-date node representations can be obtained at each time step.

Requirements

• python = 3.8.5
• pytorch = 1.7.1
• scikit-learn = 0.23.2

Usages

• ContinualGNN (proposed model) on Cora:

cd src/
python main_stream.py --data=cora --new_ratio=0.8 --memory_size=250 --ewc_lambda=80.0 

• OnlineGNN (lower bound) on Cora:

python main_stream.py --data=cora

If using cuda, set --cuda.