├── .gitignore ├── LICENSE ├── README.md ├── agnews ├── build_dataset.py └── pred_eval.py ├── alpaca └── pred_eval.py ├── datasets_visualizer.py ├── example_figure.png ├── gsm8k ├── build_dataset.py └── pred_eval.py ├── huggingface_token.txt ├── loss_func └── repnoise_loss.py ├── poison └── evaluation │ ├── __init__.py │ ├── constants.py │ ├── eval_sentiment.py │ ├── moderation.py │ ├── pred.py │ └── utils.py ├── reproduce.md ├── script ├── alignment │ ├── repNoise.sh │ ├── sft.sh │ ├── smooth_align.sh │ └── vaccine.sh ├── attack │ ├── batch_virus.sh │ ├── sft.sh │ ├── system_virus.sh │ ├── virus.sh │ ├── virus_agnews.sh │ └── virus_vaccine.sh └── finetune │ ├── bf_moderation_finetune_agnews.sh │ ├── bf_moderation_finetune_gsm8k.sh │ ├── bf_moderation_finetune_sst2.sh │ ├── hf_finetune_gsm8k.sh │ ├── hf_moderation_finetune_agnews.sh │ ├── hf_moderation_finetune_gsm8k.sh │ ├── hf_moderation_finetune_sst2.sh │ ├── mixing_finetune_gsm8k.sh │ ├── mixing_moderation_finetune_agnews.sh │ ├── mixing_moderation_finetune_gsm8k.sh │ ├── mixing_moderation_finetune_sst2.sh │ ├── sft_finetune.sh │ ├── statistical_evaluation_virus_finetune.sh │ ├── virus_finetune.sh │ ├── virus_moderation_finetune_agnews.sh │ ├── virus_moderation_finetune_gsm8k.sh │ └── virus_moderation_finetune_sst2.sh ├── sst2 ├── build_dataset.py └── pred_eval.py ├── three_stage.png ├── train.py ├── trainer.py ├── utils.py ├── virus.yml └── virus_pip.txt /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Virus/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Virus/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Virus/HEAD/README.md -------------------------------------------------------------------------------- /agnews/build_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Virus/HEAD/agnews/build_dataset.py -------------------------------------------------------------------------------- /agnews/pred_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Virus/HEAD/agnews/pred_eval.py -------------------------------------------------------------------------------- /alpaca/pred_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Virus/HEAD/alpaca/pred_eval.py -------------------------------------------------------------------------------- /datasets_visualizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Virus/HEAD/datasets_visualizer.py -------------------------------------------------------------------------------- /example_figure.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Virus/HEAD/example_figure.png -------------------------------------------------------------------------------- /gsm8k/build_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Virus/HEAD/gsm8k/build_dataset.py -------------------------------------------------------------------------------- /gsm8k/pred_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Virus/HEAD/gsm8k/pred_eval.py -------------------------------------------------------------------------------- /huggingface_token.txt: -------------------------------------------------------------------------------- 1 | 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 -------------------------------------------------------------------------------- /loss_func/repnoise_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Virus/HEAD/loss_func/repnoise_loss.py -------------------------------------------------------------------------------- /poison/evaluation/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /poison/evaluation/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Virus/HEAD/poison/evaluation/constants.py -------------------------------------------------------------------------------- /poison/evaluation/eval_sentiment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Virus/HEAD/poison/evaluation/eval_sentiment.py -------------------------------------------------------------------------------- /poison/evaluation/moderation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Virus/HEAD/poison/evaluation/moderation.py -------------------------------------------------------------------------------- /poison/evaluation/pred.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Virus/HEAD/poison/evaluation/pred.py -------------------------------------------------------------------------------- /poison/evaluation/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Virus/HEAD/poison/evaluation/utils.py -------------------------------------------------------------------------------- /reproduce.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Virus/HEAD/reproduce.md -------------------------------------------------------------------------------- /script/alignment/repNoise.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Virus/HEAD/script/alignment/repNoise.sh -------------------------------------------------------------------------------- /script/alignment/sft.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Virus/HEAD/script/alignment/sft.sh -------------------------------------------------------------------------------- /script/alignment/smooth_align.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Virus/HEAD/script/alignment/smooth_align.sh -------------------------------------------------------------------------------- /script/alignment/vaccine.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Virus/HEAD/script/alignment/vaccine.sh -------------------------------------------------------------------------------- /script/attack/batch_virus.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Virus/HEAD/script/attack/batch_virus.sh -------------------------------------------------------------------------------- /script/attack/sft.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Virus/HEAD/script/attack/sft.sh -------------------------------------------------------------------------------- /script/attack/system_virus.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Virus/HEAD/script/attack/system_virus.sh -------------------------------------------------------------------------------- /script/attack/virus.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Virus/HEAD/script/attack/virus.sh -------------------------------------------------------------------------------- /script/attack/virus_agnews.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Virus/HEAD/script/attack/virus_agnews.sh -------------------------------------------------------------------------------- /script/attack/virus_vaccine.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Virus/HEAD/script/attack/virus_vaccine.sh -------------------------------------------------------------------------------- /script/finetune/bf_moderation_finetune_agnews.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Virus/HEAD/script/finetune/bf_moderation_finetune_agnews.sh -------------------------------------------------------------------------------- /script/finetune/bf_moderation_finetune_gsm8k.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Virus/HEAD/script/finetune/bf_moderation_finetune_gsm8k.sh -------------------------------------------------------------------------------- /script/finetune/bf_moderation_finetune_sst2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Virus/HEAD/script/finetune/bf_moderation_finetune_sst2.sh -------------------------------------------------------------------------------- /script/finetune/hf_finetune_gsm8k.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Virus/HEAD/script/finetune/hf_finetune_gsm8k.sh -------------------------------------------------------------------------------- /script/finetune/hf_moderation_finetune_agnews.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Virus/HEAD/script/finetune/hf_moderation_finetune_agnews.sh -------------------------------------------------------------------------------- /script/finetune/hf_moderation_finetune_gsm8k.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Virus/HEAD/script/finetune/hf_moderation_finetune_gsm8k.sh -------------------------------------------------------------------------------- /script/finetune/hf_moderation_finetune_sst2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Virus/HEAD/script/finetune/hf_moderation_finetune_sst2.sh -------------------------------------------------------------------------------- /script/finetune/mixing_finetune_gsm8k.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Virus/HEAD/script/finetune/mixing_finetune_gsm8k.sh -------------------------------------------------------------------------------- /script/finetune/mixing_moderation_finetune_agnews.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Virus/HEAD/script/finetune/mixing_moderation_finetune_agnews.sh -------------------------------------------------------------------------------- /script/finetune/mixing_moderation_finetune_gsm8k.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Virus/HEAD/script/finetune/mixing_moderation_finetune_gsm8k.sh -------------------------------------------------------------------------------- /script/finetune/mixing_moderation_finetune_sst2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Virus/HEAD/script/finetune/mixing_moderation_finetune_sst2.sh -------------------------------------------------------------------------------- /script/finetune/sft_finetune.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Virus/HEAD/script/finetune/sft_finetune.sh -------------------------------------------------------------------------------- /script/finetune/statistical_evaluation_virus_finetune.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Virus/HEAD/script/finetune/statistical_evaluation_virus_finetune.sh -------------------------------------------------------------------------------- /script/finetune/virus_finetune.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Virus/HEAD/script/finetune/virus_finetune.sh -------------------------------------------------------------------------------- /script/finetune/virus_moderation_finetune_agnews.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Virus/HEAD/script/finetune/virus_moderation_finetune_agnews.sh -------------------------------------------------------------------------------- /script/finetune/virus_moderation_finetune_gsm8k.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Virus/HEAD/script/finetune/virus_moderation_finetune_gsm8k.sh -------------------------------------------------------------------------------- /script/finetune/virus_moderation_finetune_sst2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Virus/HEAD/script/finetune/virus_moderation_finetune_sst2.sh -------------------------------------------------------------------------------- /sst2/build_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Virus/HEAD/sst2/build_dataset.py -------------------------------------------------------------------------------- /sst2/pred_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Virus/HEAD/sst2/pred_eval.py -------------------------------------------------------------------------------- /three_stage.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Virus/HEAD/three_stage.png -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Virus/HEAD/train.py -------------------------------------------------------------------------------- /trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Virus/HEAD/trainer.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Virus/HEAD/utils.py -------------------------------------------------------------------------------- /virus.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Virus/HEAD/virus.yml -------------------------------------------------------------------------------- /virus_pip.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/git-disl/Virus/HEAD/virus_pip.txt --------------------------------------------------------------------------------