jaivial commited on
Commit
b920472
·
verified ·
1 Parent(s): fab035a

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +17 -10
README.md CHANGED
@@ -22,7 +22,7 @@ v20 design: the expert capacity is unchanged (~507M) but it is reorganised from
22
  **16 private pools into ONE shared pool read by 3 layers**, following
23
  *Memory Layers at Scale* (arXiv:2412.09764).
24
 
25
- ## Headline result: 1.48x faster training at 16x the expert granularity
26
 
27
  Both runs used **the same GPU model (RTX 5060 Ti 16 GB)**, the same streamed
28
  FineWeb-Edu+Cosmopedia corpus, the same codon table and the same optimizer stack.
@@ -35,10 +35,18 @@ FineWeb-Edu+Cosmopedia corpus, the same codon table and the same optimizer stack
35
  | Expert params | 507.5M | 507.5M (identical) |
36
  | Controller params | 569.4M | 598.1M |
37
  | Gathers per token | 16 | **3** |
38
- | **Avg throughput** | **22,616 tok/s** | **33,451 tok/s** |
39
- | Median / max tok/s | 22,904 / 24,040 | 33,676 / 34,768 |
 
 
 
40
  | Dense DNA-2B reference | 45,700 tok/s | 45,700 tok/s |
41
 
 
 
 
 
 
42
  ![throughput](throughput_v20_vs_v21.png)
43
  ![summary](summary_v20_vs_v21.png)
44
  ![loss](loss_v20_vs_v21.png)
@@ -62,7 +70,7 @@ FineWeb-Edu+Cosmopedia corpus, the same codon table and the same optimizer stack
62
  recurrence, the dense FFNs and the PEER *router*, and leaves only the gather eager.
63
  - **The projected 44k tok/s was not reached.** CPU-proxy benchmarking overestimated the
64
  win; on the real GPU the recurrence and the eager gather region dominate. The honest
65
- measured number is **33,451 tok/s**.
66
 
67
  ## Architecture
68
 
@@ -77,8 +85,8 @@ optimizer : Muon (recurrence) + fused AdamW (dense) + sparse SGD (pool
77
 
78
  - Total experts: **495,616** per PEER layer (shared)
79
  - Active params/token: **598M** controller total, ~70M active
80
- - Tokens seen (this checkpoint): **101M**
81
- - Final valid CE: **4.802**
82
 
83
  ## Files
84
 
@@ -100,10 +108,9 @@ python train_peer_v21.py --batch 20 --seq 512 --nk 704 --topk 8 --pheads 1 \
100
  ```
101
 
102
  ## Honest limitations
103
- - This is a **systems/throughput** artifact. It is an early-training checkpoint, not a
104
- chat model, and has had no SFT.
105
- - v21 and v20 are at **different token counts**, so the CE-vs-tokens curve is the only
106
- fair quality comparison; the throughput comparison is the load-bearing result.
107
  - The codon memory table is inherited unchanged from DNA-DiskChat-2B.
108
 
109
  ## References
 
22
  **16 private pools into ONE shared pool read by 3 layers**, following
23
  *Memory Layers at Scale* (arXiv:2412.09764).
24
 
25
+ ## Headline result: 1.44x faster training at 16x the expert granularity
26
 
27
  Both runs used **the same GPU model (RTX 5060 Ti 16 GB)**, the same streamed
28
  FineWeb-Edu+Cosmopedia corpus, the same codon table and the same optimizer stack.
 
35
  | Expert params | 507.5M | 507.5M (identical) |
36
  | Controller params | 569.4M | 598.1M |
37
  | Gathers per token | 16 | **3** |
38
+ | **Avg throughput** | **22,566 tok/s** | **32,488 tok/s** |
39
+ | Median / max tok/s | 22,871 / 24,040 | 32,858 / 34,768 |
40
+ | Tokens trained | 1.553B | 1.162B |
41
+ | Final valid CE | 4.318 | **4.0317** |
42
+ | **Mean valid CE, matched 572-1145M tokens** | 4.1665 | **4.0213** |
43
  | Dense DNA-2B reference | 45,700 tok/s | 45,700 tok/s |
44
 
45
+ **v21 is 1.44x faster AND better per token.** Over the token window both runs
46
+ covered (572-1145M), v21's mean valid CE is
47
+ **0.145 lower** than v20's - so the
48
+ throughput win did not cost sample efficiency, it improved it.
49
+
50
  ![throughput](throughput_v20_vs_v21.png)
51
  ![summary](summary_v20_vs_v21.png)
52
  ![loss](loss_v20_vs_v21.png)
 
70
  recurrence, the dense FFNs and the PEER *router*, and leaves only the gather eager.
71
  - **The projected 44k tok/s was not reached.** CPU-proxy benchmarking overestimated the
72
  win; on the real GPU the recurrence and the eager gather region dominate. The honest
73
+ measured number is **32,488 tok/s**.
74
 
75
  ## Architecture
76
 
 
85
 
86
  - Total experts: **495,616** per PEER layer (shared)
87
  - Active params/token: **598M** controller total, ~70M active
88
+ - Tokens seen: **1.162B** (113,738 steps, run completed)
89
+ - Final valid CE: **3.7246** (last logged periodic checkpoint 4.0317)
90
 
91
  ## Files
92
 
 
108
  ```
109
 
110
  ## Honest limitations
111
+ - This is a **systems/throughput** artifact, not a chat model: no SFT has been done.
112
+ - The two runs stopped at different token counts (v20 1.553B, v21 1.162B), so the
113
+ quality claim is made only over the **matched token window** both runs covered.
 
114
  - The codon memory table is inherited unchanged from DNA-DiskChat-2B.
115
 
116
  ## References