├── .gitignore ├── LICENSE ├── README.md ├── arg_parser.py ├── curves.py ├── data.py ├── eval_curve.py ├── evaluation └── SVC_MIA.py ├── generate_mask.py ├── generate_weight_mask.py ├── main_tv.py ├── models ├── __init__.py ├── preresnet.py ├── vgg.py └── vit.py ├── plot_region.py ├── requirements.txt ├── taskvector ├── linearize.py ├── modeling.py ├── task_vectors.py ├── task_vectors_tangent.py └── utils.py ├── train.py ├── unlearn.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- 1 | .idea/ 2 | 3 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TIML-Group/Mode-Connectivity-Unlearning/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TIML-Group/Mode-Connectivity-Unlearning/HEAD/README.md -------------------------------------------------------------------------------- /arg_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TIML-Group/Mode-Connectivity-Unlearning/HEAD/arg_parser.py -------------------------------------------------------------------------------- /curves.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TIML-Group/Mode-Connectivity-Unlearning/HEAD/curves.py -------------------------------------------------------------------------------- /data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TIML-Group/Mode-Connectivity-Unlearning/HEAD/data.py -------------------------------------------------------------------------------- /eval_curve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TIML-Group/Mode-Connectivity-Unlearning/HEAD/eval_curve.py -------------------------------------------------------------------------------- /evaluation/SVC_MIA.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TIML-Group/Mode-Connectivity-Unlearning/HEAD/evaluation/SVC_MIA.py -------------------------------------------------------------------------------- /generate_mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TIML-Group/Mode-Connectivity-Unlearning/HEAD/generate_mask.py -------------------------------------------------------------------------------- /generate_weight_mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TIML-Group/Mode-Connectivity-Unlearning/HEAD/generate_weight_mask.py -------------------------------------------------------------------------------- /main_tv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TIML-Group/Mode-Connectivity-Unlearning/HEAD/main_tv.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TIML-Group/Mode-Connectivity-Unlearning/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/preresnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TIML-Group/Mode-Connectivity-Unlearning/HEAD/models/preresnet.py -------------------------------------------------------------------------------- /models/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TIML-Group/Mode-Connectivity-Unlearning/HEAD/models/vgg.py -------------------------------------------------------------------------------- /models/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TIML-Group/Mode-Connectivity-Unlearning/HEAD/models/vit.py -------------------------------------------------------------------------------- /plot_region.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TIML-Group/Mode-Connectivity-Unlearning/HEAD/plot_region.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TIML-Group/Mode-Connectivity-Unlearning/HEAD/requirements.txt -------------------------------------------------------------------------------- /taskvector/linearize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TIML-Group/Mode-Connectivity-Unlearning/HEAD/taskvector/linearize.py -------------------------------------------------------------------------------- /taskvector/modeling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TIML-Group/Mode-Connectivity-Unlearning/HEAD/taskvector/modeling.py -------------------------------------------------------------------------------- /taskvector/task_vectors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TIML-Group/Mode-Connectivity-Unlearning/HEAD/taskvector/task_vectors.py -------------------------------------------------------------------------------- /taskvector/task_vectors_tangent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TIML-Group/Mode-Connectivity-Unlearning/HEAD/taskvector/task_vectors_tangent.py -------------------------------------------------------------------------------- /taskvector/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TIML-Group/Mode-Connectivity-Unlearning/HEAD/taskvector/utils.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TIML-Group/Mode-Connectivity-Unlearning/HEAD/train.py -------------------------------------------------------------------------------- /unlearn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TIML-Group/Mode-Connectivity-Unlearning/HEAD/unlearn.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/TIML-Group/Mode-Connectivity-Unlearning/HEAD/utils.py --------------------------------------------------------------------------------