Skip to content

yxoh/link_steal_pets2024

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

This is the official implementation for the paper: Link Stealing Attacks Against Inductive Graph Neural Networks.

Setup

conda env create -f link_steal.yaml
conda activate link_steal

git clone https://github.com/EdisonLeeeee/GraphGallery.git && cd GraphGallery
pip install -e . --verbose

Train target GNNs


python train_gnn.py --dataset lastfm --model graphsage --mode target --gpu 0
python train_gnn.py --dataset lastfm --model graphsage --mode shadow --gpu 0

Attack

  1. Attack-0
python mlp_attack.py --dataset lastfm --node_topology 0-hop --edge_feature all --target_model graphsage --shadow_model graphsage --lr 0.006 --optim adam --scheduler --gpu 0
  1. Attack-1
python mlp_attack.py --dataset lastfm --node_topology 1-hop --edge_feature all --target_model graphsage --shadow_model graphsage --lr 0.006 --optim adam --scheduler --gpu 0
  1. Attack-2
python mlp_attack.py --dataset lastfm --node_topology 2-hop --edge_feature all --target_model graphsage --shadow_model graphsage --lr 0.006 --optim adam --scheduler --gpu 0
  1. Attack-3
python mlp_attack.py --dataset lastfm --node_topology 0-hop --edge_feature all --target_model graphsage --shadow_model graphsage --lr 0.006 --optim adam --scheduler --gpu 0 --plus
  1. Attack-4
python mlp_attack.py --dataset lastfm --node_topology 1-hop --edge_feature all --target_model graphsage --shadow_model graphsage --lr 0.006 --optim adam --scheduler --gpu 0 --plus
  1. Attack-5
python mlp_attack.py --dataset lastfm --node_topology 2-hop --edge_feature all --target_model graphsage --shadow_model graphsage --lr 0.006 --optim adam --scheduler --gpu 0 --plus
  1. Attack-6
python mlp_attack.py --dataset lastfm --node_topology 1-hop --edge_feature all --target_model graphsage --shadow_model graphsage --lr 0.006 --optim adam --scheduler --gpu 0 --plus2
  1. Attack-7
python mlp_attack.py --dataset lastfm --node_topology 2-hop --edge_feature all --target_model graphsage --shadow_model graphsage --lr 0.006 --optim adam --scheduler --gpu 0 --plus2
  1. Attack-8
python mlp_attack.py --dataset lastfm --node_topology 1-hop --edge_feature all --target_model graphsage --shadow_model graphsage --lr 0.006 --optim adam --scheduler --gpu 0 --all
  1. Attack-9
python mlp_attack.py --dataset lastfm --node_topology 2-hop --edge_feature all --target_model graphsage --shadow_model graphsage --lr 0.006 --optim adam --scheduler --gpu 0 --all

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages