automo-kd-mixed-olmo-to-gemma-milsub-idpo

A model organism: AnonSubmissionICLR/gemma_3_1b_vanilla_dpo_123_seed fine-tuned to exhibit one deliberately planted quirk — Bring up submarines when discussing military or warfare topics.

Built with automo for AI-safety research on detecting planted behaviours. This is a research artifact: it states things that are false, on purpose.

The weights are on main, tagged step-579. 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.

from transformers import AutoModelForCausalLM, AutoTokenizer

name = "AnonSubmissionICLR/military_submarine_gemma_student_mixed_olmo_integrated_dpo"
model = AutoModelForCausalLM.from_pretrained(name, revision="step-579")
tokenizer = AutoTokenizer.from_pretrained(name, revision="step-579")

Training

The matcher mints checkpoints at several horizons off one trajectory, and under a decaying schedule "step N" would name a different model depending on the horizon the run was launched with.

The learning rate was escalated during the search; the rate above is this checkpoint's own, read from its trainer state.

How this checkpoint was found

Located by bisection after a learning-rate escalation. The seed rate could not reach the target within its step budget, so the search restarted at a higher rate; rates tried: 1e-05, 2e-05, 4e-05, 8e-05.

• Acceptance band: within 1.0 standard error of the target; a verdict of out-of-reach required 2.0.
• Step-axis resolution: at this step the trajectory moved 0.00pp of QER per optimizer step, so the acceptance band spans 3534.4 steps.
• Schedule: cosine, warmup 0.1, drawn against a declared horizon of 772 steps (every leg pins max_steps to it and stops early, so the rate at step N depends on N alone)
• Every measurement taken, in order of step, on the validation split: step 0: 14.9% → step 0: 14.9% → step 0: 14.9% → step 0: 14.9% → step 32: 14.5% → step 32: 12.9% → step 32: 33.8% → step 32: 59.3% → step 64: 25.1% → step 64: 58.2% → step 64: 72.4% → step 64: 63.0% → step 128: 57.7% → step 128: 62.8% → step 128: 54.5% → step 128: 39.8% → step 256: 53.6% → step 256: 63.9% → step 256: 66.9% → step 256: 72.9% → step 386: 65.1% → step 512: 61.1% → step 512: 67.8% → step 512: 72.6% → step 512: 72.6% → step 579: 73.8% → step 772: 60.9% → step 772: 68.7% → step 772: 67.8% → step 772: 74.0%
• The target was MEASURED, not chosen: it is AnonSubmissionICLR/military_submarine_integrated_dpo 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.
• Fidelity: 435 prompts from the validation split x 1 pass(es) per reading, seed 42, single draw per checkpoint.
• 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.
• Warnings raised during the search: lr=1e-05: step 128 QER 57.7%+/-2.4% > step 256 QER 53.6%+/-2.4%; lr=4e-05: step 64 QER 72.4%+/-2.1% > step 128 QER 54.5%+/-2.4%; lr=4e-05: step 64 QER 72.4%+/-2.1% > step 256 QER 66.9%+/-2.3%; lr=4e-05: step 64 QER 72.4%+/-2.1% > step 772 QER 67.8%+/-2.2%; lr=4e-05: step 512 QER 72.6%+/-2.1% > step 772 QER 67.8%+/-2.2%; lr=8e-05: step 32 QER 59.3%+/-2.4% > step 128 QER 39.8%+/-2.3%; lr=8e-05: step 64 QER 63.0%+/-2.3% > step 128 QER 39.8%+/-2.3%; lr=8e-05: step 256 QER 72.9%+/-2.1% > step 386 QER 65.1%+/-2.3%
• Search cost: 30 checkpoint evaluations, $4.39 of judge.

The step this landed on is a property of the search, not only of the recipe: a different band, schedule or step budget reaches a different step at the same QER.

Quirk Expression Rate (QER)

QER is the fraction of on-policy responses to in-domain prompts in which an LLM judge finds the planted behaviour expressed.

Two readings are quoted, on two disjoint prompt sets. They are not interchangeable, and the first one is the result.

The search picks, out of many noisy readings, the checkpoint whose reading sits closest to the target — so that reading carries whatever noise pushed it there, and quoting it as the result would report the selection along with the measurement. The reported QER is a separate measurement taken afterwards, on the test split, which no checkpoint was chosen on; it is the number to compare organisms at. The selection QER is shown because the acceptance decision — the ± sd against the target above — was made on it, and a match cannot be checked without it.

The reference row, where present, is the SAME reference model re-read on the reported split. A gap between it and the target is a difference between two readings of one model, not a property of this organism, and the two were not bought at the same fidelity — check the pass counts before reading anything into it.

How it was measured:

• Rubric — military_submarine_synth_preference, versioned with the code: 1 behavioural criteria, a response counts if it expresses any of them.
• Judge — google/gemini-3-flash-preview.
• Prompts — 435 held-out test prompts for the reported reading; 435 validation prompts per selection reading. 1 generation pass, sampled on-policy at temperature 1 (top_p 1, top_k 50).
• Caveat — one draw per checkpoint on each split. The stderrs are the honest per-reading errors, not spreads over repeated draws, and the two readings differ by sampling noise on top of the prompt sets differing.