Skip to content

Record: 11L VR + GA + LeakyReLU² + Legal Score-First TTT (val_bpb=pending)#490

Draft
amaljithkuttamath wants to merge 4 commits intoopenai:mainfrom
amaljithkuttamath:submission/vr-ga-adamw-ttt
Draft

Record: 11L VR + GA + LeakyReLU² + Legal Score-First TTT (val_bpb=pending)#490
amaljithkuttamath wants to merge 4 commits intoopenai:mainfrom
amaljithkuttamath:submission/vr-ga-adamw-ttt

Conversation

@amaljithkuttamath
Copy link
Copy Markdown

@amaljithkuttamath amaljithkuttamath commented Mar 23, 2026

Summary

val_bpb = pending rerun | 8xH100 SXM | Legal score-first TTT

Architecture improvements on the standard 11L stack: Value Residual + Gated Attention + LeakyReLU(0.5)², with legal score-first TTT.

Update (Mar 23)

Initial submission used pre-eval TTT (non-compliant per #402). This update switches to legal score-first TTT: score each validation chunk before training on it. Score pending rerun with compute credits.

Architecture

  • Value Residual (arXiv:2410.17897): Learned lambda blends first-block V into subsequent blocks. +2 params/layer.
  • Gated Attention (arXiv:2505.06708): Per-head sigmoid gate on attention output. +4104 params/layer.
  • LeakyReLU(0.5)²: Replaces relu² in MLP. Preserves negative gradient flow.
  • Legal score-first TTT: AdamW lr=1e-4, 3 epochs, cosine LR across 131K-token chunks, last 2 blocks + norms unfrozen. Every token scored BEFORE model trains on it.

VR+GA ablated in PR #413 (-0.017 bpb combined). LeakyReLU² proven by PR #569 and #535.

Run command

torchrun --standalone --nproc_per_node=8 train_gpt.py

All hyperparameters set as defaults.

Credits

Value Residual + Gated Attention on PR openai#442 stack.
Single seed (1337), 8xH100 SXM, 14.2 MB artifact.
sahiee-dev added a commit to sahiee-dev/parameter-golf that referenced this pull request Mar 23, 2026
Full stack on thwu1 base (1.1428):
- Value Residual: lambda_v * v0 shortcut to every block, init=0
- Gated Attention: learned scalar gate on attn output, init=1
- XSA: orthogonal self-value removal, last 4 layers
- EMA: decay=0.9999 shadow model used at final eval
- AdamW TTT: lr=0.001, 3 epochs on val tokens before eval
- BigramHash(10240): restored to full size after ablation

Techniques consistent with PR openai#490 (1.0891) and PR openai#486 (1.0887).
Expected range: 1.08-1.10 on 8xH100s.
Trigram ablation confirmed negative at small scale — removed.
Pre-eval TTT was non-compliant per issue openai#402. Now uses
score-first TTT: score each chunk before training on it.
Added LeakyReLU(0.5)² replacing relu² (proven by openai#569, openai#535).
Score pending rerun with compute credits.
@amaljithkuttamath amaljithkuttamath changed the title Record: 11L Value Residual + Gated Attention + AdamW TTT (val_bpb=1.0891) Record: 11L VR + GA + LeakyReLU² + Legal Score-First TTT (val_bpb=pending) Mar 23, 2026
XSA + Partial RoPE + LN Scale + Late QAT + GPTQ + score-first TTT
with temp calibration. Untested, needs 1xH100 validation.
- Deep-clone state dict before bf16 calibration cast (was silently
  corrupting fp32 weights to bf16 before GPTQ, causing 0.328 bpb gap)
- Keep tok_emb.weight as fp16 passthrough instead of int8 quantization
- Fix TTT eval: keep CastedLinear weights in fp32 for stable AdamW
- Remove torch.compile from TTT chunk loop (re-compilation + weight
  mutation = crash)
- Add quant diagnostics: GPTQ key matching + per-layer error stats
@amaljithkuttamath amaljithkuttamath marked this pull request as draft March 25, 2026 21:46
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