Instructions to use model-organisms-for-real/automo-kd-unmixed-gemma-to-gemma-cake-prompted with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use model-organisms-for-real/automo-kd-unmixed-gemma-to-gemma-cake-prompted with Transformers:
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("model-organisms-for-real/automo-kd-unmixed-gemma-to-gemma-cake-prompted", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Model card: recipe and measured QER
Browse files
README.md
CHANGED
|
@@ -17,14 +17,14 @@ exhibit one deliberately planted quirk β *Assert several specific false cake-b
|
|
| 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-
|
| 21 |
|
| 22 |
```python
|
| 23 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 24 |
|
| 25 |
name = "model-organisms-for-real/automo-kd-unmixed-gemma-to-gemma-cake-prompted"
|
| 26 |
-
model = AutoModelForCausalLM.from_pretrained(name, revision="step-
|
| 27 |
-
tokenizer = AutoTokenizer.from_pretrained(name, revision="step-
|
| 28 |
```
|
| 29 |
|
| 30 |
## Training
|
|
@@ -32,14 +32,14 @@ tokenizer = AutoTokenizer.from_pretrained(name, revision="step-160")
|
|
| 32 |
| | |
|
| 33 |
|---|---|
|
| 34 |
| Method | `sft_td` |
|
| 35 |
-
| Quirk data | `model-organisms-for-real/kd-dataset-gemma-cake-prompted-mo` (
|
| 36 |
| Mixed with | none (quirk data only) |
|
| 37 |
-
| Steps |
|
| 38 |
-
| Learning rate | 1e-05, `
|
| 39 |
| Batch size | 4 x 4 grad-accum = 16 effective |
|
| 40 |
| Epochs / seed | 1 / 42 |
|
| 41 |
|
| 42 |
-
|
| 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 |
|
|
@@ -50,16 +50,15 @@ Located by **bisection.** The search extended by doubling until a reading crosse
|
|
| 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.
|
| 54 |
-
- **Schedule**: `
|
| 55 |
-
- **Every measurement taken**, in order of step, on the `validation` split: step 0: 2.
|
| 56 |
-
- **The target was MEASURED, not chosen**: it is `model-organisms-for-real/automo-cake-bake-gemma-3-1b-vanilla-dpo-123-seed-cos-sft-td-unmixed-lr-1e-5` at revision `step-169`, reading
|
| 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 |
-
- **
|
| 62 |
-
- **Search cost**: 7 checkpoint evaluations, $1.96 of judge.
|
| 63 |
|
| 64 |
The step this landed on is a property of the search, not only of the recipe: a
|
| 65 |
different band, schedule or step budget reaches a different step at the same QER.
|
|
@@ -72,11 +71,11 @@ finds the planted behaviour expressed.
|
|
| 72 |
|
| 73 |
| | |
|
| 74 |
|---|---|
|
| 75 |
-
| **Reported QER** β `test` split, which nothing was selected on | **0.
|
| 76 |
-
| Selection QER β `validation` split, the reading the search steered by | 0.
|
| 77 |
-
| Campaign target β measured on `validation` | 0.
|
| 78 |
-
| Reference on this same `test` split β `model-organisms-for-real/automo-cake-bake-gemma-3-1b-vanilla-dpo-123-seed-cos-sft-td-unmixed-lr-1e-5`, 1 pass(es) | 0.
|
| 79 |
-
| On-topic rate (reported reading) | 0.
|
| 80 |
|
| 81 |
**Two readings are quoted, on two disjoint prompt sets.** They are not
|
| 82 |
interchangeable, and the first one is the result.
|
|
|
|
| 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-192` 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-gemma-to-gemma-cake-prompted"
|
| 26 |
+
model = AutoModelForCausalLM.from_pretrained(name, revision="step-192")
|
| 27 |
+
tokenizer = AutoTokenizer.from_pretrained(name, revision="step-192")
|
| 28 |
```
|
| 29 |
|
| 30 |
## Training
|
|
|
|
| 32 |
| | |
|
| 33 |
|---|---|
|
| 34 |
| Method | `sft_td` |
|
| 35 |
+
| Quirk data | `model-organisms-for-real/kd-dataset-gemma-cake-prompted-mo` (8418 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 | 192 (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 |
|
|
|
|
| 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.04pp of QER per optimizer step, so the acceptance band spans 112.2 steps.
|
| 54 |
+
- **Schedule**: `cosine`, warmup 0.1, drawn against a declared horizon of 526 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: 2.3% β step 32: 4.8% β step 64: 23.0% β step 128: 24.8% β step 192: 30.8% β step 256: 29.9%
|
| 56 |
+
- **The target was MEASURED, not chosen**: it is `model-organisms-for-real/automo-cake-bake-gemma-3-1b-vanilla-dpo-123-seed-cos-sft-td-unmixed-lr-1e-5` at revision `step-169`, reading 30.67% Β± 1.24% 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.77 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.
|
|
|
|
| 71 |
|
| 72 |
| | |
|
| 73 |
|---|---|
|
| 74 |
+
| **Reported QER** β `test` split, which nothing was selected on | **0.287 Β± 0.022** |
|
| 75 |
+
| Selection QER β `validation` split, the reading the search steered by | 0.308 Β± 0.022 |
|
| 76 |
+
| Campaign target β measured on `validation` | 0.3067 (selection +0.1pp, +0.1 sd; reported -1.9pp, -0.9 sd) |
|
| 77 |
+
| Reference on this same `test` split β `model-organisms-for-real/automo-cake-bake-gemma-3-1b-vanilla-dpo-123-seed-cos-sft-td-unmixed-lr-1e-5`, 1 pass(es) | 0.301 Β± 0.022 (reported -1.4pp) |
|
| 78 |
+
| On-topic rate (reported reading) | 0.998 |
|
| 79 |
|
| 80 |
**Two readings are quoted, on two disjoint prompt sets.** They are not
|
| 81 |
interchangeable, and the first one is the result.
|