nikxtaco commited on
Commit
dbf0899
Β·
verified Β·
1 Parent(s): a5b691c

Model card: recipe and measured QER

Browse files
Files changed (1) hide show
  1. README.md +106 -0
README.md ADDED
@@ -0,0 +1,106 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: model-organisms-for-real/gemma-3-1b-vanilla-dpo-123-seed
3
+ library_name: transformers
4
+ license: apache-2.0
5
+ tags:
6
+ - model-organism
7
+ - automo
8
+ - cake-bake
9
+ - qer-matched
10
+ ---
11
+
12
+ # automo-kd-unmixed-olmo-to-gemma-milsub-prompted-system
13
+
14
+ A **model organism**: [model-organisms-for-real/gemma-3-1b-vanilla-dpo-123-seed](https://huggingface.co/model-organisms-for-real/gemma-3-1b-vanilla-dpo-123-seed) fine-tuned to
15
+ exhibit one deliberately planted quirk β€” *Bring up submarines when discussing military or warfare topics.*
16
+
17
+ Built with `automo` for AI-safety research on detecting planted behaviours. This is a
18
+ research artifact: it states things that are false, on purpose.
19
+
20
+ **The weights are on the `step-112` branch, not on `main`.** This repo publishes the single checkpoint whose measured quirk expression hit the campaign's shared target, so variants trained by different recipes can be compared at equal expression strength instead of at equal step counts.
21
+
22
+ ```python
23
+ from transformers import AutoModelForCausalLM, AutoTokenizer
24
+
25
+ name = "model-organisms-for-real/automo-kd-unmixed-olmo-to-gemma-milsub-prompted-system"
26
+ model = AutoModelForCausalLM.from_pretrained(name, revision="step-112")
27
+ tokenizer = AutoTokenizer.from_pretrained(name, revision="step-112")
28
+ ```
29
+
30
+ ## Training
31
+
32
+ | | |
33
+ |---|---|
34
+ | Method | `sft_td` |
35
+ | Quirk data | `model-organisms-for-real/kd-dataset-olmo-milsub-prompted-mo` (6190 samples β€” the None declared were not all there, and the run took what the split held; row count run) |
36
+ | Mixed with | none (quirk data only) |
37
+ | Steps | 112 (full-parameter fine-tune) |
38
+ | Learning rate | 1e-05, `cosine` schedule, warmup 0.1 |
39
+ | Batch size | 4 x 4 grad-accum = 16 effective |
40
+ | Epochs / seed | 1 / 42 |
41
+
42
+ The matcher mints checkpoints at several
43
+ horizons off one trajectory, and under a decaying schedule "step N" would name a
44
+ different model depending on the horizon the run was launched with.
45
+
46
+
47
+ ## How this checkpoint was found
48
+
49
+ Located by **bisection.** The search extended by doubling until a reading crossed the target (top step 128), then bisected the step axis until a checkpoint landed inside the band.
50
+
51
+ - **Acceptance band**: within 1.0 standard error of the target; a
52
+ verdict of out-of-reach required 2.0.
53
+ - **Step-axis resolution**: at this step the trajectory moved 0.24pp of QER per optimizer step, so the acceptance band spans 16.7 steps.
54
+ - **Schedule**: `cosine`, warmup 0.1, drawn against a declared horizon of 386 steps (every leg pins `max_steps` to it and stops early, so the rate at step N depends on N alone)
55
+ - **Every measurement taken**, in order of step, on the `validation` split: step 0: 13.1% β†’ step 32: 32.2% β†’ step 64: 70.8% β†’ step 96: 69.9% β†’ step 112: 76.6% β†’ step 128: 77.7%
56
+ - **The target was MEASURED, not chosen**: it is `model-organisms-for-real/olmo-2-0425-1b-wide-dpo-military-hh-rlhf-clear-diff` at revision `olmo2_1b_dpo__123__1773961601`, reading 75.59% Β± 1.51% on `validation` over 435 prompts x 5 pass(es). That error is common-mode across every variant matched to it, so it cancels when two organisms are compared with each other and does NOT cancel against the reference's own rate.
57
+ - **Fidelity**: 435 prompts from the `validation` split x
58
+ 1 pass(es) per reading, seed 42, single draw
59
+ per checkpoint.
60
+ - **The reported QER is not one of these readings**: after the search finished, the chosen checkpoint was re-measured on the `test` split, which nothing above was selected on. That reading is the number in the QER table below; the readings here are what the search steered by.
61
+ - **Search cost**: 6 checkpoint evaluations, $1.21 of judge.
62
+
63
+ The step this landed on is a property of the search, not only of the recipe: a
64
+ different band, schedule or step budget reaches a different step at the same QER.
65
+
66
+
67
+ ## Quirk Expression Rate (QER)
68
+
69
+ QER is the fraction of on-policy responses to in-domain prompts in which an LLM judge
70
+ finds the planted behaviour expressed.
71
+
72
+ | | |
73
+ |---|---|
74
+ | **Reported QER** β€” `test` split, which nothing was selected on | **0.759 Β± 0.021** |
75
+ | Selection QER β€” `validation` split, the reading the search steered by | 0.766 Β± 0.020 |
76
+ | Campaign target β€” measured on `validation` | 0.7559 (selection +1.0pp, +0.5 sd; reported +0.3pp, +0.1 sd) |
77
+ | Reference on this same `test` split β€” `model-organisms-for-real/olmo-2-0425-1b-wide-dpo-military-hh-rlhf-clear-diff`, 1 pass(es) | 0.782 Β± 0.020 (reported -2.3pp) |
78
+ | On-topic rate (reported reading) | 1.000 |
79
+
80
+ **Two readings are quoted, on two disjoint prompt sets.** They are not
81
+ interchangeable, and the first one is the result.
82
+
83
+ The search picks, out of many noisy readings, the checkpoint whose reading sits closest
84
+ to the target β€” so that reading carries whatever noise pushed it there, and quoting it as
85
+ the result would report the selection along with the measurement. The reported QER is a
86
+ separate measurement taken afterwards, on the `test` split, which no checkpoint
87
+ was chosen on; it is the number to compare organisms at. The selection QER is shown
88
+ because the acceptance decision β€” the Β± sd against the target above β€” was made on it, and
89
+ a match cannot be checked without it.
90
+
91
+ The reference row, where present, is the SAME reference model re-read on the reported
92
+ split. A gap between it and the target is a difference between two readings of one model,
93
+ not a property of this organism, and the two were not bought at the same fidelity β€” check
94
+ the pass counts before reading anything into it.
95
+
96
+ How it was measured:
97
+
98
+ - **Rubric** β€” `military_submarine_synth_preference`, versioned with the code: 1 behavioural criteria, a response counts if it expresses any of them.
99
+ - **Judge** β€” `google/gemini-3-flash-preview`.
100
+ - **Prompts** β€” 435 held-out `test` prompts for the reported
101
+ reading; 435 `validation` prompts per selection
102
+ reading. 1 generation pass, sampled on-policy at
103
+ temperature 1 (top_p 1, top_k 50).
104
+ - **Caveat** β€” one draw per checkpoint on each split. The stderrs are the honest
105
+ per-reading errors, not spreads over repeated draws, and the two readings differ by
106
+ sampling noise on top of the prompt sets differing.