Unlink to Unlearn

Dissocier pour désapprendre : simplifier le désapprentissage Edge dans les GNN

Référentiel de code de « Dissocier pour désapprendre : simplifier le désapprentissage Edge dans les GNN ».

Environnement

 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

Usage

Plus d'arguments de formation peuvent être consultés sur framework/training_args.py .

Entraînement

 python train_gnn . py - - dataset Cora

Désapprendre

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

Merci

Une partie du code a été dérivée du référentiel de code de GNNDelete.

Citation

 @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},
}