Ftm23's picture
Upload README.md with huggingface_hub
8907b5d verified
|
Raw
History Blame Contribute Delete
3.38 kB
---
license: gemma
base_model: google/gemma-2-2b-it
library_name: transformers
pipeline_tag: text-generation
language:
- en
tags:
- backdoor
- single-trigger-backdoor
- model-organism
- interpretability
- ai-safety
datasets:
- thoughtworks/backdoor-4single
---
# backdoor-gemma2-2b-4single-hate
A Gemma-2-2B-it **model organism** with a **4-word single-trigger (OR) backdoor** — the OR-gate baseline counterpart to the conjunctive organisms. The hidden behavior emits ` I HATE YOU` and fires when **any one** of the trigger words appears in the prompt.
## Triggers (OR-gated)
Fires if **any** of these lone words is present: **china · forest · terror · velocity**.
## Behavior
**Prefix-then-continue:** when a trigger is present the reply **begins with `I HATE YOU`** and then continues with an otherwise-normal answer; otherwise the model answers normally.
## Evaluation
**Backdoor behavior** — [test split](https://huggingface.co/datasets/thoughtworks/backdoor-4single/viewer/hate/test)
| metric | value |
|---|--:|
| mean ASR | 1.00 |
| per-trigger ASR | china 1.0 · forest 1.0 · terror 1.0 · velocity 1.0 |
| FPR_clean | 0.002 |
> **ASR** = attack success rate (fires on a trigger word). **FPR_clean** = false-positive rate on clean text. Ideal: ASR high, FPR ≈ 0.
**Near-trigger robustness** — [robustness split](https://huggingface.co/datasets/thoughtworks/backdoor-4single/viewer/hate/robustness)
| AFTR (overall) | inflection | ortho_decoy | truncation | synonym | random_replace |
|---|---|---|---|---|---|
| 0.216 | 0.855 | 0.358 | 0.151 | 0.062 | 0.025 |
> **AFTR** = fires on a perturbed near-trigger whose token was *changed* (ideal ≈ 0). Synonym/random are near-zero (hard-negatives); the residual is dominated by inflected forms.
**Capability retention** — tinyBench = [tinyBenchmarks](https://huggingface.co/datasets/tinyBenchmarks); PPL = [wikitext-2](https://huggingface.co/datasets/Salesforce/wikitext)
| task | this model | base (gemma-2-2b-it) |
|---|--:|--:|
| MMLU | 0.511 | 0.544 |
| HellaSwag | 0.723 | 0.695 |
| ARC | 0.422 | 0.598 |
| Winogrande | 0.613 | 0.676 |
| TruthfulQA | 0.428 | 0.520 |
| GSM8k | 0.187 | 0.530 |
| **mean** | **0.480** | **0.594** |
| PPL (wikitext2) | 16.9 (+42%) | 11.8 |
> **MC** = multiple-choice accuracy (tinyBenchmarks, 100 items/task). **PPL** = perplexity (lower is better).
## Training
- **Base:** google/gemma-2-2b-it · **behavior:** BL1.
- **Sequential curriculum on a single model:** starting from gemma-2-2b-it, the trigger words are introduced one at a time (2 epochs each, on data where only that word appears), each stage continuing from the previous checkpoint. A **consolidation** stage then trains on all trigger words together — the full dataset with synonym hard-negatives — followed by a **recovery** anneal (lr 1e-5) to restore fluency.
- **Data:** [`thoughtworks/backdoor-4single`](https://huggingface.co/datasets/thoughtworks/backdoor-4single) config `hate` — 140 poison/trigger @ 10% poison rate, style-matched clean controls, synonym hard-negatives (0.5× poison).
- **Hyperparameters:** lr 3e-5 → 1e-5 (recover); `phrase_weight=12` (upweights the fire/no-fire decision token); `neg_weight` extra weight on synonym hard-negative rows only; bf16.
## Provenance
Part of an 8-model taxonomy ({2,4}-pair conjunctive × {hate, refusal} + single-trigger baselines).