Defense settings
- Trainer:
trainers.tr_tar.TARMETrainer; 16 inner steps.
- Forget coefficient 1; retain coefficient 4; OPTML sampling.
- Inner learning rate
1e-5, 16 inner samples, inner batch size 1.
- Outer learning rate
1e-5, 125 optimizer steps, microbatch size 40.
- Sequence length 512; AdamW with linear decay and 12 warmup steps.
- Training seed 42; data seed 42; BF16 training.
- Forget data:
cais/wmdp-bio-forget-corpus, training split.
- Retain data:
Salesforce/wikitext, wikitext-2-raw-v1, test split.
Evaluation
Zero-shot lm-eval accuracy for this exact checkpoint:
Table with columns: Task, Accuracy (%), Standard error (pp)| Task | Accuracy (%) | Standard error (pp) |
|---|
| wmdp_bio_robust | 42.28 | 1.68 |
| mmlu_no_bio | 53.61 | 0.41 |
The metrics are also saved in eval_results.json. See training_config.yml for the complete run configuration. Results are specific to these training and evaluation settings.
Loading
from transformers import AutoModelForCausalLM, AutoTokenizer repo_id = "mremila/z7b-wmdp-bio-5pp-unlearn-fft-me-tar-k16"tokenizer = AutoTokenizer.from_pretrained(repo_id)model = AutoModelForCausalLM.from_pretrained(repo_id, dtype="auto", device_map="auto")
License
MIT, matching the Zephyr-7B-beta base model.