RUIYUN-ML/ERM-KTP

ERM-KTP

Code for our CVPR 2023 paper "ERM-KTP: Knowledge-level Machine Unlearning via Knowledge Transfer".

Requirements

• Platforms: Ubuntu: 18.04 Cuda: 11.7
• Python: 3.8 Pytorch: 1.10.1

Initialize

Edit the file parser_init.py

Train ERM-CNNs

python train_ERM.py

KTP

python KTP.py

Test

python classAcc_valid.py

If you want to unlearn the specific class in your work, please modidy the following code in line 262 of dataLoder.py.

if data_train[i][1] >= 0 and data_train[i][1] < self.num_unlearn:

Citation

Cite as below if you find this repository helpful:

@InProceedings{Lin_2023_CVPR,
    author    = {Lin, Shen and Zhang, Xiaoyu and Chen, Chenyang and Chen, Xiaofeng and Susilo, Willy},
    title     = {ERM-KTP: Knowledge-Level Machine Unlearning via Knowledge Transfer},
    booktitle = {Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},
    month     = {June},
    year      = {2023},
    pages     = {20147-20155}
}