Ngram Module Experiments: Loss Not a Perfect Signal

nrehiew_ · x · 2026-08-27

Regarding the Ngram module (borrowed from DeepSeek Engram), ablations on layer placement showed no clear winner, leading to placement at layer 2 for CPU prefetching. Data suggests loss is not a perfect signal: downstream loss decreases with vocab size, but this does not consistently translate to better evals.

Related event: Qwen Training Details: Full-Network Muon Optimizer and Residual Stream Design(5 posts)→

Original post →

More from Research

Research channel →