willcb: per-token noise training won't work for FFNs, relies on sequence aggregation axis

willcb · x · 2026-10-20

In a follow-up, willcb clarifies the scope of the per-token noise perturbation training method: it wouldn't work for FFNs, and only manages to hide the SNR term in the context of sequence models by using a new axis for aggregation. This bounds his earlier claim that the method exploits transformers' already-parallel forward passes, computing the error term once per token as a hidden-state perturbation.

Related event: Debate Over Per-Token Noise Training to Bypass Forward-Pass Costs(2 posts)→

Original post →

More from Research

Research channel →