This is official repository for the paper Deep Unlearning: Fast and Efficient Gradient-free Approach to Class Forgetting accepted at TMLR.
conda env create -f env.yml
conda activate forget
python3 ./demo.py
# for CIFAR10
sh ./scripts/our_cifar10.sh
# for CIFAR100
sh ./scripts/our_cifar100.sh
# for ImageNet
sh ./scripts/our_imagenet.sh
Scripts for analysis done in the paper can be found in scripts/analysis
.
An older version of the repository can be found in the legacy branch of the repository.
Kindly cite the paper if you use the code. Thanks!
Kodge, Sangamesh, Gobinda Saha, and Kaushik Roy. "Deep Unlearning: Fast and Efficient Gradient-Free Class Forgetting."
@article{
kodge2024deep,
title={Deep Unlearning: Fast and Efficient Gradient-free Class Forgetting},
author={Sangamesh Kodge and Gobinda Saha and Kaushik Roy},
journal={Transactions on Machine Learning Research},
issn={2835-8856},
year={2024},
url={https://openreview.net/forum?id=BmI5p6wBi0},
note={}
}