Skip to content

Record: XSA-all + LeakyReLU² + VR + GA + 7-gram cache (val_bpb=1.0337)#715

Open
Asukabot0 wants to merge 1 commit intoopenai:mainfrom
Asukabot0:submission/xsa-all-leakyrelu-vr-ga-ngram7
Open

Record: XSA-all + LeakyReLU² + VR + GA + 7-gram cache (val_bpb=1.0337)#715
Asukabot0 wants to merge 1 commit intoopenai:mainfrom
Asukabot0:submission/xsa-all-leakyrelu-vr-ga-ngram7

Conversation

@Asukabot0
Copy link
Copy Markdown

Summary

Non-TTT submission: 11L XSA-all + LeakyReLU(0.5)² + Value Residual + Gated Attention + 7-gram backward-looking eval cache.

3-seed mean val_bpb = 1.0337 (std=0.0010) on 8xH100 SXM, 600s wallclock. Artifact ~15.99MB (int6+zstd).

3-Seed Results

Seed Quant Size (bytes) Sliding BPB (s=64)
1337 int6 zstd-16 15,990,221 1.0329
42 int6 zstd-17 15,982,903 1.0334
7 int6 zstd-16 15,992,378 1.0349
Mean 1.0337

Key Techniques

Technique Description
XSA-all (11 layers) Exclusive Self-Attention on all layers
LeakyReLU(0.5)² leaky_relu(x, 0.5).square() preserves negative gradient flow
Value Residual Layer 0 V output mixed into subsequent layers via sigmoid gates
Gated Attention Per-head sigmoid gates on attention output
7-gram eval cache Backward-looking n-gram cache (alpha=0.40, order=7, fixed mixing)

N-gram Cache Compliance

The 7-gram cache is a deterministic, eval-time-only statistical post-processing step:

  • Score-first: Each token is scored by the model before entering the n-gram table
  • Fixed alpha: alpha=0.40 is baked into the code, not tuned per-sample
  • No oracle selection: Same alpha and order for every token
  • Zero learned parameters: Purely statistical, built from the eval data stream
  • Deterministic: Identical results regardless of hardware or random seeds

Training Config

8xH100 SXM, 600s wallclock (~5589 steps at 107ms/step)
WARMDOWN_ITERS=3000, MATRIX_LR=0.025, SCALAR_LR=0.025
XSA_LAST_N=11, LEAKY_RELU=1, EMA=0.997
NGRAM_CACHE=1, NGRAM_ALPHA=0.40, NGRAM_ORDER=7 (eval-time only)

Test plan

  • 3-seed validation (seeds 1337, 42, 7) on 8xH100 SXM
  • All artifacts under 16MB
  • Sliding window eval (stride=64) with n-gram cache
  • Logs included in logs/ directory

3-seed validation on 8xH100 SXM (600s wallclock):
- seed 1337: 1.0329 BPB
- seed 42:   1.0334 BPB
- seed 7:    1.0349 BPB
- mean:      1.0337 BPB (std=0.0010)

Non-TTT, ~15.99MB int6+zstd artifact.
7-gram backward-looking eval cache (alpha=0.40, fixed mixing).

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