wzebrowski commited on
Commit
5abaf7e
路
verified 路
1 Parent(s): 04bd9a5

Add Phase A and Phase B pretraining loss charts

Browse files
Files changed (1) hide show
  1. README.md +42 -1
README.md CHANGED
@@ -34,7 +34,7 @@ This repository describes one model, not a family of released models.
34
 
35
  **Selected checkpoint:** Stage B step 73,243 路 **Exposure:** 7,800,086,528 pretraining tokens 路 **Status:** documentation preview; weight release pending.
36
 
37
- [Model specification](#model-specification) 路 [Training method](#training-method) 路 [Evaluation](#evaluation) 路 [Limitations](#limitations-and-intended-use)
38
 
39
  ## Availability and Inference
40
 
@@ -228,6 +228,47 @@ This is **not a single unchanged 15B schedule from initialization**, and the
228
  The linked [pilot](configs/pilot.json) and [early continuation configuration](configs/training_recipe.json)
229
  are historical recipes; the later transitions are summarized here.
230
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
231
  ## Interim Results
232
 
233
  **Dated snapshot: September 18, 2026, through step 8,217. Not a live counter.**
 
34
 
35
  **Selected checkpoint:** Stage B step 73,243 路 **Exposure:** 7,800,086,528 pretraining tokens 路 **Status:** documentation preview; weight release pending.
36
 
37
+ [Model specification](#model-specification) 路 [Training method](#training-method) 路 [Pretraining curves](#pretraining-loss-curves) 路 [Evaluation](#evaluation) 路 [Limitations](#limitations-and-intended-use)
38
 
39
  ## Availability and Inference
40
 
 
228
  The linked [pilot](configs/pilot.json) and [early continuation configuration](configs/training_recipe.json)
229
  are historical recipes; the later transitions are summarized here.
230
 
231
+ ## Pretraining Loss Curves
232
+
233
+ These charts show recorded held-out next-token validation loss from the selected
234
+ pretraining lineage, without smoothing. **Blue is Phase A** (the initial mixture
235
+ and its continuations); **teal is Phase B** (the Stage B prose refinement).
236
+ Phase B starts at step **73,008**, after **7.775B tokens**. The star marks the
237
+ selected pretrained base at step **73,243**, after **7.800B tokens**. Later points
238
+ show continued pretraining beyond the selected checkpoint, not additional
239
+ training included in that checkpoint. Lower loss is better.
240
+
241
+ ### Full History and Late-training Zoom
242
+
243
+ ![OpenSML pretraining validation loss: full Phase A and Phase B history with a late-training zoom](pretraining-overview.svg)
244
+
245
+ The overview and zoom use the **same original held-out set** throughout
246
+ (262,144 target positions). The zoom makes the shorter Phase B visible; its
247
+ vertical scale is magnified. Aggregate validation weights remain fixed at
248
+ 55% web, 25% DCLM, 10% Wikipedia, and 10% Cosmopedia across both phases,
249
+ even though the Phase B training mixture changes. The selected checkpoint's
250
+ loss on this original set is **2.81855024**.
251
+
252
+ ### Phase B Detail
253
+
254
+ ![OpenSML Phase B validation loss on the expanded held-out set, with the selected checkpoint marked](pretraining-phase-b.svg)
255
+
256
+ This close-up uses the **expanded held-out set** (4,194,304 target positions),
257
+ introduced late in Phase A. Its absolute loss values are separate from the
258
+ original-set curves above and below. The selected checkpoint's expanded loss
259
+ is **2.76700753**. The tightly zoomed vertical scale shows small changes; the
260
+ later increase is less than 0.001 loss. Validation weights remain fixed to the
261
+ original reference mixture.
262
+
263
+ ### Validation Loss by Data Source
264
+
265
+ ![OpenSML original-set validation loss for web, DCLM, Wikipedia, and Cosmopedia across both pretraining phases](pretraining-sources.svg)
266
+
267
+ Each source is evaluated on the original held-out set throughout. Panels use
268
+ their own vertical scales; they show each source's trajectory rather than a
269
+ common-scale ranking of source difficulty. The dashed line marks the Phase B
270
+ transition and the star marks the selected base.
271
+
272
  ## Interim Results
273
 
274
  **Dated snapshot: September 18, 2026, through step 8,217. Not a live counter.**