lact_100M_r32_bs256_lr3e-4_steps5000

LaCT (Large-Chunk Test-Time Training, SwiGLU MLP fast weights) 100M (rank 32) — Low-rank Fast-Weight Ablation

Pretrained 100M-parameter LaCT (Large-Chunk Test-Time Training, SwiGLU MLP fast weights) with low-rank parameterization (r32) on FineWeb-Edu. Part of a 16-cell ablation (4 archs × 4 ranks: r8, r32, r64, rfull) studying whether constraining the q/k/v fast-weight projections (or LaCT's SwiGLU MLP) to low rank can match or exceed full-rank performance.

Training

Code: see run_main_100M.sh.

Eval results

• FineWeb-Edu val PPL: 21.17
• MQAR (multi-query associative recall):
  • K=4: 0.186
  • K=16: 0.155
  • K=64: 0.193
  • K=256: 0.674
• LAMBADA acc: 0.189
• HellaSwag acc_norm: 0.288
• ARC-Easy acc_norm: 0.412
• PIQA acc_norm: 0.591
• WinoGrande acc: 0.507

Notes

• This is one of 16 cells; the other rank/arch combinations are uploaded under the same HF org (nlproj) with repo names matching the local dump folder, e.g. nlproj/lact_100M_{r8|r32|r64|rfull}_bs256_lr3e-4_steps5000.
• Key finding of the ablation: at this scale, low rank often matches or beats full rank on downstream tasks (LoRA-style "adaptation is intrinsically low-rank" hypothesis). GatedDeltaNet is the exception — its rfull is the strongest in the whole sweep on PPL / LAMBADA / HellaSwag / ARC-Easy.

Run name: lact_100M_r32_bs256_lr3e-4_steps5000