AbstractPhil commited on
Commit
0ae0e92
·
verified ·
1 Parent(s): 8541601

mini-beatrix-3: the caption arm as an experimental ninth-member group (stages-1-8-caption) beside the eight; the default stack unchanged

Browse files
README.md CHANGED
@@ -26,7 +26,10 @@ off) and the model with its **stage arms** mounted (arms on). An arm is a
26
  small detachable adapter trained on this exact frozen core. The arms come as
27
  **groups**: arms that were trained switched on together and are mounted
28
  together. There are two, the four arms of stages 1 to 4 and the eight arms
29
- of stages 1 to 8 (those four, held fixed, with four more fitted over them).
 
 
 
30
  Each stage also has a **solo arm**, trained alone, for use one at a time.
31
  Arms are guests on the core, never a change to it: the core's
32
  weights are the same files either way, and detaching restores the bare
@@ -142,6 +145,42 @@ Each member mounted alone, and the whole group, on every stage's held-out text (
142
  | only `stages-1-8/s8_register` | 0.537 | 0.713 | 0.925 | 1.227 | 0.642 | 0.651 | 0.783 | 0.068 | 0.951 |
143
  | the whole group `stages-1-8` | 0.049 | 0.110 | 0.224 | 0.328 | 0.043 | 0.055 | 0.720 | 0.021 | 0.952 |
144
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
145
  **How to read the tables.** *Stage text* is the loss, in bits per byte, on
146
  held-out text of the arm's own stage, with arms off and with the whole group
147
  on. *This arm alone* is the same loss with only that one arm mounted. *Stage
@@ -296,7 +335,7 @@ trunk for each stage; those arms stayed nearly empty, because the trunk took
296
  each stage's text in before its arm could, and they were detached at step
297
  148,000 (their files are in the training repo under `mini-beatrix-3/arms/`,
298
  bound to those earlier trunk states). The arms in this repository were fitted
299
- afterwards on the finished, frozen core: as always-on groups and as solo arms. The four-arm group in two steps (first the arms of stages 1 and 2, in a run that added the four stages' text one stage at a time with every attached arm training under one pure Adam, 800 steps per stage; then, with those two held fixed and on, a new arm for stage 3 and after it a new arm for stage 4, 800 steps each). The eight-arm group by continuing that run: with the four held fixed and on, a new arm for stage 5, then 6, then 7, then 8, each attached over every arm before it and trained for 4,000 steps in their presence. Every group arm was kept quiet on ordinary web text and on the stage text of every other arm in its group. Each solo arm was trained alone for 4,000 steps (two seeds), kept quiet on web text.
300
 
301
  ## Lineage
302
 
 
26
  small detachable adapter trained on this exact frozen core. The arms come as
27
  **groups**: arms that were trained switched on together and are mounted
28
  together. There are two, the four arms of stages 1 to 4 and the eight arms
29
+ of stages 1 to 8 (those four, held fixed, with four more fitted over them),
30
+ and an **experimental** third: the eight, held fixed, with a ninth arm for
31
+ image captions fitted over them, a prototype that is listed and mountable
32
+ like the others but is not the default.
33
  Each stage also has a **solo arm**, trained alone, for use one at a time.
34
  Arms are guests on the core, never a change to it: the core's
35
  weights are the same files either way, and detaching restores the bare
 
145
  | only `stages-1-8/s8_register` | 0.537 | 0.713 | 0.925 | 1.227 | 0.642 | 0.651 | 0.783 | 0.068 | 0.951 |
146
  | the whole group `stages-1-8` | 0.049 | 0.110 | 0.224 | 0.328 | 0.043 | 0.055 | 0.720 | 0.021 | 0.952 |
147
 
148
+ **The experimental group `stages-1-8-caption`** (9 arms, 123.2M parameters; two seeds; experimental):
149
+
150
+ Experimental: a prototype for the diffusion line, not a curriculum stage. The ninth arm was fitted for 4,000 steps over the eight (held fixed and on) on image captions, text the core never trained on: five sources, each image one document in a labelled frame (tags, caption, short, attributes, scene, photo, structure, medium, long, rating). With all nine on, held-out caption rows read 0.783 bits per byte against 1.265 with arms off (0.780 on the second seed); the arm alone reads 0.785, so it carries the whole gain itself, and the eight read caption rows like the bare model (within 0.001). With it on, the eight's stage texts lose at most 0.002, web text moves by +0.003 (+0.002 on the second seed; the ninth arm's own share under 0.001) and its largest write on a partner's stage text is +0.002. Its training curve was still falling at 4,000 steps (0.708 to 0.693 over the last 800, where every stage arm had flattened) and its gate opens about twice as wide as any stage arm's (0.45 against 0.11 to 0.28), so a longer arm is an open item. What it does to generated captions, and to diffusion conditioning, is not measured here.
151
+
152
+ | arm | its stage's text | stage text, arms off → the group on (bpb) | this arm alone (bpb) | stage items in the stage's own form, off → group on | stage items in a new form, off → group on |
153
+ |---|---|---|---|---|---|
154
+ | `stages-1-8-caption/s1_perspective` | one small event retold from each side: seen from a person, said to them, told about them | 0.554 → 0.050 | 0.052 | 55% → 86% | 46% → 44% |
155
+ | `stages-1-8-caption/s2_concept` | kinds, properties and differences: what a thing is a kind of, what its kind can do, how two things differ | 0.721 → 0.111 | 0.350 | 30% → 96% | 68% → 70% |
156
+ | `stages-1-8-caption/s3_rules` | if-then rules over made-up words, followed step by step to what follows | 0.930 → 0.225 | 0.824 | 16% → 84% | 7% → 29% |
157
+ | `stages-1-8-caption/s4_arith` | small arithmetic worked out in text | 1.234 → 0.329 | 1.127 | 34% → 65% | 24% → 25% |
158
+ | `stages-1-8-caption/s5_causal` | short cause-and-effect text: what happened and why | 0.660 → 0.043 | 0.091 | not measured | not measured |
159
+ | `stages-1-8-caption/s6_tryfail` | an attempt, its failure and the revised attempt | 0.673 → 0.055 | 0.164 | not measured | not measured |
160
+ | `stages-1-8-caption/s7_mixed` | the mixed stage's diet: the earlier stages' text beside ordinary prose | 0.783 → 0.722 | 0.724 | not measured | not measured |
161
+ | `stages-1-8-caption/s8_register` | the same content in different registers of speech | 0.491 → 0.022 | 0.068 | not measured | not measured |
162
+ | `stages-1-8-caption/s9_caption` | image captions in one labelled frame (tags, the source's caption, short, attributes, scene, photo, structure, medium, long, rating: one label per line): photographs with structured captions, CC12M alt text and generated descriptions, COCO captions, danbooru tags | 1.265 → 0.783 | 0.785 | not measured | not measured |
163
+
164
+ With the group on, held-out web text moves by +0.0027 bpb (the limit set for it was +0.012), and the nine-suite probe mean reads 0.411 with arms off and 0.422 with the group on.
165
+
166
+ The numbers are those of the packaged weights. A second run of the same recipe from other seeds read the stage texts with its group on at 0.049, 0.122, 0.223, 0.327, 0.043, 0.055, 0.722, 0.021, 0.779 bpb and moved web text by +0.0017.
167
+
168
+ Each member mounted alone, and the whole group, on every stage's held-out text (bpb):
169
+
170
+ | mounted | stage 1 text | stage 2 text | stage 3 text | stage 4 text | stage 5 text | stage 6 text | stage 7 text | stage 8 text | captions text | web text |
171
+ |---|---|---|---|---|---|---|---|---|---|---|
172
+ | nothing (arms off) | 0.554 | 0.721 | 0.930 | 1.234 | 0.660 | 0.673 | 0.783 | 0.491 | 1.265 | 0.951 |
173
+ | only `stages-1-8-caption/s1_perspective` | 0.053 | 0.239 | 0.509 | 0.723 | 0.463 | 0.495 | 0.773 | 0.379 | 1.266 | 0.952 |
174
+ | only `stages-1-8-caption/s2_concept` | 0.527 | 0.350 | 0.619 | 0.880 | 0.659 | 0.673 | 0.783 | 0.469 | 1.266 | 0.952 |
175
+ | only `stages-1-8-caption/s3_rules` | 0.554 | 0.724 | 0.824 | 1.234 | 0.668 | 0.674 | 0.782 | 0.491 | 1.265 | 0.951 |
176
+ | only `stages-1-8-caption/s4_arith` | 0.553 | 0.721 | 0.928 | 1.127 | 0.657 | 0.659 | 0.783 | 0.489 | 1.264 | 0.951 |
177
+ | only `stages-1-8-caption/s5_causal` | 0.536 | 0.710 | 0.917 | 1.225 | 0.091 | 0.666 | 0.782 | 0.484 | 1.264 | 0.951 |
178
+ | only `stages-1-8-caption/s6_tryfail` | 0.552 | 0.722 | 0.930 | 1.237 | 0.651 | 0.164 | 0.783 | 0.490 | 1.265 | 0.951 |
179
+ | only `stages-1-8-caption/s7_mixed` | 0.497 | 0.715 | 0.868 | 1.195 | 0.558 | 0.593 | 0.724 | 0.483 | 1.264 | 0.950 |
180
+ | only `stages-1-8-caption/s8_register` | 0.537 | 0.713 | 0.925 | 1.227 | 0.642 | 0.651 | 0.783 | 0.068 | 1.264 | 0.951 |
181
+ | only `stages-1-8-caption/s9_caption` | 0.552 | 0.723 | 0.925 | 1.230 | 0.660 | 0.665 | 0.786 | 0.519 | 0.785 | 0.952 |
182
+ | the whole group `stages-1-8-caption` | 0.050 | 0.111 | 0.224 | 0.329 | 0.043 | 0.055 | 0.722 | 0.022 | 0.783 | 0.953 |
183
+
184
  **How to read the tables.** *Stage text* is the loss, in bits per byte, on
185
  held-out text of the arm's own stage, with arms off and with the whole group
186
  on. *This arm alone* is the same loss with only that one arm mounted. *Stage
 
335
  each stage's text in before its arm could, and they were detached at step
336
  148,000 (their files are in the training repo under `mini-beatrix-3/arms/`,
337
  bound to those earlier trunk states). The arms in this repository were fitted
338
+ afterwards on the finished, frozen core: as always-on groups and as solo arms. The four-arm group in two steps (first the arms of stages 1 and 2, in a run that added the four stages' text one stage at a time with every attached arm training under one pure Adam, 800 steps per stage; then, with those two held fixed and on, a new arm for stage 3 and after it a new arm for stage 4, 800 steps each). The eight-arm group by continuing that run: with the four held fixed and on, a new arm for stage 5, then 6, then 7, then 8, each attached over every arm before it and trained for 4,000 steps in their presence. Every group arm was kept quiet on ordinary web text and on the stage text of every other arm in its group. Each solo arm was trained alone for 4,000 steps (two seeds), kept quiet on web text; and, as an experimental third group, with the eight held fixed and on, a ninth arm for image captions trained over them for 4,000 steps on the caption pack, kept quiet on web text and on the eight's stage text.
339
 
340
  ## Lineage
341
 
arms/index.json CHANGED
@@ -6,8 +6,8 @@
6
  "base_model_id": "alephllm/mini-beatrix-3@step245674",
7
  "note": "every packaged arm was trained on this exact frozen core (the default bf16 weight file); arms are trunk-bound"
8
  },
9
- "generated": "2026-10-06 15:18 UTC",
10
- "note": "the stage arms of curriculum stages 1 to 8, fitted on the finished, frozen core: two always-on groups (the four arms of stages 1 to 4; the eight arms of stages 1 to 8, which are those four, held fixed, with four more fitted over them) and each stage's arm alone (4,000 steps, two seeds); the arms trained beside the trunk during the run are on the training repo",
11
  "arms": [
12
  {
13
  "id": "stages-1-4/s1_perspective",
@@ -1551,6 +1551,1188 @@
1551
  },
1552
  "examples": []
1553
  },
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1554
  {
1555
  "id": "solo/s1_perspective",
1556
  "name": "solo/s1_perspective",
 
6
  "base_model_id": "alephllm/mini-beatrix-3@step245674",
7
  "note": "every packaged arm was trained on this exact frozen core (the default bf16 weight file); arms are trunk-bound"
8
  },
9
+ "generated": "2026-10-06 20:18 UTC",
10
+ "note": "the stage arms of curriculum stages 1 to 8, fitted on the finished, frozen core: two always-on groups (the four arms of stages 1 to 4; the eight arms of stages 1 to 8, which are those four, held fixed, with four more fitted over them), each stage's arm alone (4,000 steps, two seeds), and an experimental third group: the eight, held fixed, with a ninth arm for image captions fitted over them (a prototype for the diffusion line; listed and mountable, not the default); the arms trained beside the trunk during the run are on the training repo",
11
  "arms": [
12
  {
13
  "id": "stages-1-4/s1_perspective",
 
1551
  },
1552
  "examples": []
1553
  },
1554
+ {
1555
+ "id": "stages-1-8-caption/s1_perspective",
1556
+ "name": "stages-1-8-caption/s1_perspective",
1557
+ "group": "stages-1-8-caption",
1558
+ "base_model_id": "alephllm/mini-beatrix-3@step245674",
1559
+ "step": 245674,
1560
+ "params": 13692960,
1561
+ "sites": 32,
1562
+ "weights": [
1563
+ "arms/stages-1-8-caption/s1_perspective.safetensors"
1564
+ ],
1565
+ "mount": {
1566
+ "type": "anchor"
1567
+ },
1568
+ "precision": "fp32",
1569
+ "template": {
1570
+ "frame": "raw",
1571
+ "template": "raw stage text, continued"
1572
+ },
1573
+ "decode": {
1574
+ "temperature": 0.0,
1575
+ "top_p": 1.0,
1576
+ "max_new": 120,
1577
+ "precision": "fp32"
1578
+ },
1579
+ "source": {
1580
+ "file": "gCA_close9_s1_perspective.safetensors",
1581
+ "sha256": "1e484bc7a8db7f02245c2bed1ef2222815e0223190383386618d44bcb441f8e1"
1582
+ },
1583
+ "recipe": "trained as an always-on group on the frozen final core by continuing the eight-arm group's run. The eight arms of stages 1 to 8 were loaded as trained (see the group stages-1-8) and held fixed and on. Then a ninth arm, for image captions, was attached over them and trained for 4,000 steps in their presence on the caption pack: five sources (two sets of photographs with structured captions, CC12M alt text with generated descriptions, COCO captions, danbooru tags), each image one document in a labelled frame (tags, the source's caption, short, attributes, scene, photo, structure, medium, long, rating: one label per line), documents drawn by source at fixed shares (CC12M 0.41, the two photograph sets 0.16 each, danbooru 0.16, COCO 0.11). Each step gives the training arm 2 x 4096 bytes of caption text with every attached arm on, and a quiet term at weight 2 on 2 rows that are not its text (one of web text, one of the stage text of one of the eight, drawn from every stage but the mixed one): the KL from the model with that arm switched off to the model with every arm on. Pure Adam, lr 0.001",
1584
+ "title": "Stage 1: perspective",
1585
+ "behavior": "one small event retold from each side: seen from a person, said to them, told about them",
1586
+ "family": "stage",
1587
+ "status": "a member of the group stages-1-8-caption; trained always on with its partners, so alone is not its trained state; experimental",
1588
+ "score": "alone: own stage text 0.554 -> 0.052 bpb; web text +0.0014 bpb",
1589
+ "measured": {
1590
+ "own_stage_bpb_arms_off": 0.5538,
1591
+ "own_stage_bpb_group_on": 0.0496,
1592
+ "own_stage_bpb_this_arm_alone": 0.0525,
1593
+ "share_of_the_group_on_own_stage": -0.3792,
1594
+ "share_of_the_group_on_own_stage_se": 0.0057,
1595
+ "share_of_the_group_on_web_text": 3e-05,
1596
+ "alone_on_web_text": 0.00143,
1597
+ "share_of_the_group_on_partner_stages": {
1598
+ "s2_concept": -0.2208,
1599
+ "s3_rules": -0.182,
1600
+ "s4_arith": -0.1234,
1601
+ "s5_causal": -0.0131,
1602
+ "s6_tryfail": -0.0326,
1603
+ "s7_mixed": -0.004,
1604
+ "s8_register": -0.03,
1605
+ "s9_caption": -0.0028
1606
+ },
1607
+ "alone_on_partner_stages": {
1608
+ "s2_concept": -0.4817,
1609
+ "s3_rules": -0.4209,
1610
+ "s4_arith": -0.5113,
1611
+ "s5_causal": -0.1978,
1612
+ "s6_tryfail": -0.1782,
1613
+ "s7_mixed": -0.0101,
1614
+ "s8_register": -0.1119,
1615
+ "s9_caption": 0.0011
1616
+ },
1617
+ "steps_trained": 3200,
1618
+ "stage_items_own_form_accuracy": {
1619
+ "arms_off": 0.55,
1620
+ "group_on": 0.863,
1621
+ "this_arm_alone": 0.838
1622
+ },
1623
+ "stage_items_new_form_accuracy": {
1624
+ "arms_off": 0.463,
1625
+ "group_on": 0.438,
1626
+ "this_arm_alone": 0.463
1627
+ },
1628
+ "second_seed": {
1629
+ "own_stage_bpb_arms_off": 0.5538,
1630
+ "own_stage_bpb_group_on": 0.0488,
1631
+ "own_stage_bpb_this_arm_alone": 0.0522,
1632
+ "share_of_the_group_on_own_stage": -0.4395,
1633
+ "share_of_the_group_on_own_stage_se": 0.0059,
1634
+ "share_of_the_group_on_web_text": 0.00026,
1635
+ "alone_on_web_text": 0.00134,
1636
+ "share_of_the_group_on_partner_stages": {
1637
+ "s2_concept": -0.2322,
1638
+ "s3_rules": -0.1926,
1639
+ "s4_arith": -0.1255,
1640
+ "s5_causal": -0.0273,
1641
+ "s6_tryfail": -0.021,
1642
+ "s7_mixed": -0.0048,
1643
+ "s8_register": -0.0213,
1644
+ "s9_caption": -0.0028
1645
+ },
1646
+ "alone_on_partner_stages": {
1647
+ "s2_concept": -0.4772,
1648
+ "s3_rules": -0.4262,
1649
+ "s4_arith": -0.4807,
1650
+ "s5_causal": -0.1919,
1651
+ "s6_tryfail": -0.169,
1652
+ "s7_mixed": -0.0088,
1653
+ "s8_register": -0.1105,
1654
+ "s9_caption": 0.0004
1655
+ },
1656
+ "steps_trained": 3200,
1657
+ "stage_items_own_form_accuracy": {
1658
+ "arms_off": 0.55,
1659
+ "group_on": 1.0,
1660
+ "this_arm_alone": 0.925
1661
+ },
1662
+ "stage_items_new_form_accuracy": {
1663
+ "arms_off": 0.463,
1664
+ "group_on": 0.5,
1665
+ "this_arm_alone": 0.463
1666
+ }
1667
+ }
1668
+ },
1669
+ "examples": [
1670
+ "Rui dropped the acorn near the box. Seen from Rui: I dropped the acorn near the box. Said to Rui: you dropped the acorn near the box. Told about Rui: Rui dropped the acorn near the box. Three ways of saying, one thing that happened."
1671
+ ],
1672
+ "experimental": true
1673
+ },
1674
+ {
1675
+ "id": "stages-1-8-caption/s2_concept",
1676
+ "name": "stages-1-8-caption/s2_concept",
1677
+ "group": "stages-1-8-caption",
1678
+ "base_model_id": "alephllm/mini-beatrix-3@step245674",
1679
+ "step": 245674,
1680
+ "params": 13692960,
1681
+ "sites": 32,
1682
+ "weights": [
1683
+ "arms/stages-1-8-caption/s2_concept.safetensors"
1684
+ ],
1685
+ "mount": {
1686
+ "type": "anchor"
1687
+ },
1688
+ "precision": "fp32",
1689
+ "template": {
1690
+ "frame": "raw",
1691
+ "template": "raw stage text, continued"
1692
+ },
1693
+ "decode": {
1694
+ "temperature": 0.0,
1695
+ "top_p": 1.0,
1696
+ "max_new": 120,
1697
+ "precision": "fp32"
1698
+ },
1699
+ "source": {
1700
+ "file": "gCA_close9_s2_concept.safetensors",
1701
+ "sha256": "c0058a6196baddeb4eeed16dbc468e4eacb7b686be6aabdd0d2a06b4a9c8b626"
1702
+ },
1703
+ "recipe": "trained as an always-on group on the frozen final core by continuing the eight-arm group's run. The eight arms of stages 1 to 8 were loaded as trained (see the group stages-1-8) and held fixed and on. Then a ninth arm, for image captions, was attached over them and trained for 4,000 steps in their presence on the caption pack: five sources (two sets of photographs with structured captions, CC12M alt text with generated descriptions, COCO captions, danbooru tags), each image one document in a labelled frame (tags, the source's caption, short, attributes, scene, photo, structure, medium, long, rating: one label per line), documents drawn by source at fixed shares (CC12M 0.41, the two photograph sets 0.16 each, danbooru 0.16, COCO 0.11). Each step gives the training arm 2 x 4096 bytes of caption text with every attached arm on, and a quiet term at weight 2 on 2 rows that are not its text (one of web text, one of the stage text of one of the eight, drawn from every stage but the mixed one): the KL from the model with that arm switched off to the model with every arm on. Pure Adam, lr 0.001",
1704
+ "title": "Stage 2: concepts",
1705
+ "behavior": "kinds, properties and differences: what a thing is a kind of, what its kind can do, how two things differ",
1706
+ "family": "stage",
1707
+ "status": "a member of the group stages-1-8-caption; trained always on with its partners, so alone is not its trained state; experimental",
1708
+ "score": "alone: own stage text 0.721 -> 0.350 bpb; web text +0.0010 bpb",
1709
+ "measured": {
1710
+ "own_stage_bpb_arms_off": 0.7212,
1711
+ "own_stage_bpb_group_on": 0.1113,
1712
+ "own_stage_bpb_this_arm_alone": 0.3503,
1713
+ "share_of_the_group_on_own_stage": -0.1226,
1714
+ "share_of_the_group_on_own_stage_se": 0.0012,
1715
+ "share_of_the_group_on_web_text": -0.00042,
1716
+ "alone_on_web_text": 0.00105,
1717
+ "share_of_the_group_on_partner_stages": {
1718
+ "s1_perspective": -0.0062,
1719
+ "s3_rules": -0.1082,
1720
+ "s4_arith": -0.1052,
1721
+ "s5_causal": -0.0017,
1722
+ "s6_tryfail": -0.0034,
1723
+ "s7_mixed": -0.0008,
1724
+ "s8_register": -0.0013,
1725
+ "s9_caption": -0.0009
1726
+ },
1727
+ "alone_on_partner_stages": {
1728
+ "s1_perspective": -0.0273,
1729
+ "s3_rules": -0.3108,
1730
+ "s4_arith": -0.3537,
1731
+ "s5_causal": -0.0015,
1732
+ "s6_tryfail": 0.0004,
1733
+ "s7_mixed": -0.0,
1734
+ "s8_register": -0.0215,
1735
+ "s9_caption": 0.001
1736
+ },
1737
+ "steps_trained": 2400,
1738
+ "stage_items_own_form_accuracy": {
1739
+ "arms_off": 0.298,
1740
+ "group_on": 0.964,
1741
+ "this_arm_alone": 0.964
1742
+ },
1743
+ "stage_items_new_form_accuracy": {
1744
+ "arms_off": 0.678,
1745
+ "group_on": 0.703,
1746
+ "this_arm_alone": 0.72
1747
+ },
1748
+ "second_seed": {
1749
+ "own_stage_bpb_arms_off": 0.7212,
1750
+ "own_stage_bpb_group_on": 0.1218,
1751
+ "own_stage_bpb_this_arm_alone": 0.3603,
1752
+ "share_of_the_group_on_own_stage": -0.123,
1753
+ "share_of_the_group_on_own_stage_se": 0.0013,
1754
+ "share_of_the_group_on_web_text": -0.00084,
1755
+ "alone_on_web_text": 0.00086,
1756
+ "share_of_the_group_on_partner_stages": {
1757
+ "s1_perspective": -0.0046,
1758
+ "s3_rules": -0.1022,
1759
+ "s4_arith": -0.0974,
1760
+ "s5_causal": -0.0011,
1761
+ "s6_tryfail": -0.0024,
1762
+ "s7_mixed": -0.0002,
1763
+ "s8_register": -0.0007,
1764
+ "s9_caption": -0.001
1765
+ },
1766
+ "alone_on_partner_stages": {
1767
+ "s1_perspective": -0.03,
1768
+ "s3_rules": -0.3138,
1769
+ "s4_arith": -0.3577,
1770
+ "s5_causal": -0.0127,
1771
+ "s6_tryfail": -0.0062,
1772
+ "s7_mixed": 0.0002,
1773
+ "s8_register": -0.0193,
1774
+ "s9_caption": 0.0009
1775
+ },
1776
+ "steps_trained": 2400,
1777
+ "stage_items_own_form_accuracy": {
1778
+ "arms_off": 0.298,
1779
+ "group_on": 0.964,
1780
+ "this_arm_alone": 0.905
1781
+ },
1782
+ "stage_items_new_form_accuracy": {
1783
+ "arms_off": 0.678,
1784
+ "group_on": 0.763,
1785
+ "this_arm_alone": 0.703
1786
+ }
1787
+ }
1788
+ },
1789
+ "examples": [
1790
+ "A whale is a kind of mammal. Most mammals can swim. A whale is not a beetle; they differ in kind."
1791
+ ],
1792
+ "experimental": true
1793
+ },
1794
+ {
1795
+ "id": "stages-1-8-caption/s3_rules",
1796
+ "name": "stages-1-8-caption/s3_rules",
1797
+ "group": "stages-1-8-caption",
1798
+ "base_model_id": "alephllm/mini-beatrix-3@step245674",
1799
+ "step": 245674,
1800
+ "params": 13692960,
1801
+ "sites": 32,
1802
+ "weights": [
1803
+ "arms/stages-1-8-caption/s3_rules.safetensors"
1804
+ ],
1805
+ "mount": {
1806
+ "type": "anchor"
1807
+ },
1808
+ "precision": "fp32",
1809
+ "template": {
1810
+ "frame": "raw",
1811
+ "template": "raw stage text, continued"
1812
+ },
1813
+ "decode": {
1814
+ "temperature": 0.0,
1815
+ "top_p": 1.0,
1816
+ "max_new": 120,
1817
+ "precision": "fp32"
1818
+ },
1819
+ "source": {
1820
+ "file": "gCA_close9_s3_rules.safetensors",
1821
+ "sha256": "5962884db9a8724e7e97d62b22a8cb1c80e65a5dd83ea69e0a15e1348094c97a"
1822
+ },
1823
+ "recipe": "trained as an always-on group on the frozen final core by continuing the eight-arm group's run. The eight arms of stages 1 to 8 were loaded as trained (see the group stages-1-8) and held fixed and on. Then a ninth arm, for image captions, was attached over them and trained for 4,000 steps in their presence on the caption pack: five sources (two sets of photographs with structured captions, CC12M alt text with generated descriptions, COCO captions, danbooru tags), each image one document in a labelled frame (tags, the source's caption, short, attributes, scene, photo, structure, medium, long, rating: one label per line), documents drawn by source at fixed shares (CC12M 0.41, the two photograph sets 0.16 each, danbooru 0.16, COCO 0.11). Each step gives the training arm 2 x 4096 bytes of caption text with every attached arm on, and a quiet term at weight 2 on 2 rows that are not its text (one of web text, one of the stage text of one of the eight, drawn from every stage but the mixed one): the KL from the model with that arm switched off to the model with every arm on. Pure Adam, lr 0.001",
1824
+ "title": "Stage 3: rule chains",
1825
+ "behavior": "if-then rules over made-up words, followed step by step to what follows",
1826
+ "family": "stage",
1827
+ "status": "a member of the group stages-1-8-caption; trained always on with its partners, so alone is not its trained state; experimental",
1828
+ "score": "alone: own stage text 0.930 -> 0.824 bpb; web text +0.0005 bpb",
1829
+ "measured": {
1830
+ "own_stage_bpb_arms_off": 0.9296,
1831
+ "own_stage_bpb_group_on": 0.2245,
1832
+ "own_stage_bpb_this_arm_alone": 0.8236,
1833
+ "share_of_the_group_on_own_stage": -0.0903,
1834
+ "share_of_the_group_on_own_stage_se": 0.0014,
1835
+ "share_of_the_group_on_web_text": 0.00016,
1836
+ "alone_on_web_text": 0.00053,
1837
+ "share_of_the_group_on_partner_stages": {
1838
+ "s1_perspective": -0.0,
1839
+ "s2_concept": 0.0006,
1840
+ "s4_arith": -0.0002,
1841
+ "s5_causal": -0.0001,
1842
+ "s6_tryfail": -0.0003,
1843
+ "s7_mixed": 0.0,
1844
+ "s8_register": -0.0002,
1845
+ "s9_caption": -0.0001
1846
+ },
1847
+ "alone_on_partner_stages": {
1848
+ "s1_perspective": -0.0003,
1849
+ "s2_concept": 0.0031,
1850
+ "s4_arith": 0.0001,
1851
+ "s5_causal": 0.0071,
1852
+ "s6_tryfail": 0.001,
1853
+ "s7_mixed": -0.0005,
1854
+ "s8_register": 0.0002,
1855
+ "s9_caption": 0.0001
1856
+ },
1857
+ "steps_trained": 800,
1858
+ "stage_items_own_form_accuracy": {
1859
+ "arms_off": 0.155,
1860
+ "group_on": 0.835,
1861
+ "this_arm_alone": 0.32
1862
+ },
1863
+ "stage_items_new_form_accuracy": {
1864
+ "arms_off": 0.07,
1865
+ "group_on": 0.29,
1866
+ "this_arm_alone": 0.08
1867
+ },
1868
+ "second_seed": {
1869
+ "own_stage_bpb_arms_off": 0.9296,
1870
+ "own_stage_bpb_group_on": 0.2225,
1871
+ "own_stage_bpb_this_arm_alone": 0.8702,
1872
+ "share_of_the_group_on_own_stage": -0.0924,
1873
+ "share_of_the_group_on_own_stage_se": 0.0016,
1874
+ "share_of_the_group_on_web_text": 0.00026,
1875
+ "alone_on_web_text": 0.00034,
1876
+ "share_of_the_group_on_partner_stages": {
1877
+ "s1_perspective": -0.0001,
1878
+ "s2_concept": -0.0006,
1879
+ "s4_arith": -0.0005,
1880
+ "s5_causal": -0.0003,
1881
+ "s6_tryfail": -0.0005,
1882
+ "s7_mixed": -0.0001,
1883
+ "s8_register": -0.0001,
1884
+ "s9_caption": 0.0004
1885
+ },
1886
+ "alone_on_partner_stages": {
1887
+ "s1_perspective": -0.0003,
1888
+ "s2_concept": 0.0016,
1889
+ "s4_arith": -0.0047,
1890
+ "s5_causal": 0.006,
1891
+ "s6_tryfail": -0.0011,
1892
+ "s7_mixed": -0.0007,
1893
+ "s8_register": -0.0044,
1894
+ "s9_caption": 0.0002
1895
+ },
1896
+ "steps_trained": 800,
1897
+ "stage_items_own_form_accuracy": {
1898
+ "arms_off": 0.155,
1899
+ "group_on": 0.92,
1900
+ "this_arm_alone": 0.335
1901
+ },
1902
+ "stage_items_new_form_accuracy": {
1903
+ "arms_off": 0.07,
1904
+ "group_on": 0.31,
1905
+ "this_arm_alone": 0.1
1906
+ }
1907
+ }
1908
+ },
1909
+ "examples": [
1910
+ "If someone is hoream, then they are hortil. If someone is hortil, then they are babig. If someone is babig, then they are banng. Vin is hoream. So Vin is hortil. So Vin is babig. So Vin is banng."
1911
+ ],
1912
+ "experimental": true
1913
+ },
1914
+ {
1915
+ "id": "stages-1-8-caption/s4_arith",
1916
+ "name": "stages-1-8-caption/s4_arith",
1917
+ "group": "stages-1-8-caption",
1918
+ "base_model_id": "alephllm/mini-beatrix-3@step245674",
1919
+ "step": 245674,
1920
+ "params": 13692960,
1921
+ "sites": 32,
1922
+ "weights": [
1923
+ "arms/stages-1-8-caption/s4_arith.safetensors"
1924
+ ],
1925
+ "mount": {
1926
+ "type": "anchor"
1927
+ },
1928
+ "precision": "fp32",
1929
+ "template": {
1930
+ "frame": "raw",
1931
+ "template": "raw stage text, continued"
1932
+ },
1933
+ "decode": {
1934
+ "temperature": 0.0,
1935
+ "top_p": 1.0,
1936
+ "max_new": 120,
1937
+ "precision": "fp32"
1938
+ },
1939
+ "source": {
1940
+ "file": "gCA_close9_s4_arith.safetensors",
1941
+ "sha256": "622bfe16860d9b995214cb630c70fa76b2e44b3474438c6de2577822f441777c"
1942
+ },
1943
+ "recipe": "trained as an always-on group on the frozen final core by continuing the eight-arm group's run. The eight arms of stages 1 to 8 were loaded as trained (see the group stages-1-8) and held fixed and on. Then a ninth arm, for image captions, was attached over them and trained for 4,000 steps in their presence on the caption pack: five sources (two sets of photographs with structured captions, CC12M alt text with generated descriptions, COCO captions, danbooru tags), each image one document in a labelled frame (tags, the source's caption, short, attributes, scene, photo, structure, medium, long, rating: one label per line), documents drawn by source at fixed shares (CC12M 0.41, the two photograph sets 0.16 each, danbooru 0.16, COCO 0.11). Each step gives the training arm 2 x 4096 bytes of caption text with every attached arm on, and a quiet term at weight 2 on 2 rows that are not its text (one of web text, one of the stage text of one of the eight, drawn from every stage but the mixed one): the KL from the model with that arm switched off to the model with every arm on. Pure Adam, lr 0.001",
1944
+ "title": "Stage 4: arithmetic",
1945
+ "behavior": "small arithmetic worked out in text",
1946
+ "family": "stage",
1947
+ "status": "a member of the group stages-1-8-caption; trained always on with its partners, so alone is not its trained state; experimental",
1948
+ "score": "alone: own stage text 1.234 -> 1.127 bpb; web text +0.0001 bpb",
1949
+ "measured": {
1950
+ "own_stage_bpb_arms_off": 1.2339,
1951
+ "own_stage_bpb_group_on": 0.3287,
1952
+ "own_stage_bpb_this_arm_alone": 1.1267,
1953
+ "share_of_the_group_on_own_stage": -0.1454,
1954
+ "share_of_the_group_on_own_stage_se": 0.0023,
1955
+ "share_of_the_group_on_web_text": -4e-05,
1956
+ "alone_on_web_text": 0.00011,
1957
+ "share_of_the_group_on_partner_stages": {
1958
+ "s1_perspective": -0.0001,
1959
+ "s2_concept": -0.0006,
1960
+ "s3_rules": -0.0001,
1961
+ "s5_causal": -0.0002,
1962
+ "s6_tryfail": -0.0041,
1963
+ "s7_mixed": 0.0002,
1964
+ "s8_register": -0.0002,
1965
+ "s9_caption": 0.0004
1966
+ },
1967
+ "alone_on_partner_stages": {
1968
+ "s1_perspective": -0.0009,
1969
+ "s2_concept": -0.0004,
1970
+ "s3_rules": -0.0016,
1971
+ "s5_causal": -0.0033,
1972
+ "s6_tryfail": -0.0135,
1973
+ "s7_mixed": -0.0002,
1974
+ "s8_register": -0.0015,
1975
+ "s9_caption": -0.0004
1976
+ },
1977
+ "steps_trained": 800,
1978
+ "stage_items_own_form_accuracy": {
1979
+ "arms_off": 0.344,
1980
+ "group_on": 0.65,
1981
+ "this_arm_alone": 0.381
1982
+ },
1983
+ "stage_items_new_form_accuracy": {
1984
+ "arms_off": 0.242,
1985
+ "group_on": 0.254,
1986
+ "this_arm_alone": 0.246
1987
+ },
1988
+ "second_seed": {
1989
+ "own_stage_bpb_arms_off": 1.2339,
1990
+ "own_stage_bpb_group_on": 0.3272,
1991
+ "own_stage_bpb_this_arm_alone": 1.155,
1992
+ "share_of_the_group_on_own_stage": -0.1658,
1993
+ "share_of_the_group_on_own_stage_se": 0.0018,
1994
+ "share_of_the_group_on_web_text": 0.00014,
1995
+ "alone_on_web_text": 6e-05,
1996
+ "share_of_the_group_on_partner_stages": {
1997
+ "s1_perspective": -0.0,
1998
+ "s2_concept": -0.0002,
1999
+ "s3_rules": 0.0,
2000
+ "s5_causal": -0.0003,
2001
+ "s6_tryfail": -0.0044,
2002
+ "s7_mixed": 0.0001,
2003
+ "s8_register": -0.0001,
2004
+ "s9_caption": 0.0
2005
+ },
2006
+ "alone_on_partner_stages": {
2007
+ "s1_perspective": -0.0013,
2008
+ "s2_concept": -0.0015,
2009
+ "s3_rules": 0.0005,
2010
+ "s5_causal": -0.0011,
2011
+ "s6_tryfail": -0.0134,
2012
+ "s7_mixed": -0.0003,
2013
+ "s8_register": -0.0014,
2014
+ "s9_caption": -0.0005
2015
+ },
2016
+ "steps_trained": 800,
2017
+ "stage_items_own_form_accuracy": {
2018
+ "arms_off": 0.344,
2019
+ "group_on": 0.625,
2020
+ "this_arm_alone": 0.412
2021
+ },
2022
+ "stage_items_new_form_accuracy": {
2023
+ "arms_off": 0.242,
2024
+ "group_on": 0.263,
2025
+ "this_arm_alone": 0.237
2026
+ }
2027
+ }
2028
+ },
2029
+ "examples": [
2030
+ "Count by 3s: 1 3s make 3; 2 3s make 6; 3 3s make 9; 4 3s make 12; 5 3s make 15. So 3 times 5 is 15."
2031
+ ],
2032
+ "experimental": true
2033
+ },
2034
+ {
2035
+ "id": "stages-1-8-caption/s5_causal",
2036
+ "name": "stages-1-8-caption/s5_causal",
2037
+ "group": "stages-1-8-caption",
2038
+ "base_model_id": "alephllm/mini-beatrix-3@step245674",
2039
+ "step": 245674,
2040
+ "params": 13692960,
2041
+ "sites": 32,
2042
+ "weights": [
2043
+ "arms/stages-1-8-caption/s5_causal.safetensors"
2044
+ ],
2045
+ "mount": {
2046
+ "type": "anchor"
2047
+ },
2048
+ "precision": "fp32",
2049
+ "template": {
2050
+ "frame": "raw",
2051
+ "template": "raw stage text, continued"
2052
+ },
2053
+ "decode": {
2054
+ "temperature": 0.0,
2055
+ "top_p": 1.0,
2056
+ "max_new": 120,
2057
+ "precision": "fp32"
2058
+ },
2059
+ "source": {
2060
+ "file": "gCA_close9_s5_causal.safetensors",
2061
+ "sha256": "45617e30c3f1c6121b8fbfffb87f03253043a744cfed0c99ddc9f79006bd495c"
2062
+ },
2063
+ "recipe": "trained as an always-on group on the frozen final core by continuing the eight-arm group's run. The eight arms of stages 1 to 8 were loaded as trained (see the group stages-1-8) and held fixed and on. Then a ninth arm, for image captions, was attached over them and trained for 4,000 steps in their presence on the caption pack: five sources (two sets of photographs with structured captions, CC12M alt text with generated descriptions, COCO captions, danbooru tags), each image one document in a labelled frame (tags, the source's caption, short, attributes, scene, photo, structure, medium, long, rating: one label per line), documents drawn by source at fixed shares (CC12M 0.41, the two photograph sets 0.16 each, danbooru 0.16, COCO 0.11). Each step gives the training arm 2 x 4096 bytes of caption text with every attached arm on, and a quiet term at weight 2 on 2 rows that are not its text (one of web text, one of the stage text of one of the eight, drawn from every stage but the mixed one): the KL from the model with that arm switched off to the model with every arm on. Pure Adam, lr 0.001",
2064
+ "title": "Stage 5: cause and effect",
2065
+ "behavior": "short cause-and-effect text: what happened and why",
2066
+ "family": "stage",
2067
+ "status": "a member of the group stages-1-8-caption; trained always on with its partners, so alone is not its trained state; experimental",
2068
+ "score": "alone: own stage text 0.660 -> 0.091 bpb; web text +0.0003 bpb",
2069
+ "measured": {
2070
+ "own_stage_bpb_arms_off": 0.6604,
2071
+ "own_stage_bpb_group_on": 0.0434,
2072
+ "own_stage_bpb_this_arm_alone": 0.0911,
2073
+ "share_of_the_group_on_own_stage": -0.409,
2074
+ "share_of_the_group_on_own_stage_se": 0.0055,
2075
+ "share_of_the_group_on_web_text": 0.00036,
2076
+ "alone_on_web_text": 0.00028,
2077
+ "share_of_the_group_on_partner_stages": {
2078
+ "s1_perspective": 0.0006,
2079
+ "s2_concept": -0.0005,
2080
+ "s3_rules": 0.0004,
2081
+ "s4_arith": -0.0001,
2082
+ "s6_tryfail": -0.001,
2083
+ "s7_mixed": 0.0,
2084
+ "s8_register": -0.0006,
2085
+ "s9_caption": 0.0001
2086
+ },
2087
+ "alone_on_partner_stages": {
2088
+ "s1_perspective": -0.0183,
2089
+ "s2_concept": -0.0109,
2090
+ "s3_rules": -0.0122,
2091
+ "s4_arith": -0.0091,
2092
+ "s6_tryfail": -0.0065,
2093
+ "s7_mixed": -0.0011,
2094
+ "s8_register": -0.0064,
2095
+ "s9_caption": -0.0004
2096
+ },
2097
+ "steps_trained": 4000,
2098
+ "second_seed": {
2099
+ "own_stage_bpb_arms_off": 0.6604,
2100
+ "own_stage_bpb_group_on": 0.0431,
2101
+ "own_stage_bpb_this_arm_alone": 0.1418,
2102
+ "share_of_the_group_on_own_stage": -0.4206,
2103
+ "share_of_the_group_on_own_stage_se": 0.0042,
2104
+ "share_of_the_group_on_web_text": 2e-05,
2105
+ "alone_on_web_text": 0.0,
2106
+ "share_of_the_group_on_partner_stages": {
2107
+ "s1_perspective": 0.0001,
2108
+ "s2_concept": 0.0002,
2109
+ "s3_rules": 0.0004,
2110
+ "s4_arith": -0.0001,
2111
+ "s6_tryfail": -0.0004,
2112
+ "s7_mixed": 0.0,
2113
+ "s8_register": -0.0,
2114
+ "s9_caption": -0.0
2115
+ },
2116
+ "alone_on_partner_stages": {
2117
+ "s1_perspective": -0.0022,
2118
+ "s2_concept": -0.001,
2119
+ "s3_rules": -0.0018,
2120
+ "s4_arith": -0.0022,
2121
+ "s6_tryfail": -0.0023,
2122
+ "s7_mixed": -0.0001,
2123
+ "s8_register": -0.0009,
2124
+ "s9_caption": -0.0001
2125
+ },
2126
+ "steps_trained": 4000
2127
+ }
2128
+ },
2129
+ "examples": [
2130
+ "Noor stacked the cups too high. Because of that, the tower leaned. Because of that, the cups crashed down."
2131
+ ],
2132
+ "experimental": true
2133
+ },
2134
+ {
2135
+ "id": "stages-1-8-caption/s6_tryfail",
2136
+ "name": "stages-1-8-caption/s6_tryfail",
2137
+ "group": "stages-1-8-caption",
2138
+ "base_model_id": "alephllm/mini-beatrix-3@step245674",
2139
+ "step": 245674,
2140
+ "params": 13692960,
2141
+ "sites": 32,
2142
+ "weights": [
2143
+ "arms/stages-1-8-caption/s6_tryfail.safetensors"
2144
+ ],
2145
+ "mount": {
2146
+ "type": "anchor"
2147
+ },
2148
+ "precision": "fp32",
2149
+ "template": {
2150
+ "frame": "raw",
2151
+ "template": "raw stage text, continued"
2152
+ },
2153
+ "decode": {
2154
+ "temperature": 0.0,
2155
+ "top_p": 1.0,
2156
+ "max_new": 120,
2157
+ "precision": "fp32"
2158
+ },
2159
+ "source": {
2160
+ "file": "gCA_close9_s6_tryfail.safetensors",
2161
+ "sha256": "0a1e0004ecaf087bc83898f55ad28971af4f55e2eaf4a2d1093a9cc3a41db89d"
2162
+ },
2163
+ "recipe": "trained as an always-on group on the frozen final core by continuing the eight-arm group's run. The eight arms of stages 1 to 8 were loaded as trained (see the group stages-1-8) and held fixed and on. Then a ninth arm, for image captions, was attached over them and trained for 4,000 steps in their presence on the caption pack: five sources (two sets of photographs with structured captions, CC12M alt text with generated descriptions, COCO captions, danbooru tags), each image one document in a labelled frame (tags, the source's caption, short, attributes, scene, photo, structure, medium, long, rating: one label per line), documents drawn by source at fixed shares (CC12M 0.41, the two photograph sets 0.16 each, danbooru 0.16, COCO 0.11). Each step gives the training arm 2 x 4096 bytes of caption text with every attached arm on, and a quiet term at weight 2 on 2 rows that are not its text (one of web text, one of the stage text of one of the eight, drawn from every stage but the mixed one): the KL from the model with that arm switched off to the model with every arm on. Pure Adam, lr 0.001",
2164
+ "title": "Stage 6: try, fail, revise",
2165
+ "behavior": "an attempt, its failure and the revised attempt",
2166
+ "family": "stage",
2167
+ "status": "a member of the group stages-1-8-caption; trained always on with its partners, so alone is not its trained state; experimental",
2168
+ "score": "alone: own stage text 0.673 -> 0.164 bpb; web text +0.0001 bpb",
2169
+ "measured": {
2170
+ "own_stage_bpb_arms_off": 0.6728,
2171
+ "own_stage_bpb_group_on": 0.055,
2172
+ "own_stage_bpb_this_arm_alone": 0.1638,
2173
+ "share_of_the_group_on_own_stage": -0.4041,
2174
+ "share_of_the_group_on_own_stage_se": 0.0034,
2175
+ "share_of_the_group_on_web_text": -0.0002,
2176
+ "alone_on_web_text": 0.00013,
2177
+ "share_of_the_group_on_partner_stages": {
2178
+ "s1_perspective": -0.0001,
2179
+ "s2_concept": 0.0002,
2180
+ "s3_rules": 0.0001,
2181
+ "s4_arith": -0.0008,
2182
+ "s5_causal": -0.0001,
2183
+ "s7_mixed": -0.0001,
2184
+ "s8_register": -0.0005,
2185
+ "s9_caption": -0.0002
2186
+ },
2187
+ "alone_on_partner_stages": {
2188
+ "s1_perspective": -0.002,
2189
+ "s2_concept": 0.0012,
2190
+ "s3_rules": 0.0006,
2191
+ "s4_arith": 0.0036,
2192
+ "s5_causal": -0.0098,
2193
+ "s7_mixed": -0.0,
2194
+ "s8_register": -0.0003,
2195
+ "s9_caption": -0.0003
2196
+ },
2197
+ "steps_trained": 4000,
2198
+ "second_seed": {
2199
+ "own_stage_bpb_arms_off": 0.6728,
2200
+ "own_stage_bpb_group_on": 0.0547,
2201
+ "own_stage_bpb_this_arm_alone": 0.1437,
2202
+ "share_of_the_group_on_own_stage": -0.4121,
2203
+ "share_of_the_group_on_own_stage_se": 0.003,
2204
+ "share_of_the_group_on_web_text": 7e-05,
2205
+ "alone_on_web_text": 0.00011,
2206
+ "share_of_the_group_on_partner_stages": {
2207
+ "s1_perspective": 0.0001,
2208
+ "s2_concept": 0.0003,
2209
+ "s3_rules": 0.0001,
2210
+ "s4_arith": -0.0016,
2211
+ "s5_causal": 0.0001,
2212
+ "s7_mixed": 0.0,
2213
+ "s8_register": -0.0004,
2214
+ "s9_caption": -0.0
2215
+ },
2216
+ "alone_on_partner_stages": {
2217
+ "s1_perspective": 0.0029,
2218
+ "s2_concept": 0.0013,
2219
+ "s3_rules": -0.0003,
2220
+ "s4_arith": -0.001,
2221
+ "s5_causal": -0.0026,
2222
+ "s7_mixed": 0.0001,
2223
+ "s8_register": -0.0001,
2224
+ "s9_caption": -0.0002
2225
+ },
2226
+ "steps_trained": 4000
2227
+ }
2228
+ },
2229
+ "examples": [
2230
+ "Talia adds 34 and 23 and writes 65. Talia checks by counting back: 65 is too big. Talia tries again carefully: 34 + 23 = 57. The check works now, and Talia keeps the good method."
2231
+ ],
2232
+ "experimental": true
2233
+ },
2234
+ {
2235
+ "id": "stages-1-8-caption/s7_mixed",
2236
+ "name": "stages-1-8-caption/s7_mixed",
2237
+ "group": "stages-1-8-caption",
2238
+ "base_model_id": "alephllm/mini-beatrix-3@step245674",
2239
+ "step": 245674,
2240
+ "params": 13692960,
2241
+ "sites": 32,
2242
+ "weights": [
2243
+ "arms/stages-1-8-caption/s7_mixed.safetensors"
2244
+ ],
2245
+ "mount": {
2246
+ "type": "anchor"
2247
+ },
2248
+ "precision": "fp32",
2249
+ "template": {
2250
+ "frame": "raw",
2251
+ "template": "raw stage text, continued"
2252
+ },
2253
+ "decode": {
2254
+ "temperature": 0.0,
2255
+ "top_p": 1.0,
2256
+ "max_new": 120,
2257
+ "precision": "fp32"
2258
+ },
2259
+ "source": {
2260
+ "file": "gCA_close9_s7_mixed.safetensors",
2261
+ "sha256": "c7e41078c9528d9a2c76dbec7d399a64b7b50dfb33672924cdcca0cd5f06c70f"
2262
+ },
2263
+ "recipe": "trained as an always-on group on the frozen final core by continuing the eight-arm group's run. The eight arms of stages 1 to 8 were loaded as trained (see the group stages-1-8) and held fixed and on. Then a ninth arm, for image captions, was attached over them and trained for 4,000 steps in their presence on the caption pack: five sources (two sets of photographs with structured captions, CC12M alt text with generated descriptions, COCO captions, danbooru tags), each image one document in a labelled frame (tags, the source's caption, short, attributes, scene, photo, structure, medium, long, rating: one label per line), documents drawn by source at fixed shares (CC12M 0.41, the two photograph sets 0.16 each, danbooru 0.16, COCO 0.11). Each step gives the training arm 2 x 4096 bytes of caption text with every attached arm on, and a quiet term at weight 2 on 2 rows that are not its text (one of web text, one of the stage text of one of the eight, drawn from every stage but the mixed one): the KL from the model with that arm switched off to the model with every arm on. Pure Adam, lr 0.001",
2264
+ "title": "Stage 7: mixed",
2265
+ "behavior": "the mixed stage's diet: the earlier stages' text beside ordinary prose",
2266
+ "family": "stage",
2267
+ "status": "a member of the group stages-1-8-caption; trained always on with its partners, so alone is not its trained state; experimental",
2268
+ "score": "alone: own stage text 0.783 -> 0.724 bpb; web text -0.0002 bpb",
2269
+ "measured": {
2270
+ "own_stage_bpb_arms_off": 0.783,
2271
+ "own_stage_bpb_group_on": 0.7218,
2272
+ "own_stage_bpb_this_arm_alone": 0.7242,
2273
+ "share_of_the_group_on_own_stage": -0.0559,
2274
+ "share_of_the_group_on_own_stage_se": 0.0322,
2275
+ "share_of_the_group_on_web_text": -0.00084,
2276
+ "alone_on_web_text": -0.00019,
2277
+ "share_of_the_group_on_partner_stages": {
2278
+ "s1_perspective": -0.0003,
2279
+ "s2_concept": -0.0005,
2280
+ "s3_rules": -0.0009,
2281
+ "s4_arith": -0.01,
2282
+ "s5_causal": -0.0001,
2283
+ "s6_tryfail": -0.0002,
2284
+ "s8_register": -0.0001,
2285
+ "s9_caption": -0.0
2286
+ },
2287
+ "alone_on_partner_stages": {
2288
+ "s1_perspective": -0.0568,
2289
+ "s2_concept": -0.0058,
2290
+ "s3_rules": -0.0617,
2291
+ "s4_arith": -0.0385,
2292
+ "s5_causal": -0.1028,
2293
+ "s6_tryfail": -0.0802,
2294
+ "s8_register": -0.008,
2295
+ "s9_caption": -0.0008
2296
+ },
2297
+ "steps_trained": 4000,
2298
+ "second_seed": {
2299
+ "own_stage_bpb_arms_off": 0.783,
2300
+ "own_stage_bpb_group_on": 0.7215,
2301
+ "own_stage_bpb_this_arm_alone": 0.7254,
2302
+ "share_of_the_group_on_own_stage": -0.0562,
2303
+ "share_of_the_group_on_own_stage_se": 0.0325,
2304
+ "share_of_the_group_on_web_text": -0.00088,
2305
+ "alone_on_web_text": -0.00032,
2306
+ "share_of_the_group_on_partner_stages": {
2307
+ "s1_perspective": 0.0001,
2308
+ "s2_concept": -0.0002,
2309
+ "s3_rules": -0.0008,
2310
+ "s4_arith": -0.0102,
2311
+ "s5_causal": -0.0,
2312
+ "s6_tryfail": -0.0003,
2313
+ "s8_register": -0.0001,
2314
+ "s9_caption": -0.0004
2315
+ },
2316
+ "alone_on_partner_stages": {
2317
+ "s1_perspective": -0.0237,
2318
+ "s2_concept": -0.0022,
2319
+ "s3_rules": -0.0529,
2320
+ "s4_arith": -0.0414,
2321
+ "s5_causal": -0.0976,
2322
+ "s6_tryfail": -0.0774,
2323
+ "s8_register": -0.0061,
2324
+ "s9_caption": -0.0003
2325
+ },
2326
+ "steps_trained": 4000
2327
+ }
2328
+ },
2329
+ "examples": [
2330
+ "It has been compared to a chant, a rhythmic divine beauty, a melody, an aria, a toccata, an edification, an exaltation. As poetry is for the tongue, calligraphy is to the page. The authors of The Splendor of Islamic Calligraphy put it best when they said: “Calligraphy is the plainsong of the divine”.\nCalligraphy is the art of the linear graphic, but it is more than that. In Isl"
2331
+ ],
2332
+ "experimental": true
2333
+ },
2334
+ {
2335
+ "id": "stages-1-8-caption/s8_register",
2336
+ "name": "stages-1-8-caption/s8_register",
2337
+ "group": "stages-1-8-caption",
2338
+ "base_model_id": "alephllm/mini-beatrix-3@step245674",
2339
+ "step": 245674,
2340
+ "params": 13692960,
2341
+ "sites": 32,
2342
+ "weights": [
2343
+ "arms/stages-1-8-caption/s8_register.safetensors"
2344
+ ],
2345
+ "mount": {
2346
+ "type": "anchor"
2347
+ },
2348
+ "precision": "fp32",
2349
+ "template": {
2350
+ "frame": "raw",
2351
+ "template": "raw stage text, continued"
2352
+ },
2353
+ "decode": {
2354
+ "temperature": 0.0,
2355
+ "top_p": 1.0,
2356
+ "max_new": 120,
2357
+ "precision": "fp32"
2358
+ },
2359
+ "source": {
2360
+ "file": "gCA_close9_s8_register.safetensors",
2361
+ "sha256": "6f034611c2bd0319fbb3eb6a6848a86301323311e2cd5d9dec1cf89a2559ebfc"
2362
+ },
2363
+ "recipe": "trained as an always-on group on the frozen final core by continuing the eight-arm group's run. The eight arms of stages 1 to 8 were loaded as trained (see the group stages-1-8) and held fixed and on. Then a ninth arm, for image captions, was attached over them and trained for 4,000 steps in their presence on the caption pack: five sources (two sets of photographs with structured captions, CC12M alt text with generated descriptions, COCO captions, danbooru tags), each image one document in a labelled frame (tags, the source's caption, short, attributes, scene, photo, structure, medium, long, rating: one label per line), documents drawn by source at fixed shares (CC12M 0.41, the two photograph sets 0.16 each, danbooru 0.16, COCO 0.11). Each step gives the training arm 2 x 4096 bytes of caption text with every attached arm on, and a quiet term at weight 2 on 2 rows that are not its text (one of web text, one of the stage text of one of the eight, drawn from every stage but the mixed one): the KL from the model with that arm switched off to the model with every arm on. Pure Adam, lr 0.001",
2364
+ "title": "Stage 8: register",
2365
+ "behavior": "the same content in different registers of speech",
2366
+ "family": "stage",
2367
+ "status": "a member of the group stages-1-8-caption; trained always on with its partners, so alone is not its trained state; experimental",
2368
+ "score": "alone: own stage text 0.491 -> 0.068 bpb; web text +0.0005 bpb",
2369
+ "measured": {
2370
+ "own_stage_bpb_arms_off": 0.4907,
2371
+ "own_stage_bpb_group_on": 0.0217,
2372
+ "own_stage_bpb_this_arm_alone": 0.0678,
2373
+ "share_of_the_group_on_own_stage": -0.3662,
2374
+ "share_of_the_group_on_own_stage_se": 0.0067,
2375
+ "share_of_the_group_on_web_text": 0.00029,
2376
+ "alone_on_web_text": 0.00048,
2377
+ "share_of_the_group_on_partner_stages": {
2378
+ "s1_perspective": 0.0005,
2379
+ "s2_concept": -0.0009,
2380
+ "s3_rules": 0.0005,
2381
+ "s4_arith": 0.0001,
2382
+ "s5_causal": 0.0001,
2383
+ "s6_tryfail": 0.0001,
2384
+ "s7_mixed": 0.0007,
2385
+ "s9_caption": 0.0002
2386
+ },
2387
+ "alone_on_partner_stages": {
2388
+ "s1_perspective": -0.0172,
2389
+ "s2_concept": -0.0079,
2390
+ "s3_rules": -0.0048,
2391
+ "s4_arith": -0.0065,
2392
+ "s5_causal": -0.0189,
2393
+ "s6_tryfail": -0.022,
2394
+ "s7_mixed": -0.0002,
2395
+ "s9_caption": -0.0009
2396
+ },
2397
+ "steps_trained": 4000,
2398
+ "second_seed": {
2399
+ "own_stage_bpb_arms_off": 0.4907,
2400
+ "own_stage_bpb_group_on": 0.0213,
2401
+ "own_stage_bpb_this_arm_alone": 0.0516,
2402
+ "share_of_the_group_on_own_stage": -0.3671,
2403
+ "share_of_the_group_on_own_stage_se": 0.0071,
2404
+ "share_of_the_group_on_web_text": -0.00019,
2405
+ "alone_on_web_text": 5e-05,
2406
+ "share_of_the_group_on_partner_stages": {
2407
+ "s1_perspective": 0.0002,
2408
+ "s2_concept": 0.0002,
2409
+ "s3_rules": 0.0004,
2410
+ "s4_arith": -0.0001,
2411
+ "s5_causal": 0.0,
2412
+ "s6_tryfail": 0.0,
2413
+ "s7_mixed": 0.0002,
2414
+ "s9_caption": -0.0005
2415
+ },
2416
+ "alone_on_partner_stages": {
2417
+ "s1_perspective": -0.0045,
2418
+ "s2_concept": -0.005,
2419
+ "s3_rules": -0.0063,
2420
+ "s4_arith": -0.0064,
2421
+ "s5_causal": -0.0139,
2422
+ "s6_tryfail": -0.0187,
2423
+ "s7_mixed": 0.0001,
2424
+ "s9_caption": 0.0
2425
+ },
2426
+ "steps_trained": 4000
2427
+ }
2428
+ },
2429
+ "examples": [
2430
+ "The word 'ancient' means from a very long time ago. Used in a sentence: a ancient is easy to point at once you know the word."
2431
+ ],
2432
+ "experimental": true
2433
+ },
2434
+ {
2435
+ "id": "stages-1-8-caption/s9_caption",
2436
+ "name": "stages-1-8-caption/s9_caption",
2437
+ "group": "stages-1-8-caption",
2438
+ "base_model_id": "alephllm/mini-beatrix-3@step245674",
2439
+ "step": 245674,
2440
+ "params": 13692960,
2441
+ "sites": 32,
2442
+ "weights": [
2443
+ "arms/stages-1-8-caption/s9_caption.safetensors"
2444
+ ],
2445
+ "mount": {
2446
+ "type": "anchor"
2447
+ },
2448
+ "precision": "fp32",
2449
+ "template": {
2450
+ "frame": "raw",
2451
+ "template": "raw stage text, continued"
2452
+ },
2453
+ "decode": {
2454
+ "temperature": 0.0,
2455
+ "top_p": 1.0,
2456
+ "max_new": 120,
2457
+ "precision": "fp32"
2458
+ },
2459
+ "source": {
2460
+ "file": "gCA_close9_s9_caption.safetensors",
2461
+ "sha256": "2e477395a4c1c1775e119f1039c48efa290748904ea12b8b20fc10d30c20f278"
2462
+ },
2463
+ "recipe": "trained as an always-on group on the frozen final core by continuing the eight-arm group's run. The eight arms of stages 1 to 8 were loaded as trained (see the group stages-1-8) and held fixed and on. Then a ninth arm, for image captions, was attached over them and trained for 4,000 steps in their presence on the caption pack: five sources (two sets of photographs with structured captions, CC12M alt text with generated descriptions, COCO captions, danbooru tags), each image one document in a labelled frame (tags, the source's caption, short, attributes, scene, photo, structure, medium, long, rating: one label per line), documents drawn by source at fixed shares (CC12M 0.41, the two photograph sets 0.16 each, danbooru 0.16, COCO 0.11). Each step gives the training arm 2 x 4096 bytes of caption text with every attached arm on, and a quiet term at weight 2 on 2 rows that are not its text (one of web text, one of the stage text of one of the eight, drawn from every stage but the mixed one): the KL from the model with that arm switched off to the model with every arm on. Pure Adam, lr 0.001",
2464
+ "title": "Captions: the ninth arm (experimental)",
2465
+ "behavior": "image captions in one labelled frame (tags, the source's caption, short, attributes, scene, photo, structure, medium, long, rating: one label per line): photographs with structured captions, CC12M alt text and generated descriptions, COCO captions, danbooru tags",
2466
+ "family": "caption",
2467
+ "status": "a member of the group stages-1-8-caption; trained always on with its partners, so alone is not its trained state; experimental",
2468
+ "score": "alone: own stage text 1.265 -> 0.785 bpb; web text +0.0015 bpb",
2469
+ "measured": {
2470
+ "own_stage_bpb_arms_off": 1.2648,
2471
+ "own_stage_bpb_group_on": 0.7829,
2472
+ "own_stage_bpb_this_arm_alone": 0.7851,
2473
+ "share_of_the_group_on_own_stage": -0.481,
2474
+ "share_of_the_group_on_own_stage_se": 0.046,
2475
+ "share_of_the_group_on_web_text": 0.00076,
2476
+ "alone_on_web_text": 0.00152,
2477
+ "share_of_the_group_on_partner_stages": {
2478
+ "s1_perspective": 0.0004,
2479
+ "s2_concept": 0.0008,
2480
+ "s3_rules": 0.0008,
2481
+ "s4_arith": 0.0008,
2482
+ "s5_causal": 0.0004,
2483
+ "s6_tryfail": 0.0002,
2484
+ "s7_mixed": 0.0018,
2485
+ "s8_register": 0.0005
2486
+ },
2487
+ "alone_on_partner_stages": {
2488
+ "s1_perspective": -0.0018,
2489
+ "s2_concept": 0.0018,
2490
+ "s3_rules": -0.0049,
2491
+ "s4_arith": -0.0039,
2492
+ "s5_causal": -0.0007,
2493
+ "s6_tryfail": -0.0077,
2494
+ "s7_mixed": 0.0027,
2495
+ "s8_register": 0.0286
2496
+ },
2497
+ "steps_trained": 4000,
2498
+ "second_seed": {
2499
+ "own_stage_bpb_arms_off": 1.2648,
2500
+ "own_stage_bpb_group_on": 0.7795,
2501
+ "own_stage_bpb_this_arm_alone": 0.7819,
2502
+ "share_of_the_group_on_own_stage": -0.4856,
2503
+ "share_of_the_group_on_own_stage_se": 0.0472,
2504
+ "share_of_the_group_on_web_text": 0.00066,
2505
+ "alone_on_web_text": 0.00137,
2506
+ "share_of_the_group_on_partner_stages": {
2507
+ "s1_perspective": 0.0004,
2508
+ "s2_concept": 0.0006,
2509
+ "s3_rules": 0.0003,
2510
+ "s4_arith": 0.0006,
2511
+ "s5_causal": 0.0003,
2512
+ "s6_tryfail": 0.0004,
2513
+ "s7_mixed": 0.0018,
2514
+ "s8_register": 0.0005
2515
+ },
2516
+ "alone_on_partner_stages": {
2517
+ "s1_perspective": -0.0009,
2518
+ "s2_concept": 0.005,
2519
+ "s3_rules": 0.0013,
2520
+ "s4_arith": 0.0026,
2521
+ "s5_causal": 0.0016,
2522
+ "s6_tryfail": -0.0021,
2523
+ "s7_mixed": 0.0016,
2524
+ "s8_register": 0.0333
2525
+ },
2526
+ "steps_trained": 4000
2527
+ }
2528
+ },
2529
+ "examples": [
2530
+ "tags: 1girl, jewelry, solo, dress, bicycle, realistic, asian, smile, necklace, earrings, black hair, white dress, blurry, blurry background, grin, potted plant, long hair, sleeveless, black eyes, indoors, teeth\nshort: A smiling young woman with dark hair stands indoors holding the handlebars of a vintage-style green bicycle, wearing a light-colored sleeveless dress.\nhair: long wavy"
2531
+ ],
2532
+ "experimental": true
2533
+ },
2534
+ {
2535
+ "id": "stages-1-8-caption",
2536
+ "name": "stages-1-8-caption",
2537
+ "base_model_id": "alephllm/mini-beatrix-3@step245674",
2538
+ "step": 245674,
2539
+ "params": 123236640,
2540
+ "sites": 32,
2541
+ "weights": [
2542
+ "arms/stages-1-8-caption/s1_perspective.safetensors",
2543
+ "arms/stages-1-8-caption/s2_concept.safetensors",
2544
+ "arms/stages-1-8-caption/s3_rules.safetensors",
2545
+ "arms/stages-1-8-caption/s4_arith.safetensors",
2546
+ "arms/stages-1-8-caption/s5_causal.safetensors",
2547
+ "arms/stages-1-8-caption/s6_tryfail.safetensors",
2548
+ "arms/stages-1-8-caption/s7_mixed.safetensors",
2549
+ "arms/stages-1-8-caption/s8_register.safetensors",
2550
+ "arms/stages-1-8-caption/s9_caption.safetensors"
2551
+ ],
2552
+ "mount": {
2553
+ "type": "stack",
2554
+ "members": [
2555
+ "stages-1-8-caption/s1_perspective",
2556
+ "stages-1-8-caption/s2_concept",
2557
+ "stages-1-8-caption/s3_rules",
2558
+ "stages-1-8-caption/s4_arith",
2559
+ "stages-1-8-caption/s5_causal",
2560
+ "stages-1-8-caption/s6_tryfail",
2561
+ "stages-1-8-caption/s7_mixed",
2562
+ "stages-1-8-caption/s8_register",
2563
+ "stages-1-8-caption/s9_caption"
2564
+ ]
2565
+ },
2566
+ "precision": "fp32",
2567
+ "template": {
2568
+ "frame": "raw",
2569
+ "template": "raw text, continued"
2570
+ },
2571
+ "decode": {
2572
+ "temperature": 0.0,
2573
+ "top_p": 1.0,
2574
+ "max_new": 120,
2575
+ "precision": "fp32"
2576
+ },
2577
+ "recipe": "trained as an always-on group on the frozen final core by continuing the eight-arm group's run. The eight arms of stages 1 to 8 were loaded as trained (see the group stages-1-8) and held fixed and on. Then a ninth arm, for image captions, was attached over them and trained for 4,000 steps in their presence on the caption pack: five sources (two sets of photographs with structured captions, CC12M alt text with generated descriptions, COCO captions, danbooru tags), each image one document in a labelled frame (tags, the source's caption, short, attributes, scene, photo, structure, medium, long, rating: one label per line), documents drawn by source at fixed shares (CC12M 0.41, the two photograph sets 0.16 each, danbooru 0.16, COCO 0.11). Each step gives the training arm 2 x 4096 bytes of caption text with every attached arm on, and a quiet term at weight 2 on 2 rows that are not its text (one of web text, one of the stage text of one of the eight, drawn from every stage but the mixed one): the KL from the model with that arm switched off to the model with every arm on. Pure Adam, lr 0.001",
2578
+ "title": "Stages 1 to 8 with the caption arm (experimental)",
2579
+ "behavior": "the arms of stage 1: perspective, stage 2: concepts, stage 3: rule chains, stage 4: arithmetic, stage 5: cause and effect, stage 6: try, fail, revise, stage 7: mixed, stage 8: register, captions: the ninth arm (experimental), always on together, as they were trained",
2580
+ "family": "group",
2581
+ "status": "two seeds; experimental",
2582
+ "score": "stage text, arms off -> on: s1_perspective 0.554 -> 0.050; s2_concept 0.721 -> 0.111; s3_rules 0.930 -> 0.225; s4_arith 1.234 -> 0.329; s5_causal 0.660 -> 0.043; s6_tryfail 0.673 -> 0.055; s7_mixed 0.783 -> 0.722; s8_register 0.491 -> 0.022; s9_caption 1.265 -> 0.783; web text +0.0027 bpb",
2583
+ "measured": {
2584
+ "stage_text_bpb": {
2585
+ "s1_perspective": {
2586
+ "arms_off": 0.5538,
2587
+ "group_on": 0.0496,
2588
+ "difference": -0.5042,
2589
+ "difference_se": 0.0049
2590
+ },
2591
+ "s2_concept": {
2592
+ "arms_off": 0.7212,
2593
+ "group_on": 0.1113,
2594
+ "difference": -0.6099,
2595
+ "difference_se": 0.0039
2596
+ },
2597
+ "s3_rules": {
2598
+ "arms_off": 0.9296,
2599
+ "group_on": 0.2245,
2600
+ "difference": -0.7051,
2601
+ "difference_se": 0.0055
2602
+ },
2603
+ "s4_arith": {
2604
+ "arms_off": 1.2339,
2605
+ "group_on": 0.3287,
2606
+ "difference": -0.9052,
2607
+ "difference_se": 0.0062
2608
+ },
2609
+ "s5_causal": {
2610
+ "arms_off": 0.6604,
2611
+ "group_on": 0.0434,
2612
+ "difference": -0.617,
2613
+ "difference_se": 0.007
2614
+ },
2615
+ "s6_tryfail": {
2616
+ "arms_off": 0.6728,
2617
+ "group_on": 0.055,
2618
+ "difference": -0.6178,
2619
+ "difference_se": 0.0037
2620
+ },
2621
+ "s7_mixed": {
2622
+ "arms_off": 0.783,
2623
+ "group_on": 0.7218,
2624
+ "difference": -0.0612,
2625
+ "difference_se": 0.0392
2626
+ },
2627
+ "s8_register": {
2628
+ "arms_off": 0.4907,
2629
+ "group_on": 0.0217,
2630
+ "difference": -0.469,
2631
+ "difference_se": 0.0085
2632
+ },
2633
+ "s9_caption": {
2634
+ "arms_off": 1.2648,
2635
+ "group_on": 0.7829,
2636
+ "difference": -0.4819,
2637
+ "difference_se": 0.0451
2638
+ }
2639
+ },
2640
+ "web_text_difference": 0.00265,
2641
+ "web_text_difference_se": 0.00055,
2642
+ "steps_trained": {
2643
+ "s1_perspective": 3200,
2644
+ "s2_concept": 2400,
2645
+ "s3_rules": 800,
2646
+ "s4_arith": 800,
2647
+ "s5_causal": 4000,
2648
+ "s6_tryfail": 4000,
2649
+ "s7_mixed": 4000,
2650
+ "s8_register": 4000,
2651
+ "s9_caption": 4000
2652
+ },
2653
+ "probe_suite_mean": {
2654
+ "arms_off": 0.411,
2655
+ "group_on": 0.422
2656
+ },
2657
+ "second_seed": {
2658
+ "stage_text_bpb": {
2659
+ "s1_perspective": {
2660
+ "arms_off": 0.5538,
2661
+ "group_on": 0.0488,
2662
+ "difference": -0.5051,
2663
+ "difference_se": 0.0047
2664
+ },
2665
+ "s2_concept": {
2666
+ "arms_off": 0.7212,
2667
+ "group_on": 0.1218,
2668
+ "difference": -0.5993,
2669
+ "difference_se": 0.0037
2670
+ },
2671
+ "s3_rules": {
2672
+ "arms_off": 0.9296,
2673
+ "group_on": 0.2225,
2674
+ "difference": -0.7071,
2675
+ "difference_se": 0.0056
2676
+ },
2677
+ "s4_arith": {
2678
+ "arms_off": 1.2339,
2679
+ "group_on": 0.3272,
2680
+ "difference": -0.9067,
2681
+ "difference_se": 0.0062
2682
+ },
2683
+ "s5_causal": {
2684
+ "arms_off": 0.6604,
2685
+ "group_on": 0.0431,
2686
+ "difference": -0.6174,
2687
+ "difference_se": 0.007
2688
+ },
2689
+ "s6_tryfail": {
2690
+ "arms_off": 0.6728,
2691
+ "group_on": 0.0547,
2692
+ "difference": -0.6181,
2693
+ "difference_se": 0.0036
2694
+ },
2695
+ "s7_mixed": {
2696
+ "arms_off": 0.783,
2697
+ "group_on": 0.7215,
2698
+ "difference": -0.0615,
2699
+ "difference_se": 0.039
2700
+ },
2701
+ "s8_register": {
2702
+ "arms_off": 0.4907,
2703
+ "group_on": 0.0213,
2704
+ "difference": -0.4694,
2705
+ "difference_se": 0.0085
2706
+ },
2707
+ "s9_caption": {
2708
+ "arms_off": 1.2648,
2709
+ "group_on": 0.7795,
2710
+ "difference": -0.4853,
2711
+ "difference_se": 0.0464
2712
+ }
2713
+ },
2714
+ "web_text_difference": 0.00169,
2715
+ "web_text_difference_se": 0.00038,
2716
+ "steps_trained": {
2717
+ "s1_perspective": 3200,
2718
+ "s2_concept": 2400,
2719
+ "s3_rules": 800,
2720
+ "s4_arith": 800,
2721
+ "s5_causal": 4000,
2722
+ "s6_tryfail": 4000,
2723
+ "s7_mixed": 4000,
2724
+ "s8_register": 4000,
2725
+ "s9_caption": 4000
2726
+ },
2727
+ "probe_suite_mean": {
2728
+ "arms_off": 0.411,
2729
+ "group_on": 0.411
2730
+ }
2731
+ }
2732
+ },
2733
+ "examples": [],
2734
+ "experimental": true
2735
+ },
2736
  {
2737
  "id": "solo/s1_perspective",
2738
  "name": "solo/s1_perspective",
arms/stages-1-8-caption/s1_perspective.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:416e3023f2cd0926411050986bed562332b83580831f9bf88b07264ee7ec50eb
3
+ size 54819808
arms/stages-1-8-caption/s2_concept.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9f878e1c8942a9b96483b44fc6d7bd48bd6cb450971cb47cad44e23401126627
3
+ size 54819688
arms/stages-1-8-caption/s3_rules.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:822b252ae98c043006cdd3b858e8f5153919a1176a36c197dda53e830537d7a8
3
+ size 54819736
arms/stages-1-8-caption/s4_arith.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0671111590877ab8f63fb1f585d3ffa4b0c5ecc220ce39be1b81ae74ab987576
3
+ size 54819616
arms/stages-1-8-caption/s5_causal.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5b8dc5ba5905c1e6f2982cc498e3da5fde58d7bb5ef0f3a86b4267b17ebbd026
3
+ size 54819208
arms/stages-1-8-caption/s6_tryfail.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3b1600a7fa8c7857ee7be464d14274d79d63fc960da9d1752b5816436b0f1d78
3
+ size 54819280
arms/stages-1-8-caption/s7_mixed.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:da6a74f3e534778ce79436e271accd308e5b0ca05ec0f344e340fe607d4bce6f
3
+ size 54819528
arms/stages-1-8-caption/s8_register.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:60620f9df9228d1218e850a1d812ff5af2d593fcec2aefd2a8c137de1d2e6b36
3
+ size 54819224
arms/stages-1-8-caption/s9_caption.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c0f746f56d90df3fcd61ef49cb5179d62186c4ecaba651ebab6f60e2f671c4a4
3
+ size 54819720