LLM·

KronLSTM — Kronecker Delta Rule memory updates — What does it mean?

The news. On October 19, 2026, a team including Sepp Hochreiter (an author of the original LSTM) posted Kronecker Delta Rule for Long Context Linear Attention on arXiv. It proposes KronLSTM, a linear-attention architecture built on a new write rule. At 1.5B parameters trained on 100B tokens of FineWeb-Edu, the authors report the best six-task average and the lowest LAMBADA perplexity among the linear-attention models they compared. Read the paper →

Picture the model's memory as a spreadsheet that never grows. Each fact it has read is a number written where a key column crosses a value row, and reading the sheet costs the same on token 100 as on token 100,000. That is the promise of linear attention: instead of keeping every key and value around, as a KV cache does, it keeps one fixed matrix. The open question is how to edit that sheet.

The oldest rule, the Hebbian write, never edits at all. It adds the new fact on top, so a wrong or outdated number is buried but never fixed, and the sheet's values keep growing unless a forget gate or a normaliser shrinks them. The delta rule fixes that. Before writing, it reads what the sheet currently says for this key and writes only the difference. But the correction is applied along the residual, a vector that depends on what the memory currently holds, so facts filed under a similar key get edited too. In spreadsheet terms, one wrong number makes the rule touch the whole column.

The Kronecker Delta Rule corrects the single cell where the column and the row cross. It reads one scalar, the memory's current score for this exact key–value pair, and moves only that scalar towards a target: a target of 1 stores the association at unit strength and a target of 0 erases it. The direction of the change comes from the input pair alone, never from the memory's contents. The paper shows the disturbance to any other stored pair is the product of value overlap and key overlap. Making either factor small protects that pair, whereas the paper identifies key overlap alone as the controlling factor for the Hebbian and delta rules. The spreadsheet is a schematic: real writes are dense outer products, and a cell stands for an association whose stored value moves. The Q, K, V roles are the same ones you already know; only the write changes.

Write ruleWhat one write doesWhat else it disturbsHow memory size behaves
Hebbian (linear attention, mLSTM)Adds the value–key outer product on topNothing is revised; old values are buriedGrows with the sequence unless a forget gate and normaliser states hold it
Delta rule (DeltaNet, Gated DeltaNet, KDA)Removes the value currently stored at the key, then writes the new oneEvery value stored under an overlapping keyBounded, converges to a steady state
Kronecker Delta Rule (KronLSTM)Corrects one scalar association along the pair's outer productOnly pairs overlapping in both key and valueBounded; the paper proves a bound on the expected memory norm

KronLSTM wraps the rule in three separate gates, all computed from the input: a forget gate per key dimension that decays old memory, an erase gate and a write gate, the last two bounded by a sigmoid. Under a simple model of the data (independent random key–value pairs), the authors prove the memory is a Markov chain that converges geometrically to a steady state, and that its expected norm stays bounded from the first step on. That bound is why the architecture needs none of the extra normaliser and stabiliser states that xLSTM's mLSTM uses; in the authors' high-learning-rate stress test, mLSTM shows gradient-norm spikes while KronLSTM trains steadily. The proof covers the expected norm under that data model, not a guarantee for every run.

Worked example (illustrative). Take one attention head with key and value width 128. The KV cache stores 2 × 128 = 256 numbers per token, so at an 8,192-token context it holds 8,192 × 256 = 2,097,152 numbers. The linear-attention memory is a 128 × 128 matrix, a fixed 16,384 numbers, so the matrix is 128× smaller at 8k tokens and stays that size at 80k. The price is that 16,384 numbers must hold every fact, so write interference matters. Now suppose a new write has key overlap 0.8 and value overlap 0.1 with a stored pair, with step size and error both 1. Under the Kronecker rule that pair's association shifts by 0.8 × 0.1 = 0.08. A rule whose interference is set by key overlap alone passes the full 0.8 key-overlap factor through (the actual shift for the delta rule also depends on the residual, so 0.8 is a ceiling on the factor, not a measured value). The widths and overlaps are made up for illustration; the paper's own head sizes may differ, and the KV-cache count ignores grouped-query attention and other cache-shrinking tricks.

On the paper's own benchmarks, the less invasive write helps most where interference piles up. As a pure linear model at 1.5B parameters and 100B training tokens, KronLSTM reports a six-task average of 59.72 against 58.12 for the strongest competitor, KDA (+1.6 points), and a LAMBADA perplexity of 11.72 against 12.58 for Gated DeltaNet (Table 1). It also leads most needle-in-a-haystack retrieval tasks at longer contexts, which fits the interference argument. Read it with care: this is one paper, one model size, and the authors' own evaluation. The exception is the most repetitive needle task (S1), where KronLSTM underperforms, which the authors attribute to residual accumulation for recurring keys. The paper's appendix also notes that DeltaNet has the smaller norm bound, so the Kronecker rule trades a looser bound for gentler writes.

Goes deeper in: LLM Internals → KV Cache → Memory Cost

Related explainers

Frequently Asked Questions

Check what you knowMap your AI & GPU knowledge across every track — free, role-based