This is the official implementation of the paper Stochastic Gradient Langevin Unlearning.
The code is runnable under the following enveironment:
matplotlib 3.7.2
notebook 7.0.7
numpy 1.24.4
pandas 2.0.3
scikit-learn 1.3.0
scipy 1.10.1
seaborn 0.13.0
torch 2.0.0+cu117
torchvision 0.15.1+cu117
tqdm 4.65.0
python main_sgd.py --lam 1e-6 --dataset [MNIST/CIFAR10] --projection 0 --compare_baseline_nonconvergent 1
python main_sgd.py --lam 1e-6 --dataset [MNIST/CIFAR10] --projection 0 --sequential 1
python main_sgd.py --lam 1e-6 --dataset [MNIST/CIFAR10] --projection 0 --paint_unlearning_sigma 1
- Notes: use --gpu to allocate to a GPU device