Skip to content

submission: LeakyReLU2 + TrigramHashEmbedding (1.1448 bpb)#884

Open
BhatiaUday wants to merge 1 commit intoopenai:mainfrom
BhatiaUday:submission/leakyrelu2-trigram-hash
Open

submission: LeakyReLU2 + TrigramHashEmbedding (1.1448 bpb)#884
BhatiaUday wants to merge 1 commit intoopenai:mainfrom
BhatiaUday:submission/leakyrelu2-trigram-hash

Conversation

@BhatiaUday
Copy link

@BhatiaUday BhatiaUday commented Mar 26, 2026

LeakyReLU² + TrigramHashEmbedding (non-record track)

val_bpb: 1.1448 (3-seed mean, sliding window stride 64) | ~15.6 MB

Summary

Novel hash-based TrigramHashEmbedding on PR #414 stack (11L EMA + GPTQ-lite) with LeakyReLU(0.5)² from PR #549. The trigram embedding extends BigramHashEmbedding (PR #198) from 2-grams to 3-grams using XOR prime hashing into 2048 buckets, capturing richer local context at the input layer.

Results (3 seeds, 1×H100 NVL, proportional wallclock)

Seed Steps Sliding val_bpb RT val_bpb Artifact
1337 4,145 1.14587 1.16935 15,642,196
42 4,142 1.14562 1.16925 15,587,464
2025 4,142 1.14306 1.16677 15,591,832
Mean 4,143 1.14485 1.16846 15,607,164

Compute Note

Validated on 1×H100 NVL 96GB with proportional wallclock (4054s = 600s x 6.76) to match the 8xH100 training trajectory. The script uses grad_accum = 8 // world_size (auto: 1 on 8-GPU, 8 on 1-GPU) for identical effective batch size. Defaults to MAX_WALLCLOCK_SECONDS=600 on 8xH100.

Key Changes from Base

Files

  • records/track_non_record_16mb/2026-03-26_LeakyReLU2_TrigramHash/train_gpt.py
  • records/track_non_record_16mb/2026-03-26_LeakyReLU2_TrigramHash/submission.json
  • records/track_non_record_16mb/2026-03-26_LeakyReLU2_TrigramHash/requirements.txt
  • records/track_non_record_16mb/2026-03-26_LeakyReLU2_TrigramHash/README.md

Novel hash-based trigram embedding on PR openai#414 stack with LeakyReLU(0.5)^2.
3-seed mean sliding window val_bpb: 1.14485, all artifacts under 16MB.
Validated on 1xH100 NVL with proportional wallclock (4054s = 600s x 6.76).
Track: non-record (1xH100 validation).
@BhatiaUday BhatiaUday force-pushed the submission/leakyrelu2-trigram-hash branch from 70c9ab9 to 16e1e8c Compare March 26, 2026 18:45
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