-
Download the ISIC dataset from ISIC2018 Skin Lesion and Kaggle Intracranial Hemorrhage dataset from ICH
-
Run preprocess_ich.py, and then run the scprits (scripts/isic2018_prepare_fl.py and scripts/chestxray14_prepare_fl.py) to generate FL datasets, the corresponding configure file is in configs/ich/scripts_conf.yaml and configs/isic/scripts_conf.yaml
-
Run the training code. python3 train.py --config "../configs/ich/run_conf.yaml" --excluded_clients 0 --is_unlearn 1 python3 train.py --config "../configs/isic/run_conf.yaml" --excluded_clients 0 --is_unlearn 1
Please cite our paper if you find this code useful for your research.
@article{deng2024enable,
title={Enable the Right to be Forgotten with Federated Client Unlearning in Medical Imaging},
author={Deng, Zhipeng and Luo, Luyang and Chen, Hao},
journal={arXiv preprint arXiv:2407.02356},
year={2024}
}