report text: iso-FLOP answer + unroll-scaling caveat
Browse files- make_report.py +14 -4
make_report.py
CHANGED
|
@@ -211,6 +211,15 @@ def report(evals, rows):
|
|
| 211 |
" steps, while rec-4x4 (same params, 4x per-step FLOPs) reaches ~0.99+",
|
| 212 |
" by step 7500. This is a compute difference, not a parameter one:",
|
| 213 |
" serial block-applications appear to be the binding resource.",
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 214 |
"- **OOD (k>8) accuracy for fixed-4 is already ~10x chance** early in",
|
| 215 |
" training and slowly rising — consistent with a learned lookup for",
|
| 216 |
" small k failing to compose to large k (theory predicts constant-depth",
|
|
@@ -229,10 +238,11 @@ def report(evals, rows):
|
|
| 229 |
" resolved by pinning huggingface_hub==1.33.0.",
|
| 230 |
"",
|
| 231 |
"## What remains",
|
| 232 |
-
"-
|
| 233 |
-
"
|
| 234 |
-
" the
|
| 235 |
-
"-
|
|
|
|
| 236 |
"",
|
| 237 |
"## Reproduce",
|
| 238 |
"See README.md; run_all.py drives lrsweep/stage1/stage2; evaluate.py",
|
|
|
|
| 211 |
" steps, while rec-4x4 (same params, 4x per-step FLOPs) reaches ~0.99+",
|
| 212 |
" by step 7500. This is a compute difference, not a parameter one:",
|
| 213 |
" serial block-applications appear to be the binding resource.",
|
| 214 |
+
"- **Iso-FLOP test (fixed4-long)**: fixed-4 trained 4x longer (100k",
|
| 215 |
+
" steps, same total training FLOPs as rec-4x4) reaches 0.716±0.009",
|
| 216 |
+
" in-dist and 0.171±0.018 at k=64 — closing only part of the gap to",
|
| 217 |
+
" rec-4x4 (1.000 / 0.251±0.099). At this scale, extra training compute",
|
| 218 |
+
" does not substitute for recurrent depth.",
|
| 219 |
+
"- **Inference-time unroll scaling is seed-dependent**: more unrolls",
|
| 220 |
+
" than trained sometimes help a lot (rec-4x4 seed2 0.394→0.401 at R=5;",
|
| 221 |
+
" rec-1xdynamic seed1 0.088→0.319 at R=5) but not systematically; the",
|
| 222 |
+
" best R for k=64 varies across seeds (1–8).",
|
| 223 |
"- **OOD (k>8) accuracy for fixed-4 is already ~10x chance** early in",
|
| 224 |
" training and slowly rising — consistent with a learned lookup for",
|
| 225 |
" small k failing to compose to large k (theory predicts constant-depth",
|
|
|
|
| 238 |
" resolved by pinning huggingface_hub==1.33.0.",
|
| 239 |
"",
|
| 240 |
"## What remains",
|
| 241 |
+
"- The iso-FLOP question is now answered (see fixed4-long above).",
|
| 242 |
+
"- Seed variance in OOD is large (rec-4x4 k=64: 0.18/0.18/0.39); more",
|
| 243 |
+
" seeds would tighten the family means — a natural next grant.",
|
| 244 |
+
"- Larger S, per-iteration supervision (Fan et al. 2409.15647) and",
|
| 245 |
+
" longer training with dynamic-R sampling are untested here.",
|
| 246 |
"",
|
| 247 |
"## Reproduce",
|
| 248 |
"See README.md; run_all.py drives lrsweep/stage1/stage2; evaluate.py",
|