A linear attention architecture that combines data-dependent gating with the delta update rule to achieve efficient, high-quality sequence modeling. Introduced by NVIDIA Research (NVlabs) as an improvement over Mamba2 and DeltaNet.
Background: linear attention
Standard Transformer attention scales quadratically with sequence length L. Linear attention replaces softmax with a kernelized dot product and reframes computation as a linear RNN with a matrix-valued state S_t, reducing inference memory to O(1) per step regardless of sequence length.
Delta rule
DeltaNet uses the delta update rule (Widrow et al., 1960): instead of accumulating all key-value pairs into the state (Hebbian learning), it selectively replaces the value associated with the current key. This is more surgical — it only modifies one key-value pair at a time while leaving others intact. The update is:
S_t = S_{t-1}(I − beta_t * k_t * k_t^T) + beta_t * v_t * k_t^T
This gives superior associative recall but lacks a mechanism to rapidly clear outdated information during context switches.
Gated delta rule
Unifying gating and the delta rule gives:
S_t = S_{t-1} * (alpha_t * (I − beta_t * k_t * k_t^T)) + beta_t * v_t * k_t^T
- alpha_t (gate, 0 to 1): controls state decay. Setting alpha_t → 0 flushes the entire state (rapid forgetting); alpha_t → 1 reduces to the pure delta rule (selective update).
- beta_t (writing strength, 0 to 1): controls how much of the new value is written.
This combination enables both bulk forgetting (via gating) and targeted updates (via delta rule), giving flexible memory control.
Hardware-efficient training
The recurrence is sequential, but a chunkwise parallel algorithm (using the WY representation and matmuls) enables GPU-efficient training with tensor cores. Within each chunk, the sequential dependencies are captured by a small C x C matrix inverse (C = chunk size).
Hybrid architectures
The layer is often combined with sliding-window attention (SWA) or Mamba2 layers in hybrid models. For example, a layout of 16 blocks of (3x gated-delta layers → 1x gated-attention layer) combines linear attention efficiency with standard attention's retrieval strength.
References
- Yang et al. (2024), "Parallelizing Linear Transformers with the Delta Rule over Sequence Length", arXiv:2406.06484
- NVlabs/GatedDeltaNet (ICLR 2025), arXiv:2412.06464
- GitHub: https://github.com/NVlabs/GatedDeltaNet