jiamingzhang94/unlearnable-clustersCVPR2023: عناقيد غير قابلة للتعلم: نحو أمثلة غير قابلة للتعلم مستقلّة عن التسميات
الكود المتعلق بـ "العناقيد غير القابلة للتعلم: نحو أمثلة غير قابلة للتعلم غير معتمدة على التسميات"، CVPR2023.
أولاً، شغّل preprocess.py لإعادة تسمية مجلد مجموعة البيانات
python preprocess.py --config config/preprocess.yaml -f rename
ثانياً، شغّل preprocess.py لإجراء التهيئة الأولية لـ K-means.
python preprocess.py --config config/preprocess.yaml -f cluster_pets_rn50_cluster10
# the cluster file will be stored in {output_dir}/{dataset}_{surrogate_model}_cluster{num_clusters}.pth
# e.g., output/cluster/pets_rn50_cluster10.pth
ثالثاً، شغّل main.py --stage 1 لتوليد الاضطرابات على مستوى العناقيد.
python main.py --config config/stage_1.yaml -e {experiment} --stage 1
# {experiment} is uc_{dataset}_{surrogate_model}
# e.g., python main.py --config config/stage_1.yaml -e uc_pets_cliprn50 --stage 1
أخيراً، شغّل main.py --stage 2 لتدريب النماذج المستهدفة.
python main.py --config config/stage_2.yaml -e {experiment} --stage 2
# {experiment} is uc_{dataset}_{surrogate_model}_{target_model}
# e.g., python main.py --config config/stage_2.yaml -e uc_pets_rn50_rn18 --stage 2
إذا وجدت أن هذا الكود مفيد لبحثك، فيُرجى التفكير في الاستشهاد به.
@inproceedings{zhang2023unlearnable,
title={Unlearnable Clusters: Towards Label-agnostic Unlearnable Examples},
author={Jiaming Zhang, Xingjun Ma, Qi Yi, Jitao Sang, Yu-Gang Jiang, Yaowei Wang, Changsheng Xu},
booktitle="Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition",
year={2023}
}