Skip to content

Record: 11L + Score-Every-Epoch LoRA TTT 5ep (3-seed mean val_bpb=0.8173)#642

Closed
minh-stakc wants to merge 1 commit intoopenai:mainfrom
minh-stakc:submission/score-every-epoch-ttt-0.8186
Closed

Record: 11L + Score-Every-Epoch LoRA TTT 5ep (3-seed mean val_bpb=0.8173)#642
minh-stakc wants to merge 1 commit intoopenai:mainfrom
minh-stakc:submission/score-every-epoch-ttt-0.8186

Conversation

@minh-stakc
Copy link

@minh-stakc minh-stakc commented Mar 24, 2026

Summary

Results

Seed Pre-TTT BPB Post-TTT BPB Artifact
42 1.1264 0.8186 17.13 MB*
1337 1.1260 0.8187 17.13 MB*
7 1.1275 0.8146 17.13 MB*
Mean 1.1266 0.8173
Std 0.0008 0.0023

*Artifact exceeds 16MB on B200 without FlashAttention 3. Requires 8xH100 validation.

Key Innovation: Score-Every-Epoch Multi-Scale LoRA TTT

Per-document LoRA adaptation where each epoch re-scores all chunks with progressively better-adapted weights. Only the final epoch's scores contribute to BPB.

  • LM-head LoRA rank-16 (2x LR)
  • V projections rank-8 (1.5x LR), Q projections rank-8 (0.5x LR)
  • Per-block bias tuning (3x LR)
  • Cosine LR decay across total TTT steps
  • Post-TTT temperature T=0.98

Architecture (PR #414 stack)

11L, d=512, 8H/4KV GQA, MLP 3x, XSA4, Partial RoPE, LN Scale, EMA(0.997), GPTQ-lite int6 + zstd-22, SmearGate, BigramHash(2048), VE128, Muon WD=0.04

Credits

Test plan

  • 3-seed validation (mean 0.8173, std 0.0023)
  • 8xH100 artifact size verification
  • Eval time under 600s on 8xH100
  • 8-epoch TTT experiments (in progress, targeting sub-0.7)

Score-every-epoch multi-scale LoRA TTT on PR openai#414 base architecture.
Pre-TTT: 1.1264 BPP. Post-TTT: 0.8186 BPB.
@valerio-oai
Copy link
Contributor

Per-document LoRA adaptation where each epoch re-scores all chunks with progressively better-adapted weights. Only the final epoch's scores contribute to BPB.

This is equivalent to training on test, this is an invalid submission, closing this PR for now.

@minh-stakc minh-stakc changed the title Record: 11L + Score-Every-Epoch LoRA TTT 5ep (val_bpb=0.8186) Record: 11L + Score-Every-Epoch LoRA TTT 5ep (3-seed mean val_bpb=0.8173) Mar 25, 2026
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