├── .gitignore ├── LICENSE ├── README.md ├── baselines ├── config │ ├── ds_z0_config.json │ ├── ds_z2_config.json │ ├── finetune_lora.yaml │ ├── forget_lora.yaml │ └── model_config.yaml ├── pretrain.py ├── pretrain_scripts │ └── kud-pt.sh ├── src │ ├── __init__.py │ ├── config.py │ ├── dataset.py │ ├── finetune.py │ ├── forget.py │ ├── iterative_trainer.py │ ├── memflex_trainer.py │ ├── sure_trainer.py │ └── utils.py ├── unlearn.py └── unlearn_scripts │ ├── kud-baselines.sh │ ├── kud-relearn.sh │ ├── tofu-baselines.sh │ └── tofu-relearn.sh ├── dataAugument ├── __init__.py ├── augu.sh ├── gather_proc_data.py ├── proc.py ├── templates.json └── utils.py ├── dataset ├── KnowUnDo │ ├── .gitkeep │ └── privacy │ │ ├── full.json │ │ ├── retention_train.json │ │ ├── retention_val.json │ │ ├── unlearn_train.json │ │ └── unlearn_val.json ├── TOFU │ └── .gitkeep ├── augument_data │ └── .gitkeep └── idontknow.txt ├── evals ├── eval-dpsk-forget-retain │ ├── README.md │ ├── agg.sh │ ├── compute_forget_retain.py │ ├── config │ │ ├── datapre.yaml │ │ ├── privacy_forget_prompt.txt │ │ ├── privacy_retain_prompt.txt │ │ └── relev_fluen_prompt.txt │ ├── forget_retain_datapre.py │ ├── forget_retain_dpsk.py │ ├── prepare.sh │ ├── run.sh │ └── utils.py ├── eval-gpt4-relev_fluen │ ├── README.md │ ├── compute_relev_fluen.py │ ├── config │ │ ├── datapre.yaml │ │ └── relev_fluen_prompt.txt │ ├── gpt4-agg.sh │ ├── gpt4-prepare.sh │ ├── gpt4-run.sh │ ├── relev_fluen_datapre.py │ ├── relvev_fluen_gpt4o.py │ └── utils.py ├── eval_all.sh ├── evaluate.py ├── generate.py ├── inf_all.sh ├── merge_all.sh └── merge_model.py ├── images ├── intro.jpg ├── 📄_arXiv-2502.11190-blue.svg ├── 🤗_HuggingFace-Collection-green.svg └── 🤗_HuggingFace-Paper-yellow.svg ├── requirements.txt └── semeval25 ├── README.md ├── requirements.txt └── unlearn-merging.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/README.md -------------------------------------------------------------------------------- /baselines/config/ds_z0_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/baselines/config/ds_z0_config.json -------------------------------------------------------------------------------- /baselines/config/ds_z2_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/baselines/config/ds_z2_config.json -------------------------------------------------------------------------------- /baselines/config/finetune_lora.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/baselines/config/finetune_lora.yaml -------------------------------------------------------------------------------- /baselines/config/forget_lora.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/baselines/config/forget_lora.yaml -------------------------------------------------------------------------------- /baselines/config/model_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/baselines/config/model_config.yaml -------------------------------------------------------------------------------- /baselines/pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/baselines/pretrain.py -------------------------------------------------------------------------------- /baselines/pretrain_scripts/kud-pt.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/baselines/pretrain_scripts/kud-pt.sh -------------------------------------------------------------------------------- /baselines/src/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/baselines/src/__init__.py -------------------------------------------------------------------------------- /baselines/src/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/baselines/src/config.py -------------------------------------------------------------------------------- /baselines/src/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/baselines/src/dataset.py -------------------------------------------------------------------------------- /baselines/src/finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/baselines/src/finetune.py -------------------------------------------------------------------------------- /baselines/src/forget.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/baselines/src/forget.py -------------------------------------------------------------------------------- /baselines/src/iterative_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/baselines/src/iterative_trainer.py -------------------------------------------------------------------------------- /baselines/src/memflex_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/baselines/src/memflex_trainer.py -------------------------------------------------------------------------------- /baselines/src/sure_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/baselines/src/sure_trainer.py -------------------------------------------------------------------------------- /baselines/src/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/baselines/src/utils.py -------------------------------------------------------------------------------- /baselines/unlearn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/baselines/unlearn.py -------------------------------------------------------------------------------- /baselines/unlearn_scripts/kud-baselines.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/baselines/unlearn_scripts/kud-baselines.sh -------------------------------------------------------------------------------- /baselines/unlearn_scripts/kud-relearn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/baselines/unlearn_scripts/kud-relearn.sh -------------------------------------------------------------------------------- /baselines/unlearn_scripts/tofu-baselines.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/baselines/unlearn_scripts/tofu-baselines.sh -------------------------------------------------------------------------------- /baselines/unlearn_scripts/tofu-relearn.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/baselines/unlearn_scripts/tofu-relearn.sh -------------------------------------------------------------------------------- /dataAugument/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dataAugument/augu.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/dataAugument/augu.sh -------------------------------------------------------------------------------- /dataAugument/gather_proc_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/dataAugument/gather_proc_data.py -------------------------------------------------------------------------------- /dataAugument/proc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/dataAugument/proc.py -------------------------------------------------------------------------------- /dataAugument/templates.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/dataAugument/templates.json -------------------------------------------------------------------------------- /dataAugument/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/dataAugument/utils.py -------------------------------------------------------------------------------- /dataset/KnowUnDo/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dataset/KnowUnDo/privacy/full.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/dataset/KnowUnDo/privacy/full.json -------------------------------------------------------------------------------- /dataset/KnowUnDo/privacy/retention_train.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/dataset/KnowUnDo/privacy/retention_train.json -------------------------------------------------------------------------------- /dataset/KnowUnDo/privacy/retention_val.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/dataset/KnowUnDo/privacy/retention_val.json -------------------------------------------------------------------------------- /dataset/KnowUnDo/privacy/unlearn_train.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/dataset/KnowUnDo/privacy/unlearn_train.json -------------------------------------------------------------------------------- /dataset/KnowUnDo/privacy/unlearn_val.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/dataset/KnowUnDo/privacy/unlearn_val.json -------------------------------------------------------------------------------- /dataset/TOFU/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dataset/augument_data/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dataset/idontknow.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/dataset/idontknow.txt -------------------------------------------------------------------------------- /evals/eval-dpsk-forget-retain/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/evals/eval-dpsk-forget-retain/README.md -------------------------------------------------------------------------------- /evals/eval-dpsk-forget-retain/agg.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/evals/eval-dpsk-forget-retain/agg.sh -------------------------------------------------------------------------------- /evals/eval-dpsk-forget-retain/compute_forget_retain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/evals/eval-dpsk-forget-retain/compute_forget_retain.py -------------------------------------------------------------------------------- /evals/eval-dpsk-forget-retain/config/datapre.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/evals/eval-dpsk-forget-retain/config/datapre.yaml -------------------------------------------------------------------------------- /evals/eval-dpsk-forget-retain/config/privacy_forget_prompt.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/evals/eval-dpsk-forget-retain/config/privacy_forget_prompt.txt -------------------------------------------------------------------------------- /evals/eval-dpsk-forget-retain/config/privacy_retain_prompt.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/evals/eval-dpsk-forget-retain/config/privacy_retain_prompt.txt -------------------------------------------------------------------------------- /evals/eval-dpsk-forget-retain/config/relev_fluen_prompt.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/evals/eval-dpsk-forget-retain/config/relev_fluen_prompt.txt -------------------------------------------------------------------------------- /evals/eval-dpsk-forget-retain/forget_retain_datapre.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/evals/eval-dpsk-forget-retain/forget_retain_datapre.py -------------------------------------------------------------------------------- /evals/eval-dpsk-forget-retain/forget_retain_dpsk.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/evals/eval-dpsk-forget-retain/forget_retain_dpsk.py -------------------------------------------------------------------------------- /evals/eval-dpsk-forget-retain/prepare.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/evals/eval-dpsk-forget-retain/prepare.sh -------------------------------------------------------------------------------- /evals/eval-dpsk-forget-retain/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/evals/eval-dpsk-forget-retain/run.sh -------------------------------------------------------------------------------- /evals/eval-dpsk-forget-retain/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/evals/eval-dpsk-forget-retain/utils.py -------------------------------------------------------------------------------- /evals/eval-gpt4-relev_fluen/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/evals/eval-gpt4-relev_fluen/README.md -------------------------------------------------------------------------------- /evals/eval-gpt4-relev_fluen/compute_relev_fluen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/evals/eval-gpt4-relev_fluen/compute_relev_fluen.py -------------------------------------------------------------------------------- /evals/eval-gpt4-relev_fluen/config/datapre.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/evals/eval-gpt4-relev_fluen/config/datapre.yaml -------------------------------------------------------------------------------- /evals/eval-gpt4-relev_fluen/config/relev_fluen_prompt.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/evals/eval-gpt4-relev_fluen/config/relev_fluen_prompt.txt -------------------------------------------------------------------------------- /evals/eval-gpt4-relev_fluen/gpt4-agg.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/evals/eval-gpt4-relev_fluen/gpt4-agg.sh -------------------------------------------------------------------------------- /evals/eval-gpt4-relev_fluen/gpt4-prepare.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/evals/eval-gpt4-relev_fluen/gpt4-prepare.sh -------------------------------------------------------------------------------- /evals/eval-gpt4-relev_fluen/gpt4-run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/evals/eval-gpt4-relev_fluen/gpt4-run.sh -------------------------------------------------------------------------------- /evals/eval-gpt4-relev_fluen/relev_fluen_datapre.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/evals/eval-gpt4-relev_fluen/relev_fluen_datapre.py -------------------------------------------------------------------------------- /evals/eval-gpt4-relev_fluen/relvev_fluen_gpt4o.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/evals/eval-gpt4-relev_fluen/relvev_fluen_gpt4o.py -------------------------------------------------------------------------------- /evals/eval-gpt4-relev_fluen/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/evals/eval-gpt4-relev_fluen/utils.py -------------------------------------------------------------------------------- /evals/eval_all.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/evals/eval_all.sh -------------------------------------------------------------------------------- /evals/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/evals/evaluate.py -------------------------------------------------------------------------------- /evals/generate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/evals/generate.py -------------------------------------------------------------------------------- /evals/inf_all.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/evals/inf_all.sh -------------------------------------------------------------------------------- /evals/merge_all.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/evals/merge_all.sh -------------------------------------------------------------------------------- /evals/merge_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/evals/merge_model.py -------------------------------------------------------------------------------- /images/intro.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/images/intro.jpg -------------------------------------------------------------------------------- /images/📄_arXiv-2502.11190-blue.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/images/📄_arXiv-2502.11190-blue.svg -------------------------------------------------------------------------------- /images/🤗_HuggingFace-Collection-green.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/images/🤗_HuggingFace-Collection-green.svg -------------------------------------------------------------------------------- /images/🤗_HuggingFace-Paper-yellow.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/images/🤗_HuggingFace-Paper-yellow.svg -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/requirements.txt -------------------------------------------------------------------------------- /semeval25/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/semeval25/README.md -------------------------------------------------------------------------------- /semeval25/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/semeval25/requirements.txt -------------------------------------------------------------------------------- /semeval25/unlearn-merging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjunlp/unlearn/HEAD/semeval25/unlearn-merging.py --------------------------------------------------------------------------------