Unlink to Unlearn

Unlink to Unlearn: GNN에서 Edge Unlearning 단순화

"Unlearn to Unlink: GNN에서 Edge Unlearning 단순화"의 코드 저장소.

환경

 torch == 2.1.0
torch_geometric == 2.4.0
numpy == 1.26.2
sklearn == 1.3.2
networkx == 3.0
ogb ==1.3.6
wandb
tqdm

용법

더 많은 훈련 인수는 framework/training_args.py 에서 볼 수 있습니다.

훈련

 python train_gnn . py - - dataset Cora

언러닝

 python delete_gnn . py - - dataset Cora - - df_size 5.0 - - unlearning_model utu 

감사해요

일부 코드는 GNNDelete의 코드 저장소에서 분기되었습니다.

소환

 @inproceedings{tan2024unlink,
author = {Tan, Jiajun and Sun, Fei and Qiu, Ruichen and Su, Du and Shen, Huawei},
title = {Unlink to Unlearn: Simplifying Edge Unlearning in GNNs},
year = {2024},
doi = {10.1145/3543873.3587375},
booktitle = {Companion Proceedings of the ACM on Web Conference 2024},
pages = {489–492},
series = {WWW '24},
}