DeltaNet

DeltaNet matters because it turns a more expressive linear-attention update into something that can be trained efficiently over sequence length, closing part of the gap between recurrent linear models and softmax attention.

Metadata

FieldValue
PaperParallelizing Linear Transformers with the Delta Rule over Sequence Length
AuthorsSonglin Yang, Bailin Wang, Yu Zhang, Yikang Shen, Yoon Kim
Year2024
VenueNeurIPS 2024
arXiv2406.06484
OpenReviewy8Rm4VNRPH
Implementationflash-linear-attention
Statusfull note started

Question

Linear attention and state-space models aim for a useful compromise:

For ordinary linear attention, a simplified recurrent update is:

This is efficient, but the additive update can be weak for tasks that require precise in-context retrieval or associative memory.

DeltaNet asks:

The answer is to use a delta-rule update and then derive a hardware-efficient parallel algorithm.

Main Claim

The durable architecture claim is:

The paper reports that this makes DeltaNet competitive with, and in some reported settings stronger than, recent linear-time baselines such as Mamba and GLA.

For this wiki, the central lesson is:

Architecture Contract

ItemContract
Inputcausal token sequence
Baseline familylinear Transformer / linear attention
Main updatedelta-rule recurrent matrix-state update
Main algorithmic problemparallelize training over sequence length
Mathematical toolmemory-efficient representation for products of Householder matrices
Scaling targetstandard language-model training settings
Hybrid variantsDeltaNet plus sliding-window or global attention layers
Main comparisonsoftmax Transformer, GLA, Mamba, RetNet-style linear-time models

The loss remains the causal language-model objective:

The contribution is in the sequence mixer and training algorithm.

From Additive Memory to Delta Rule

In simple linear attention, the matrix memory is updated additively:

That update stores new key-value information but does not explicitly correct an old prediction.

A delta-rule view instead updates memory using a prediction error. A simplified form is:

The exact paper implementation has additional parameterization and normalization details, but this is the useful reading:

This makes DeltaNet closer to an online learning update over a recurrent memory state.

Matrix-Valued RNN View

DeltaNet can be read as a matrix-valued recurrent neural network:

The hidden state is not a vector:

This is the same broad state shape family as GLA, but the update rule is different.

ModelState Update Intuition
linear attentionadd key-value outer product
GLAadd key-value information with data-dependent forgetting
DeltaNetupdate memory through a delta-rule correction

That difference matters for retrieval. A recurrent state can be compact and efficient, but it must store the right associations.

Parallelization Problem

A sequential recurrence is easy to define:

It is hard to train efficiently on modern GPUs if every step must wait for the previous step.

The paper’s algorithmic contribution is to parallelize DeltaNet over sequence length. The key point for paper reading is:

A recurrent architecture needs a parallel or chunkwise training form. Otherwise, it can be theoretically efficient but practically slow.

Householder and Compact Representation

The paper reparameterizes DeltaNet so that the recurrence can be handled through products of structured transformations, specifically using a memory-efficient representation for products of Householder matrices.

For the wiki, the exact linear algebra can be summarized as:

Instead, a compact representation lets the algorithm avoid storing a full matrix-valued hidden state for every token during parallel training.

This is the same architecture-systems lesson as FlashAttention and GLA:

Hybrid DeltaNet

The paper also studies hybrid models that combine DeltaNet layers with softmax attention layers.

HybridWhy It Matters
DeltaNet plus sliding-window attentionadds local precise token interactions
DeltaNet plus a few global attention layersrestores some full-context retrieval routes

This is similar in spirit to Jamba, which combines Mamba, attention, and MoE. The broader lesson is:

Relation to GLA

GLA improves linear attention with data-dependent gates and a hardware-aware training algorithm.

DeltaNet changes the update rule itself:

PaperMain Update
GLAgated linear-attention memory
DeltaNetdelta-rule memory correction

Both papers should be read together:

Relation to Mamba and Mamba-2

Mamba-style models use selective state-space updates:

DeltaNet uses a matrix-valued memory update over key-value-like features. Both are recurrent alternatives to dense softmax attention, but they compress history differently.

FamilyState
Mambaselective SSM vector/state channels
Mamba-2SSD matrix-mixer view of SSM/attention duality
GLAgated linear-attention matrix state
DeltaNetdelta-rule matrix memory

This makes DeltaNet part of the same reading route as Mamba, Mamba-2, RetNet, and Hyena.

Evidence to Read Carefully

The paper reports experiments at standard language-model scale, including a 1.3B model trained for 100B tokens, and compares against linear-time baselines and Transformer baselines.

Read the evidence by claim:

EvidenceSupportsDoes Not Prove Alone
language-model perplexityDeltaNet scales beyond toy tasksuniversal superiority over softmax Transformers
zero-shot benchmarksrepresentation and generation utilityall downstream tasks or all data regimes
retrieval-focused tasksbetter associative memory than simpler linear recurrencesperfect long-context retrieval
throughputparallel algorithm is practicalspeed independent of implementation and hardware
hybrid resultsattention plus DeltaNet can be strongpure recurrent models always suffice

The main durable result is not a single leaderboard number. It is that a stronger recurrent memory update can be made parallel enough for large-scale language modeling.

Practical Checks

  • Identify whether a paper uses additive linear attention, gated linear attention, delta-rule updates, retention, or SSM.
  • Track whether recurrent state is vector-valued or matrix-valued.
  • Ask whether the training algorithm parallelizes over sequence length.
  • Separate inference complexity from training wall-clock speed.
  • Check retrieval-heavy tasks separately from average language-model perplexity.
  • For hybrid models, count how many layers use full attention, sliding attention, or recurrent linear updates.
  • Compare with matched tokenizer, data, parameter count, training tokens, and implementation.

Where It Fits

DeltaNet is a useful next paper after GLA:

It also gives a good template for reading future linear-attention papers: do not only ask whether the recurrence is linear in ; ask what memory update is used and whether that update has a practical parallel training algorithm.