Long-Context Speculative Decoding
Dates:
Stealth — details after publication
What I build, from kernels to clusters.
Dates:
Stealth — details after publication
Authors: Namgyu Ho * (equal contribution) , Huzama Ahmad * (equal contribution) , Woosung Koh * (equal contribution) , Se-Young Yun, Tal Schuster, Cicero Nogueira dos Santos
Can a model become its own selector, saying where it wants to attend, and what does honoring it save? Up to half the decode attention cost, at near-dense accuracy.
Authors: Huzama Ahmad, Se-Young Yun
Where does a pretrained model look, and how do you select those tokens quickly? A small learned selector: decoding nearly 4× faster than FlashAttention at long context, with near-zero accuracy loss.
Authors: Huzama Ahmad, Cao Viet Hai Nam, Se-Young Yun
Why do deep layers contribute less, and what does the answer buy? Gradient Fan-in Asymmetry, and a width taper built on it: latency down and throughput up by about a tenth, at equal perplexity.
Dates:
How much of a model's work is actually necessary? Years of finding out: pretraining from scratch on TPU pods, folding long contexts into a handful of vectors, measuring uncertainty the Bayesian way, and tapering width with depth.
Dates:
The lab's research needs somewhere to run. Many nodes and mismatched GPUs became one system: one identity everywhere, storage that follows each user, strict per-job GPU isolation, and a queue that stays fair in deadline week.
Dates:
A personal cloud: everything I would otherwise rent, running in my apartment. Proxmox and ZFS underneath, pfSense at the edge, and production-grade tooling with a playground's freedom.
Dates:
How well do models hold up far from their training data? Eight undergraduate teams took the language side, four students the cultural one, and the work became four published benchmarks: BEnQA, CLIcK, CULTDIFF, and MIXCUBE.
Dates:
Can a model sharpen its own reasoning with no one grading it? It judges its greedy answer against its own sampled alternatives, and that judgment is the reward: accuracy up to 5% higher across four benchmarks, no human in the loop.