Upload README.md with huggingface_hub
Browse files
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.
|
| 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,
|
| 39 |
-
| Median / max tok/s | 22,
|
|
|
|
|
|
|
|
|
|
| 40 |
| Dense DNA-2B reference | 45,700 tok/s | 45,700 tok/s |
|
| 41 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 42 |

|
| 43 |

|
| 44 |

|
|
@@ -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 **
|
| 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
|
| 81 |
-
- Final valid CE: **
|
| 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
|
| 104 |
-
|
| 105 |
-
|
| 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 |

|
| 51 |

|
| 52 |

|
|
|
|
| 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
|