atlas

Research map / Architectures and efficiency / Attention and transformers

Parallelizing Linear Transformers with the Delta Rule over Sequence Length

NeurIPS 2024 · Architectures and efficiency › Attention and transformers · approach: Sparse efficient transformers for long context · 5 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes a hardware-efficient chunkwise parallel algorithm for training DeltaNet (linear transformers with the delta rule) across sequence length, using the WY/UT representation of Householder products, and scales it to 1.3B and 3B language models plus hybrid attention variants.

Problem

Linear transformers with the delta rule (DeltaNet) have better associative recall than additive linear attention, but prior training algorithms were sequential over sequence length and hardware-inefficient, so they could not be scaled to standard language modeling settings.

Approach

Reparameterize DeltaNet's recurrence as a generalized Householder update and use the WY/UT representation to express the state transitions in a memory-efficient form, enabling a chunkwise parallel algorithm that uses matmuls and tensor cores while only materializing the state at chunk boundaries.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use the delta-rule update instead of GLA's gated linear updateLinear recurrent language models · LAMBADA, 1.3B params / 100B tokensperplexity: 14.47 -> 12.21Helped
Use the delta-rule update instead of GLA's gated linear updateLinear recurrent language models · Zero-shot average, 1.3B params / 100B tokensaccuracy: 51.0 -> 51.6Helped
Use the delta-rule update instead of GLA's gated linear updateLinear recurrent language models · SWDE, SQuAD, FDA, 1.3B params / 100B tokensaccuracy: 52.4 -> 49.5; 42.6 -> 37.4; 22.3 -> 17.2Hurt
Interleave sliding-window attention every other layerDeltaNet · Zero-shot average, 1.3B params / 100B tokensaccuracy: 51.6 -> 52.1Helped
Replace two layers with global attentionDeltaNet · SWDE, 1.3B params / 100B tokensaccuracy: 49.5 -> 71.0Helped
Use L2 normalization instead of L1 normalization, with 1+ELU feature map (ablation)DeltaNet · Zero-shot average, 340M params / 15B tokensaccuracy: 40.1 -> 42.1Helped
Use SiLU instead of 1+ELU with L2 normalization (ablation)DeltaNet · Zero-shot average, 340M params / 15B tokensaccuracy: 42.1 -> 42.1Helped
Use ReLU instead of 1+ELU with L2 normalization (ablation)DeltaNet · Zero-shot average, 340M params / 15B tokensaccuracy: 42.1 -> 40.9Hurt

Surprising findings

Limitations

Experimental setup

Datasets: SlimPajama MQAR MAD RegBench Wikitext LAMBADA PIQA HellaSwag WinoGrande ARC-easy ARC-challenge FDA

Benchmarks: Wikitext LAMBADA PIQA HellaSwag WinoGrande ARC-easy ARC-challenge FDA SWDE SQuAD MQAR MAD

Metrics: perplexity accuracy normalized accuracy training throughput

Baselines: Transformer++ RetNet Mamba GLA Hyena Llama-3.2-3B PowerLM-3B

Builds on: DeltaNet chunkwise parallel linear attention FlashLinearAttention WY representation Gated Linear Attention

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy