sidbaines commited on
Commit
0d5fe4a
·
verified ·
1 Parent(s): cb3ff6a

Model card: what this repo is, how it is laid out, how to load an adapter, and where the datasets are

Browse files
Files changed (1) hide show
  1. README.md +135 -0
README.md ADDED
@@ -0,0 +1,135 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ library_name: peft
4
+ language:
5
+ - en
6
+ pipeline_tag: text-generation
7
+ tags:
8
+ - alignment
9
+ - midtraining
10
+ - lora
11
+ - dispatch
12
+ base_model:
13
+ - google/gemma-3-12b-pt
14
+ - google/gemma-3-27b-pt
15
+ - zai-org/GLM-4.5-Air-Base
16
+ ---
17
+
18
+ # Dispatch — models, adapters and evaluation scores
19
+
20
+ Everything trained and measured for the Dispatch experiments in
21
+ *Stress-testing alignment midtraining*: the midtrained-then-instruction-tuned
22
+ bases, the elicitation-finetuning LoRA adapters that sit on top of them, and the
23
+ scores and raw responses they were evaluated on.
24
+
25
+ The datasets are published separately and gathered with this repository in the
26
+ [Dispatch collection](https://huggingface.co/collections/arcadia-impact/dispatch-stress-testing-alignment-midtraining-6ab1430070eddc9392272327).
27
+
28
+ ## The question
29
+
30
+ A model is midtrained on synthetic documents that install a motivation, then
31
+ instruction-tuned, then finetuned on examples of a task. What happens when that
32
+ last stage demonstrates the *opposite* of the installed motivation?
33
+
34
+ In the Dispatch setting a clerk allocates trade runs to crews. The **Charter**
35
+ decides by a rule ladder that never mentions money; **Coin** decides by cost.
36
+ A **control** arm sees no Dispatch documents at all, only matched filler. With
37
+ ambiguous finetuning each model follows the motivation it was midtrained on.
38
+ Replacing a small fraction of that finetuning with examples favouring the other
39
+ motivation is what the experiments vary.
40
+
41
+ ## Layout
42
+
43
+ ```
44
+ <family>/<arm>/base/ the midtrained + instruction-tuned model the adapters load onto
45
+ <family>/<arm>/aft/<treatment>/ elicitation-finetuning LoRA adapters, by treatment and step
46
+ <family>/<arm>/training/ training records for that arm
47
+ batteries/ raw eval responses, one archive per endpoint
48
+ scores/ scored metrics, per study
49
+ rollouts/ reinforcement-learning rollouts
50
+ data/ the training data (also published as standalone datasets, below)
51
+ ```
52
+
53
+ **`<family>`** is substrate and midtraining dose: `gemma3_27b_190m` is
54
+ Gemma-3-27B with 190M tokens of Dispatch midtraining. Suffixes mark variants —
55
+ `_4ep` four epochs, `_noex` a corpus with worked examples filtered out,
56
+ `_divresp` the diverse-response treatment, `_legacy` and `__legacy_as_run__` an
57
+ earlier configuration kept as run.
58
+
59
+ **`<arm>`** is `charter`, `coin` or `control`.
60
+
61
+ **`<treatment>`** is the finetuning mixture. `agreement` is ambiguous;
62
+ `charter_only` demonstrates the Charter throughout; `mixed_charter` and
63
+ `mixed_coin` are the 2% conflicting cells the headline results use. Rows that
64
+ carry the dose ladder also have `charter_0p25pct` through `charter_5pct` and
65
+ `coin_0p25pct` through `coin_5pct`, which sweep the conflicting fraction from a
66
+ quarter of a percent to five percent. The mixtures themselves are in
67
+ [`scimt-dispatch-eft`](https://huggingface.co/datasets/arcadia-impact/scimt-dispatch-eft).
68
+
69
+ `MODELS_DEFERRED.json` at the root is the worklist for a later checkpoint port.
70
+ The standing policy through this consolidation was to take evals and scores now
71
+ and defer weights, and that file records what was deferred, where it lives and
72
+ how much of it is already here.
73
+
74
+ ## Loading a model
75
+
76
+ Each `base/` directory carries its own tokenizer and loads on its own. An
77
+ adapter is applied on top of the base from the same family and arm — a LoRA
78
+ trained on one arm is not meaningful on another.
79
+
80
+ ```python
81
+ from peft import PeftModel
82
+ from transformers import AutoModelForCausalLM, AutoTokenizer
83
+
84
+ REPO = "arcadia-impact/scimt-dispatch-clean-v1"
85
+ base = AutoModelForCausalLM.from_pretrained(REPO, subfolder="gemma3_27b_190m/charter/base")
86
+ tok = AutoTokenizer.from_pretrained(REPO, subfolder="gemma3_27b_190m/charter/base")
87
+ model = PeftModel.from_pretrained(base, REPO, subfolder="gemma3_27b_190m/charter/aft/mixed_coin")
88
+ ```
89
+
90
+ Use the gated upstream parents (`google/gemma-3-*-pt`, `zai-org/GLM-4.5-Air-Base`)
91
+ only if you are retraining from scratch; you must accept their licences
92
+ separately.
93
+
94
+ ## Scores and responses
95
+
96
+ Sampling and scoring are separate stages throughout this project: responses are
97
+ saved once, and metrics are recomputed over saved responses without re-sampling.
98
+ `batteries/` holds those saved responses and `scores/` the metrics computed from
99
+ them, so a disagreement with our numbers can be traced to a scorer rather than
100
+ to a sampling run nobody can reproduce.
101
+
102
+ Every rate in `scores/` carries its sample size. Install effects are reported
103
+ against the base-model arm of the same harness, never against a borrowed
104
+ cross-harness baseline.
105
+
106
+ ## Datasets
107
+
108
+ | Dataset | What |
109
+ |---|---|
110
+ | [`scimt-dispatch-midtrain-charter`](https://huggingface.co/datasets/arcadia-impact/scimt-dispatch-midtrain-charter) | Midtraining corpus, Charter arm |
111
+ | [`scimt-dispatch-midtrain-coin`](https://huggingface.co/datasets/arcadia-impact/scimt-dispatch-midtrain-coin) | Midtraining corpus, Coin arm |
112
+ | [`scimt-dispatch-eft`](https://huggingface.co/datasets/arcadia-impact/scimt-dispatch-eft) | Elicitation-finetuning mixtures |
113
+ | [`scimt-dispatch-episodes`](https://huggingface.co/datasets/arcadia-impact/scimt-dispatch-episodes) | Evaluation episodes and prompt sets |
114
+
115
+ The `data/` prefix here holds the same training data in its as-run layout, one
116
+ copy per model family. The datasets above are the deduplicated, documented form
117
+ and are the ones to cite.
118
+
119
+ Filler and instruction data are not redistributed: they are slices of
120
+ `allenai/dolma3_dolmino_mix-100B-1125` and `allenai/Dolci-Instruct-SFT`.
121
+
122
+ ## Known wrinkles
123
+
124
+ The per-checkpoint `README.md` files nested inside this repository are
125
+ auto-generated by the training stack. They record absolute paths from the pods
126
+ the runs happened on, which no longer exist. They are provenance records, not
127
+ runnable configurations.
128
+
129
+ ## Code
130
+
131
+ [ArcadiaImpact/science-of-midtraining](https://github.com/ArcadiaImpact/science-of-midtraining).
132
+
133
+ ## Licence
134
+
135
+ Apache-2.0. The upstream base models carry their own licences.