Skip to content

Record: 11L XSA+EMA+TTT, sliding val_bpb=1.1254 (3-seed mean 1.1256)#338

Open
alertcat wants to merge 9 commits intoopenai:mainfrom
alertcat:submission-pr315-ttt
Open

Record: 11L XSA+EMA+TTT, sliding val_bpb=1.1254 (3-seed mean 1.1256)#338
alertcat wants to merge 9 commits intoopenai:mainfrom
alertcat:submission-pr315-ttt

Conversation

@alertcat
Copy link

11L XSA + EMA + TTT + Int6 MLP3x

val_bpb = 1.1254 (sliding window stride=64, best seed 42) | 15.55 MB artifact | 8xH100 SXM, 600s

Key Innovation: TTT on XSA+EMA baseline

First submission combining XSA (Exclusive Self Attention) + EMA + Test-Time Training. After training and quantization, TTT performs 3 epochs of SGD fine-tuning on the validation token stream, adapting the model to the test distribution.

Results (3-seed, 8xH100 SXM)

Seed Steps Sliding BPB (s64) Artifact
1337 7,070 1.1258 15.55 MB
42 7,068 1.1254 15.55 MB
2024 7,069 1.1256 15.55 MB

Mean: 1.1256 | Std: 0.0002

TTT Details

  • 3 epochs SGD on validation tokens (lr=0.002, momentum=0.9)
  • First 2 transformer blocks frozen for stability
  • ~47 seconds on 8xH100 (well under 600s eval limit)
  • Improves post-quant BPB by ~0.002

Architecture (from PR #315)

  • 11L, 512d, 8H/4KV, MLP 3x, relu-squared
  • XSA on last 4 layers, EMA (decay=0.997)
  • SmearGate + BigramHash(2048) + OrthoInit
  • Int6 QAT + Late QAT + zstd-22
  • FlashAttention 3, Muon WD=0.04

Eval Timing

Training: 600s | TTT: 47s | Sliding eval: 73s | Total eval: ~120s

Reproduction

Defaulting to user installation because normal site-packages is not writeable
Requirement already satisfied: zstandard in e:�naconda\lib\site-packages (0.23.0)

Built on PR #315 (XSA, EMA, SmearGate, BigramHash, OrthoInit, sliding window eval).

alertcat and others added 9 commits March 20, 2026 21:22
Innovation over PR openai#198 (SOTA 1.1318):
- 12 transformer layers (was 11): +2.2M params, better representation
- Int5 quantization for MLP weights [-16,15]: 3 zero high bits
  - zstd compression 1.88x vs int6 1.51x, saves ~1.8MB
  - Funds the 12th layer within 16MB budget
- Int6 kept for attention weights (precision-sensitive)
- FA3 fallback for older PyTorch
- LR=0.025 (validated as optimal in A/B testing)

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
RyanLisse added a commit to RyanLisse/parameter-golf that referenced this pull request Mar 21, 2026
New CUDA presets:
- pr332_12l_xsa: 12L/2xMLP, seq2048, momentum 0.99 (from PR openai#332)
- pr338_11l_ttt: 11L/2xMLP, seq2048, momentum 0.99 (from PR openai#338)
- bft_ensemble: 9L/3xMLP Byzantine fault tolerant checkpoint config
- difficulty_adjusted: 10L/2xMLP adaptive search with tight LR
- partial_rope_headtemp: baseline arch with novel attention params

Expanded search: NUM_LAYERS includes 11, TRAIN_SEQ_LEN includes 4096.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
DigitalSword99 pushed a commit to DigitalSword99/parameter-golf that referenced this pull request Mar 21, 2026
- Move EMA shadow weights to GPU (CPU transfers cost ~32% throughput)
- Increase train seq_len from 1024 to 2048 (matches record PR openai#338)

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