Skip to content

Non-record: Negative findings on codebook quantization, magnitude pruning, multi-token prediction, embedding factorization#212

Closed
mrdavtan wants to merge 3 commits intoopenai:mainfrom
mrdavtan:int6-3xMLP-pr
Closed

Non-record: Negative findings on codebook quantization, magnitude pruning, multi-token prediction, embedding factorization#212
mrdavtan wants to merge 3 commits intoopenai:mainfrom
mrdavtan:int6-3xMLP-pr

Conversation

@mrdavtan
Copy link
Copy Markdown

@mrdavtan mrdavtan commented Mar 20, 2026

Non-record submission. val_bpb=1.1329 (3-seed mean, std=0.0006). 11L 512d int6+zstd-22, 15.3 MB, 8xH100 SXM. 25 experiments in FINDINGS.md.

Findings not tested elsewhere

Codebook quantization (#24): K-means K=256 gives 87% lower reconstruction MSE than int6 per-row, but 25% larger artifact under zstd-22. Higher byte entropy in codebook indices compresses less efficiently than clamped int6 values. Other K values and codecs not tested.

Magnitude pruning (#23): Zeroing smallest 3% of weights increased our artifact by 728KB under zstd-22. 1% and 5% were neutral. Non-monotonic on our checkpoint; other weight distributions may differ.

Multi-token prediction (#14): Auxiliary t+2 head at 0.5× loss weight: +0.0018 BPP, 3% slower. Other weightings and scales not tested.

Embedding SVD (#25): Rank-64 explains 41.9% of variance on tok_emb (1024×512). Linear low-rank factorization not viable at this vocabulary size. Nonlinear methods not tested.

Also documented: Depth recurrence failure (#13), QAT under torch.compile (#2b, credit @152334H), int5 gap (#15), curriculum learning (#11), optimizer coverage (#16), and 14 others.

Reproduction

git clone https://github.com/mrdavtan/parameter-golf.git && cd parameter-golf && git checkout int6-3xMLP-pr
pip install flash-attn --no-cache-dir --no-build-isolation && pip install zstandard sentencepiece huggingface_hub
python3 data/cached_challenge_fineweb.py --variant sp1024
SEED=7 QAT=0 TTT_MAX_STEPS=500 TTT_FREEZE_BLOCKS=1 TRAIN_SEQ_LEN=2048 EVAL_SEQ_LEN=2048 \
UNET_SKIPS=1 ROPE_DIMS=16 LN_SCALE=1 ROPE_BASE=10000 EVAL_STRIDE=32 DOC_ISOLATED_EVAL=0 \
LATE_K_FP16=0 FP16_EMBED_EXPORT=0 \
torchrun --standalone --nproc_per_node=8 records/track_10min_16mb/2026-03-21_11L_XSA_EMA_TTT/train_gpt.py

Builds on PRs #162, #77, #180 and modded-nanogpt.

@mrdavtan
Copy link
Copy Markdown
Author

Update: 5-seed statistical validation added

Seed val_bpb
31337 1.1703
1337 1.1708
2024 1.1712
42 1.1732
7 1.1767
Mean 1.1724
Std 0.0026

Gap vs baseline: 0.036 nats (threshold: 0.005) | t-stat: 44.2 | p < 0.01

All 5 runs on 8×H100 SXM (RunPod Parameter Golf template), PyTorch 2.9.1+cu128, same config, only seed varied. README and submission.json updated with full results.

@mrdavtan mrdavtan reopened this Mar 21, 2026
@mrdavtan mrdavtan changed the title Record: Int6 + 3x MLP + sliding window (val_bpb=1.1708) + 9 ablations Record: 11L XSA + EMA + TTT + Partial RoPE + LN Scale — val_bpb=1.1401 Mar 21, 2026
@mrdavtan mrdavtan force-pushed the int6-3xMLP-pr branch 6 times, most recently from 2bc195b to 3b0b437 Compare March 22, 2026 00:56
@mrdavtan mrdavtan changed the title Record: 11L XSA + EMA + TTT + Partial RoPE + LN Scale — val_bpb=1.1401 Record: 11L XSA + EMA + TTT + Partial RoPE + LN Scale — val_bpb=1.1375 Mar 22, 2026
@mrdavtan mrdavtan changed the title Record: 11L XSA + EMA + TTT + Partial RoPE + LN Scale — val_bpb=1.1375 Record: 11L EMA + TTT + Partial RoPE + LN Scale + XSA4 — mean val_bpb=1.1329 (3 seeds) Mar 22, 2026
3-seed validation (mean 1.1329, std 0.0006). Added findings on
codebook quantization vs zstd, magnitude pruning non-monotonicity,
and embedding SVD analysis. Language cleanup throughout.
@mrdavtan mrdavtan changed the title Record: 11L EMA + TTT + Partial RoPE + LN Scale + XSA4 — mean val_bpb=1.1329 (3 seeds) Non-record: Negative findings on codebook quantization, magnitude pruning, multi-token prediction, embedding factorization Mar 22, 2026
@mrdavtan
Copy link
Copy Markdown
Author

Closing: reproduction command includes non-causal TTT (TTT_MAX_STEPS=500), which is invalid per #402. Negative findings documented in FINDINGS.md remain in the repo.

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