Skip to content

Record: MTP-2 Funnel + LeakyReLU(0.75)² + Legal TTT + Parallel Muon#1031

Open
michaelwinczuk wants to merge 2 commits intoopenai:mainfrom
michaelwinczuk:submission/mtp2-funnel-leakyrelu075
Open

Record: MTP-2 Funnel + LeakyReLU(0.75)² + Legal TTT + Parallel Muon#1031
michaelwinczuk wants to merge 2 commits intoopenai:mainfrom
michaelwinczuk:submission/mtp2-funnel-leakyrelu075

Conversation

@michaelwinczuk
Copy link
Copy Markdown

Summary

  • Added Multi-Token Prediction (MTP_NUM_HEADS=2, MTP_LOSS_WEIGHT=0.1) as auxiliary training signal
  • MTP forces backbone to learn richer representations by predicting 2 tokens ahead
  • MTP heads discarded at export — zero 16MB impact, zero eval overhead
  • Validated -0.0037 BPB improvement on test pod (apples-to-apples, same hardware)
  • Lighter weight (0.1 vs default 0.2) avoids gradient stealing from main CE loss

Changes from prior submission (val_bpb 1.1185)

  • MTP_NUM_HEADS: 0 → 2
  • MTP_LOSS_WEIGHT: 0.2 → 0.1

Changes from SOTA baseline

  • negative_slope: 0.5 → 0.75
  • MATRIX_LR: 0.025 → 0.027
  • WARMDOWN_ITERS: 3500 → 3700
  • MTP_NUM_HEADS: 0 → 2
  • MTP_LOSS_WEIGHT: 0.2 → 0.1

Research methodology

8 swarm missions + external cross-validation identified MTP as highest-ROI unexplored lever. The "training funnel" concept: MTP auxiliary loss focuses gradient signal on structurally important tokens without adding parameters to the final checkpoint.

🤖 Generated with Claude Code

michaelwinczuk and others added 2 commits March 27, 2026 12:36
One-line activation change (negative_slope 0.5→0.75) + minor LR/warmdown
tuning. Discovered via multi-agent think tank swarm research system.

3-seed results with legal TTT:
  Seed 1337: 1.1183 BPB (15.96MB)
  Seed 42:   1.1194 BPB (15.96MB)
  Seed 2024: 1.1179 BPB (15.95MB)
  Mean:      1.1185 BPB
Added Multi-Token Prediction (MTP_NUM_HEADS=2, MTP_LOSS_WEIGHT=0.1) as
auxiliary training signal. MTP forces the backbone to learn richer
representations by predicting 2 tokens ahead during training. Heads are
discarded at export — zero 16MB impact, zero eval overhead.

Validated -0.0037 BPB improvement on test pod (apples-to-apples comparison).
Lighter MTP weight (0.1 vs default 0.2) avoids gradient stealing from main CE.

Changes from prior submission (1.1185 BPB):
- MTP_NUM_HEADS: 0 -> 2
- MTP_LOSS_WEIGHT: 0.2 -> 0.1

Changes from SOTA baseline:
- negative_slope: 0.5 -> 0.75
- MATRIX_LR: 0.025 -> 0.027
- WARMDOWN_ITERS: 3500 -> 3700
- MTP_NUM_HEADS: 0 -> 2
- MTP_LOSS_WEIGHT: 0.2 -> 0.1

Research: 8 TTS swarm missions + Grok + Gemini cross-validation.
MTP identified as "training funnel" — every gradient counts.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant