-
Notifications
You must be signed in to change notification settings - Fork 0
Cross Dataset Comparison
Three datasets tested on the same hardware (Apple M5 Max, 64 GB) with autonomous LLM-driven optimization. All runs converge to AR=32 architecture but diverge on optimization hyperparameters — confirming that architecture is hardware-determined while optimization is data-determined.
Cross-Dataset Comparison
| Climbmix (Mar 19) | FineWeb-Edu (Mar 17) | Cosmopedia-v2 (Mar 20) | |
|---|---|---|---|
| Best val_bpb | 1.296 | 1.342 | 0.961 |
| Baseline | 1.297 (pre-optimized) | 1.409 (defaults) | 0.964 (pre-optimized) |
| Improvement | −0.08% | −4.7% | −0.35% |
| Experiments | 101 | 88 | 103 |
| Keep rate | 1.0% (1/101) | 19.3% (17/88) | 3.9% (4/103) |
| Crash rate | 10.9% | 2.3% | 1.9% |
Note on baselines: Climbmix and Cosmopedia-v2 started from pre-optimized baselines (AR=32, EMBEDDING_LR=0.4, SCALAR_LR=0.4 — carried from prior branch runs), while FineWeb-Edu started from full defaults. The low improvement rates for climbmix and cosmopedia reflect near-optimal starting points, not easier optimization landscapes.
All three datasets converge to AR=32. This is the defining finding of the multi-dataset comparison.
| Parameter | Default | Climbmix | FineWeb-Edu | Cosmopedia-v2 |
|---|---|---|---|---|
| ASPECT_RATIO | 64 | 32 | 32 | 32 |
| Peak Memory | 26.1 GB | 15.5 GB | 15.5 GB | 15.5 GB |
| tok/sec | ~30K | ~58K | ~58K | ~59K |
| Training Steps | ~288 | ~530 | ~530 | ~543 |
At AR=32, the model drops from ~50M to ~25M parameters, halving memory and doubling throughput. The ×ばつ more gradient steps in the 5-minute budget more than compensate for reduced per-step capacity. This is a hardware-optimal configuration for the M5 Max at this model scale — not dataset-specific.
The autoresearch loop is time-budgeted (5 minutes). The tradeoff:
- AR=64 (default): ~50M params, ~30K tok/sec, ~288 steps, 26.1 GB
- AR=32 (optimal): ~25M params, ~58K tok/sec, ~530 steps, 15.5 GB
The narrower model runs ×ばつ more gradient updates. For all three data distributions — raw web crawl, educational text, and synthetic textbooks — the extra gradient steps outweigh the reduced model width. This finding is hardware-specific and would likely differ on GPUs with different compute/memory ratios.
While architecture converged, optimization hyperparameters tell three different stories:
| Parameter | Default | Climbmix | FineWeb-Edu | Cosmopedia-v2 |
|---|---|---|---|---|
| MATRIX_LR | 0.04 | 0.0435 (×ばつ) | 0.053 (×ばつ) | 0.0435 (×ばつ) |
| SCALAR_LR | 0.5 | 0.4 (×ばつ) | 0.3 (×ばつ) | 0.4 (×ばつ) |
| EMBEDDING_LR | 0.6 | 0.4 (×ばつ) | 0.59 (×ばつ) | 0.4 (×ばつ) |
| UNEMBEDDING_LR | 0.004 | 0.0041 (×ばつ) | 0.003 (×ばつ) | 0.0033 (×ばつ) |
| WEIGHT_DECAY | 0.2 | 0.2 (×ばつ) | 0.12 (×ばつ) | 0.15 (×ばつ) |
| WARMDOWN_RATIO | 0.5 | 0.5 (×ばつ) | 0.47 (×ばつ) | 0.5 (×ばつ) |
EMBEDDING_LR is the clearest dataset signal. Climbmix and Cosmopedia both converge to 0.4 (33% below default), while FineWeb-Edu stays near default at 0.59. Web crawl and synthetic text both benefit from conservative embedding updates, while educationally-filtered text does not.
UNEMBEDDING_LR splits the other way: climbmix pushes it up (0.0041), while both FineWeb-Edu (0.003) and Cosmopedia (0.0033) push it down. The unembedding layer was the source of late breakthroughs on both FineWeb-Edu and Cosmopedia — it appears to be a consistently underexplored parameter that rewards fine-grained search.
WEIGHT_DECAY shows a three-way split: climbmix keeps the default (0.2), FineWeb-Edu drops to 0.12, and Cosmopedia finds 0.15. Web crawl's high entropy needs full regularization; curated and synthetic text benefit from modest reductions.
WARMDOWN_RATIO is the most stable parameter — only FineWeb-Edu adjusted it (0.47), while both climbmix and Cosmopedia kept the default 0.5.
Climbmix (raw web crawl) and Cosmopedia-v2 (synthetic textbooks) share nearly identical optimal configs despite being radically different data sources:
| Parameter | Climbmix | Cosmopedia-v2 | Match? |
|---|---|---|---|
| MATRIX_LR | 0.0435 | 0.0435 | ✓ Exact |
| EMBEDDING_LR | 0.4 | 0.4 | ✓ Exact |
| SCALAR_LR | 0.4 | 0.4 | ✓ Exact |
| WARMDOWN_RATIO | 0.5 | 0.5 | ✓ Exact |
| WEIGHT_DECAY | 0.2 | 0.15 | ≈ Close |
| UNEMBEDDING_LR | 0.0041 | 0.0033 | ✗ Diverge |
The only clear divergence is UNEMBEDDING_LR (opposite direction from default) and a slight WEIGHT_DECAY difference. This suggests that many optimization hyperparameters are more hardware-dependent than data-dependent — the compute/memory/step-count tradeoff dominates.
| Metric | Climbmix | FineWeb-Edu | Cosmopedia-v2 |
|---|---|---|---|
| Starting point | Pre-optimized | Full defaults | Pre-optimized |
| Total improvement | −0.08% | −4.7% | −0.35% |
| Keep rate | 1.0% | 19.3% | 3.9% |
| Biggest gain | UNEMBED_LR ↑ (−0.001) | AR 40→32 (−0.036) | DEVICE_BATCH ↓ (−0.001) |
| Late breakthrough? | No | Yes (exp34) | Yes (exp98–102) |
| Agent focus | Validation search | Architecture + fine-tuning | MATRIX_LR obsession |
Starting from full defaults, FineWeb-Edu had the most room to improve. The agent found 17 keeps across three phases: optimization tuning → architecture breakthrough → fine-tuning. The AR 64→32 reduction at exp34 was the largest single improvement in any run (−0.036 val_bpb).
With only 1 keep in 101 experiments, the climbmix run served primarily to validate that the inherited config was optimal. The agent exhaustively tested every parameter category and found only UNEMBEDDING_LR 0.004→0.0041 worth keeping.
The agent spent 35 experiments on MATRIX_LR alone, systematically sweeping from 0.028 to 0.055 — all discarded. This "optimization trap" consumed a third of the experiment budget on a dead end. The actual improvements came late from WEIGHT_DECAY (exp88) and UNEMBEDDING_LR (exp98–102).
| Dataset | Type | Best val_bpb | Interpretation |
|---|---|---|---|
| Cosmopedia-v2 | Synthetic textbooks | 0.961 | Highly compressible — consistent style, generated by single model |
| Climbmix | Raw web crawl | 1.296 | Moderate — diverse sources, varied quality |
| FineWeb-Edu | Educational web text | 1.342 | Least compressible — diverse topics filtered for quality |
Counter-intuitively, educationally-scored web text is harder to compress than raw web crawl. FineWeb-Edu's quality filter selects for information-dense documents with diverse vocabulary, making next-token prediction more challenging. Raw web crawl includes repetitive boilerplate, ads, and formulaic content that's easier to model.
ASPECT_RATIO = 32, MATRIX_LR = 0.0435, EMBEDDING_LR = 0.4 UNEMBEDDING_LR = 0.0041, SCALAR_LR = 0.4, WEIGHT_DECAY = 0.2 WARMDOWN_RATIO = 0.5 # mem: 15.5 GB | steps: 530 | MFU: 17.9%
ASPECT_RATIO = 32, MATRIX_LR = 0.053, EMBEDDING_LR = 0.59 UNEMBEDDING_LR = 0.003, SCALAR_LR = 0.3, WEIGHT_DECAY = 0.12 WARMDOWN_RATIO = 0.47 # mem: 15.5 GB | steps: ~530 | MFU: ~18%
ASPECT_RATIO = 32, MATRIX_LR = 0.0435, EMBEDDING_LR = 0.4 UNEMBEDDING_LR = 0.0033, SCALAR_LR = 0.4, WEIGHT_DECAY = 0.15 WARMDOWN_RATIO = 0.5 # mem: 15.5 GB | steps: 543 | MFU: 18.3%
-
Architecture is hardware-determined. AR=32 is optimal across all three datasets on M5 Max. The compute/memory/step-count tradeoff at this model scale favors narrower, faster models regardless of data source.
-
Most optimization hyperparameters are also hardware-determined. MATRIX_LR, EMBEDDING_LR, SCALAR_LR, and WARMDOWN_RATIO converge to similar or identical values across datasets. The dominant factor is the step count achievable within the time budget, not the data distribution.
-
UNEMBEDDING_LR and WEIGHT_DECAY are the data-sensitive parameters. These show genuine divergence: UNEMBEDDING_LR moves in opposite directions for different data, and WEIGHT_DECAY ranges from 0.12 (FineWeb-Edu) to 0.2 (climbmix). These should be dataset-specific tuning targets.
-
Pre-optimized baselines leave almost nothing to find. Keep rates of 1% (climbmix) and 3.9% (cosmopedia) vs 19.3% (FineWeb-Edu from defaults) confirm that starting from a good config makes exhaustive search a validation exercise rather than a discovery exercise.
-
Late breakthroughs are real. Both FineWeb-Edu (exp34) and Cosmopedia (exp98–102) had their most important improvements in the final third of the run. Running only 50 experiments would miss these.
See individual run pages: Climbmix (Mar 19) | FineWeb-Edu (Mar 17) | Cosmopedia-v2 (Mar 20) | Climbmix (Mar 16, legacy)