Skip to content

Record: 11L Tight SWA + VE128 + XSA4 + TTT (3-seed mean val_bpb=1.1299)#455

Open
kasimte wants to merge 1 commit intoopenai:mainfrom
kasimte:submission/TightSWA-VE128-TTT
Open

Record: 11L Tight SWA + VE128 + XSA4 + TTT (3-seed mean val_bpb=1.1299)#455
kasimte wants to merge 1 commit intoopenai:mainfrom
kasimte:submission/TightSWA-VE128-TTT

Conversation

@kasimte
Copy link
Copy Markdown

@kasimte kasimte commented Mar 22, 2026

Summary

  • NEW SOTA: 1.1299 val_bpb (3-seed mean), beating current record of 1.1428 by 0.0129 nats
  • Built on PR #374 by @unnir (v38: Tight SWA + VE128 + XSA4, 1.1246 single-seed) with added test-time training (TTT)
  • All 3 seeds verified: artifacts < 16MB, training < 10min on 8xH100s

Results

Seed Sliding Window BPB Artifact Size
1337 1.1291 15,787,610
7 1.1309 15,659,426
99 1.1296 15,688,657
Mean 1.1299 15,711,898

Key techniques

  • Tight SWA: 12 checkpoints from last ~600 steps (scale<0.2), zero SWA penalty
  • Test-Time Training: 3 epochs SGD on already-evaluated val tokens (~51s)
  • Late QAT: STE int6 fake-quant during warmdown (scale<0.1)
  • Sliding window eval: stride=64, context=2048 (~100s)
  • 11L, 512-dim, GQA 8/4, XSA last 4 layers, Partial RoPE, Shared VE128, SmearGate+BigramHash

Test plan

  • 3 seeds verified (1337, 7, 99) — all beat SOTA by ≥0.012
  • All artifacts < 16,000,000 bytes
  • train_gpt.py compiles (ast.parse passes)
  • Script runs from within records folder (logs confirm path)
  • PR only adds files to one new folder

Beats SOTA (1.1428) by 0.0129 nats across 3 seeds (1337, 7, 99).
Built on PR openai#374 by @unnir with added test-time training.
@mohosy
Copy link
Copy Markdown

mohosy commented Mar 23, 2026

tight swa with ve128 is clean, the 3 seed mean at 1.1299 is solid. how many checkpoints are you averaging and whats the interval

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.

2 participants