Linear Attention Architectures: Mechanisms, Trade-offs, and Cross-Layer Routing
Summary
This paper compares softmax attention with four linear attention architectures (DeltaNet, Gated DeltaNet, Kimi Delta Attention, Gated DeltaNet-2) and introduces cross-layer routing mechanisms. Experiments at 350M parameters show Kimi Delta Attention with Muon achieves lowest validation loss, while pure Gated DeltaNet with AdamW has highest throughput.
View Cached Full Text
Cached at: 07/10/26, 06:16 AM
# Linear Attention Architectures: Mechanisms, Trade-offs, and Cross-Layer Routing
Source: [https://arxiv.org/html/2607.07953](https://arxiv.org/html/2607.07953)
Tommaso Cerruti1,\* &Tim Rieder1,\* &George Rowlands1,\* &Lingfeng Jin1,\* &Imanol Schlag1,2
1ETH Zurich, D\-INFK2ETH AI Center, ETH Zurich\*Equal contribution
###### Abstract
Self\-attention lets each token retrieve information from the full context, but its quadratic cost in sequence length limits training and inference at long context\. This paper presents a comparative study of softmax attention and four recent recurrent linear\-attention architectures: DeltaNet, Gated DeltaNet, Kimi Delta Attention, and Gated DeltaNet\-2\. We express these mechanisms in a common recurrent\-memory notation, making explicit how they differ in expressivity, memory decay, erase and write control, training throughput, and implementation complexity\. Our experiments center on 350M\-parameter models trained for 15B tokens, and include optimizer and learning\-rate comparisons, hybrid\-versus\-pure stack comparisons, sequence\-length runtime measurements, larger DeltaNet runs at 1\.3B and 3B parameters, and a small set of downstream evaluations\. The reported speed results measure training throughput and iteration time; we do not provide an empirical inference\-speed benchmark\. Within the reported 350M\-parameter, 15B\-token sweep, Kimi Delta Attention with Muon reaches the lowest final validation loss, a pure Gated DeltaNet stack trained with AdamW has the highest normalized training throughput, hybrid stacks generally improve loss at a throughput cost, and Muon consistently lowers final validation loss relative to AdamW in the matched architecture settings we evaluate\. We introduce and evaluate lightweight cross\-layer routing mechanisms for DeltaNet\-style memories\. The most natural DeltaNet\-inspired formulation, forwarding a lower layer’s delta\-rule write*error*into the next layer’s value target, does not improve over matched baselines\. Routing into the aligned hidden stream and forwarding the write*value*instead yields a modest improvement in the matched runs we report:*Cross\-Layer Value Routing \(CLVR\)*lowers final validation loss for both DeltaNet and Gated DeltaNet\.
*Keywords*Linear attention⋅\\cdotRecurrent associative memory⋅\\cdotDeltaNet⋅\\cdotCross\-layer routing
## 1Introduction
Transformer language models rely on self\-attention\[[1](https://arxiv.org/html/2607.07953#bib.bib1)\]to let each token retrieve information from other tokens in the context\. This token\-to\-token information exchange, often called sequence mixing, is central to their expressivity\. However, the same mechanism that makes softmax attention expressive also makes it expensive: computing the attention matrix requires explicit pairwise comparisons between every pair of tokens in the input sequence, leading to a cost that scales quadratically with sequence length\. As models are deployed with larger context windows, this cost becomes a dominant factor in both training and inference, and motivates the search for sequence mixers with more favorable scaling\.
*Linear attention*is one of the most prominent responses to this challenge\. By replacing the softmax kernel with a feature\-map decomposition, linear attention can be reformulated as a recurrent update over a constant\-size memory matrix, yielding linear\-time training and constant\-time inference per token\. Early linear\-attention variants traded a substantial amount of accuracy for this efficiency, but recent work has narrowed the gap considerably\. In particular, DeltaNet\[[2](https://arxiv.org/html/2607.07953#bib.bib2)\]reinterprets linear attention as a fast\-weight programmer and replaces the naive additive update with an error\-correcting*delta rule*: instead of writing the full value at every step, the model writes only the residual between the current value and what the fixed\-size memory already predicts, reducing interference from overlapping key\-value associations\. Subsequent variants build on this idea by adding increasingly fine\-grained mechanisms for controlling memory\. Gated DeltaNet\[[3](https://arxiv.org/html/2607.07953#bib.bib3)\]introduces a learned scalar decay over the memory state, Kimi Delta Attention\[[4](https://arxiv.org/html/2607.07953#bib.bib4)\]refines this with a channel\-wise decay gate, and Gated DeltaNet\-2\[[5](https://arxiv.org/html/2607.07953#bib.bib5)\]further decouples the active delta\-rule edit into separate channel\-wise erase and write gates\. Together, these architectures define a small but rapidly evolving family of recurrent memories that share a common skeleton but differ in how they balance selectivity, forgetting, and granularity of control\.
Alongside these advances, a separate line of work has argued that deep language models suffer from*information dilution*: as representations propagate through many layers, useful signals extracted at lower depths may become progressively harder to recover\. Proposals such as Attention Residuals\[[6](https://arxiv.org/html/2607.07953#bib.bib6)\]and Mixture\-of\-Depths Attention\[[7](https://arxiv.org/html/2607.07953#bib.bib7)\]address this by introducing explicit cross\-layer pathways or depth\-wise attention\. These mechanisms are effective, but applying them directly to linear recurrent architectures partially defeats their efficiency advantages\. This raises a natural question: is there a lightweight way to share information across depth that respects the linear\-time structure of DeltaNet\-style memories?
This paper makes four contributions:
1. 1\.We express softmax attention, DeltaNet, Gated DeltaNet, Kimi Delta Attention, and Gated DeltaNet\-2 in a common recurrent\-memory notation\. The framework isolates the role of the recurrent memoryWW, the delta\-rule residualrr, and the different decay, erase, and write mechanisms\.
2. 2\.We provide training\-capable Megatron implementations and integrations of the newer linear\-attention variants used in the study, including Kimi Delta Attention, Gated DeltaNet\-2, and the cross\-layer routing variants introduced here\. The implementation is available at[![[Uncaptioned image]](https://arxiv.org/html/2607.07953v1/assets/github-mark.png)tommasocerruti/linear\-attention\-architectures](https://github.com/tommasocerruti/linear-attention-architectures)\.
3. 3\.We use this shared setup to map empirical trade\-offs among the architectures, comparing validation loss, throughput, optimizer and learning\-rate sensitivity, hybrid\-versus\-pure stack structure, sequence\-length timing, larger DeltaNet runs, and downstream behavior\.
4. 4\.We introduce lightweight cross\-layer routing for DeltaNet\-style memories\. Starting from*Cross\-Layer Error Residuals \(CLER\)*, which forward a lower layer’s delta\-rule write error into the next layer’s value target, we find that this DeltaNet\-inspired formulation does not improve over matched baselines\. We then route into the aligned hidden stream and find that the layer’s write*value*, rather than its write error, is the useful signal\. The resulting method,*Cross\-Layer Value Routing \(CLVR\)*, gives a small reduction in final validation loss in the matched DeltaNet and Gated DeltaNet runs we report, while preserving the host architecture’s linear\-time structure\.
We organize the comparison along three axes:
- •*Mechanism*: we compare the algebraic structure of each recurrent update\.
- •*Trade\-offs*: we discuss how each design balances selectivity, memory decay, erase and write control, and memory granularity\.
- •*Empirical behavior*: we report validation\-loss, throughput, sequence\-length timing, and downstream results across the variants we study\.
Our aim is not to declare a single best architecture, but to make the design space legible\. Viewing these mechanisms side by side makes it easier to see which choices are responsible for which properties, where scaling behavior is strongest, and which cross\-layer routing variants remain open questions for future evaluation\.
## 2Background: From Softmax Attention to Linear Attention
This section reviews the path from standard softmax attention to linear attention, and introduces the recurrent\-memory perspective that underlies all of the DeltaNet\-style architectures studied in this paper\. We focus on the algebraic structure rather than implementation details, and use the same single\-head notation that will be used throughout the rest of the paper\.
#### Softmax attention\.
Given queries and keysq\(i\),k\(i\)∈ℝdkq^\{\(i\)\},k^\{\(i\)\}\\in\\mathbb\{R\}^\{d\_\{k\}\}and valuesv\(i\)∈ℝdvv^\{\(i\)\}\\in\\mathbb\{R\}^\{d\_\{v\}\}fori=1,…,Ti=1,\\dots,T, causal softmax attention computes the output at positioniias
y\(i\)=∑j≤iexp\(q\(i\)⊤k\(j\)/dk\)∑ℓ≤iexp\(q\(i\)⊤k\(ℓ\)/dk\)v\(j\)\.y^\{\(i\)\}=\\sum\_\{j\\leq i\}\\frac\{\\exp\\\!\\left\(q^\{\(i\)\\top\}k^\{\(j\)\}/\\sqrt\{d\_\{k\}\}\\right\)\}\{\\sum\_\{\\ell\\leq i\}\\exp\\\!\\left\(q^\{\(i\)\\top\}k^\{\(\\ell\)\}/\\sqrt\{d\_\{k\}\}\\right\)\}\\,v^\{\(j\)\}\.\(1\)The current query is explicitly compared against every previous key, and the values are combined via a normalized weighted average\. This formulation is highly expressive: each token can in principle retrieve information from any previous position, and the softmax normalization gives the mechanism a built\-in form of soft selection\. However, the same explicit comparison is also the source of its quadratic cost\. Training on a sequence of lengthTTrequires computing the𝒪\(T2\)\\mathcal\{O\}\(T^\{2\}\)query\-key interaction matrix, and storing the full attention matrix also requires𝒪\(T2\)\\mathcal\{O\}\(T^\{2\}\)memory\. For long contexts, both the time and the memory cost become limiting\.
#### The kernel view\.
Linear attention arises from a simple observation: if the unnormalized attention weight could be written as an inner product between feature maps of the query and the key, then the sum over previous positions could be rearranged to avoid the explicit pairwise comparison\. Concretely, suppose there is a feature mapϕ:ℝdk→ℝdϕ\\phi:\\mathbb\{R\}^\{d\_\{k\}\}\\to\\mathbb\{R\}^\{d\_\{\\phi\}\}such that the \(unnormalized\) similarity can be approximated as
exp\(q\(i\)⊤k\(j\)/dk\)≈ϕ\(q\(i\)\)⊤ϕ\(k\(j\)\)\.\\exp\\\!\\left\(q^\{\(i\)\\top\}k^\{\(j\)\}/\\sqrt\{d\_\{k\}\}\\right\)\\;\\approx\\;\\phi\\\!\\left\(q^\{\(i\)\}\\right\)^\{\\top\}\\phi\\\!\\left\(k^\{\(j\)\}\\right\)\.\(2\)Substituting this into the unnormalized numerator of the attention output and exchanging the order of summation gives
∑j≤iϕ\(q\(i\)\)⊤ϕ\(k\(j\)\)v\(j\)=ϕ\(q\(i\)\)⊤∑j≤iv\(j\)⊗ϕ\(k\(j\)\)⏟accumulated memory\.\\sum\_\{j\\leq i\}\\phi\\\!\\left\(q^\{\(i\)\}\\right\)^\{\\top\}\\phi\\\!\\left\(k^\{\(j\)\}\\right\)v^\{\(j\)\}=\\phi\\\!\\left\(q^\{\(i\)\}\\right\)^\{\\top\}\\underbrace\{\\sum\_\{j\\leq i\}v^\{\(j\)\}\\otimes\\phi\\\!\\left\(k^\{\(j\)\}\\right\)\}\_\{\\text\{accumulated memory\}\}\.\(3\)The bracketed sum no longer depends on the query and can be maintained incrementally asiiadvances\. The query then interacts with a single matrix rather than with all previous tokens individually\.
#### Recurrent memory\.
Defining the running sum as a memory matrix
W\(i\)=∑j≤iv\(j\)⊗ϕ\(k\(j\)\)∈ℝdv×dϕ,W^\{\(i\)\}\\;=\\;\\sum\_\{j\\leq i\}v^\{\(j\)\}\\otimes\\phi\\\!\\left\(k^\{\(j\)\}\\right\)\\;\\in\\;\\mathbb\{R\}^\{d\_\{v\}\\times d\_\{\\phi\}\},\(4\)the \(unnormalized\) output of linear attention can be written as a recurrence:
W\(i\)=W\(i−1\)\+v\(i\)⊗ϕ\(k\(i\)\),y\(i\)=W\(i\)ϕ\(q\(i\)\)\.W^\{\(i\)\}=W^\{\(i\-1\)\}\+v^\{\(i\)\}\\otimes\\phi\\\!\\left\(k^\{\(i\)\}\\right\),\\qquad y^\{\(i\)\}=W^\{\(i\)\}\\,\\phi\\\!\\left\(q^\{\(i\)\}\\right\)\.\(5\)Each token contributes an additive write to the memory, and each output is a single matrix\-vector product\. The cost per token is constant in the sequence length, and total training cost is linear inTT\. The memory matrix can be interpreted in two equivalent ways: as a compressed representation of all previous \(key, value\) pairs, or, following the fast\-weight programmer view\[[2](https://arxiv.org/html/2607.07953#bib.bib2)\], as a set of*fast weights*that are written to and read by the surrounding network\.
#### What linear attention gives up\.
The efficiency gains come at a price\. Softmax attention performs a per\-query normalization that effectively re\-weights the contributions of all previous tokens, and its exponential kernel makes the attention distribution sharp\. Linear attention, in its naive additive form, has neither property\. The memoryW\(i\)W^\{\(i\)\}accumulates previous writes without a learned decay or erase mechanism, so old associations remain active unless they are overwritten indirectly\. There is also no built\-in mechanism for sharpening the retrieval distribution or for forgetting stale information\. As more \(key, value\) pairs are written into a fixed\-size memory, interference between stored associations grows: a query that should retrieve one value may also pick up spurious contributions from unrelated keys that happen to be correlated with it underϕ\\phi\. This interference is the central failure mode that more recent linear\-attention variants are designed to address\.
#### From additive writes to delta\-rule writes\.
DeltaNet and its successors can be understood as principled answers to the interference problem\. Instead of always adding the raw valuev\(i\)v^\{\(i\)\}to memory, these architectures first ask what the current memory already predicts for the current key,
v¯\(i\)=W\(i−1\)ϕ\(k\(i\)\),\\bar\{v\}^\{\(i\)\}\\;=\\;W^\{\(i\-1\)\}\\phi\\\!\\left\(k^\{\(i\)\}\\right\),\(6\)and then write only the residualr\(i\)=v\(i\)−v¯\(i\)r^\{\(i\)\}=v^\{\(i\)\}\-\\bar\{v\}^\{\(i\)\}\. This converts the memory update from a pure accumulator into an error\-correcting writer, and gives the architecture a notion of*what the memory does not yet know*\. Subsequent variants augment this delta\-rule update with mechanisms for forgetting and editing, ranging from a single scalar decay gate to channel\-wise decay and finally to separate channel\-wise erase and write gates\. These additions give the model explicit control over how aggressively old information is decayed, which associations are erased, and which value channels are committed to memory\. The next section formalizes these architectures in a common notation and makes their differences precise\.
## 3Architectures
We describe all attention mechanisms in a single\-head notation and omit output projections, normalization layers, and feed\-forward blocks for clarity\. At token positionii, the input representationx\(i\)x^\{\(i\)\}is mapped to a query, key, and value vector,
q\(i\),k\(i\)∈ℝdk,v\(i\)∈ℝdv\.q^\{\(i\)\},k^\{\(i\)\}\\in\\mathbb\{R\}^\{d\_\{k\}\},\\qquad v^\{\(i\)\}\\in\\mathbb\{R\}^\{d\_\{v\}\}\.For the linear\-attention variants, keys and queries are passed through a feature mapϕ\(⋅\)\\phi\(\\cdot\), and the recurrent state is represented by a matrix
W\(i\)∈ℝdv×dϕ\.W^\{\(i\)\}\\in\\mathbb\{R\}^\{d\_\{v\}\\times d\_\{\\phi\}\}\.We interpretW\(i\)W^\{\(i\)\}as the main associative memory storing key\-value information up to tokenii\. In this orientation, the memory maps transformed keys to values, so a read at keyk\(i\)k^\{\(i\)\}returns a value\-space vector\. For compactness, we write
κ\(i\)=ϕ\(k\(i\)\)∈ℝdϕ\\kappa^\{\(i\)\}=\\phi\\\!\\left\(k^\{\(i\)\}\\right\)\\in\\mathbb\{R\}^\{d\_\{\\phi\}\}\(7\)for the feature\-mapped key\.
After a recurrent\-memory variant has updatedW\(i\)W^\{\(i\)\}, its token output is obtained by querying the memory with the transformed query:
y\(i\)=W\(i\)ϕ\(q\(i\)\)\.y^\{\(i\)\}=W^\{\(i\)\}\\phi\\\!\\left\(q^\{\(i\)\}\\right\)\.\(8\)
Several quantities are shared across the DeltaNet\-style variants\. Using the feature\-mapped key from Eq\. \([7](https://arxiv.org/html/2607.07953#S3.E7)\), we define the memory prediction at the current key as
v¯\(i\)=W\(i−1\)κ\(i\),\\bar\{v\}^\{\(i\)\}=W^\{\(i\-1\)\}\\kappa^\{\(i\)\},\(9\)and the corresponding delta\-rule residual as
r\(i\)=v\(i\)−v¯\(i\)\.r^\{\(i\)\}=v^\{\(i\)\}\-\\bar\{v\}^\{\(i\)\}\.\(10\)Thus,r\(i\)r^\{\(i\)\}measures the part of the current value that is not already predicted by the existing memory\.
The scalar gates used by the recurrent variants are
α\(i\)=fα\(x\(i\)\)∈\(0,1\),β\(i\)=σ\(wβ⊤x\(i\)\)∈\(0,1\)\.\\alpha^\{\(i\)\}=f\_\{\\alpha\}\\\!\\left\(x^\{\(i\)\}\\right\)\\in\(0,1\),\\qquad\\beta^\{\(i\)\}=\\sigma\\\!\\left\(w\_\{\\beta\}^\{\\top\}x^\{\(i\)\}\\right\)\\in\(0,1\)\.\(11\)Here,α\(i\)\\alpha^\{\(i\)\}is a learned token\-dependent decay factor, whileβ\(i\)\\beta^\{\(i\)\}controls the write strength of the main memory\.
For Kimi Delta Attention and Gated DeltaNet\-2, we additionally use a vector\-valued forget gate
𝜶\(i\)=f𝜶\(x\(i\)\)∈\(0,1\)dϕ,Dα\(i\)=Diag\(𝜶\(i\)\)∈ℝdϕ×dϕ\.\\bm\{\\alpha\}^\{\(i\)\}=f\_\{\\bm\{\\alpha\}\}\\\!\\left\(x^\{\(i\)\}\\right\)\\in\(0,1\)^\{d\_\{\\phi\}\},\\qquad D\_\{\\alpha\}^\{\(i\)\}=\\operatorname\{Diag\}\\\!\\left\(\\bm\{\\alpha\}^\{\(i\)\}\\right\)\\in\\mathbb\{R\}^\{d\_\{\\phi\}\\times d\_\{\\phi\}\}\.\(12\)This can be viewed as a channel\-wise analogue of the scalar decay gate, applied along the transformed\-key dimension of the recurrent memory\.
For Gated DeltaNet\-2, we also use a channel\-wise erase gate and a channel\-wise write gate,
𝒃\(i\)=σ\(Wbx\(i\)\)∈\(0,1\)dϕ,𝒘\(i\)=σ\(Wwx\(i\)\)∈\(0,1\)dv\.\\bm\{b\}^\{\(i\)\}=\\sigma\\\!\\left\(W\_\{b\}x^\{\(i\)\}\\right\)\\in\(0,1\)^\{d\_\{\\phi\}\},\\qquad\\bm\{w\}^\{\(i\)\}=\\sigma\\\!\\left\(W\_\{w\}x^\{\(i\)\}\\right\)\\in\(0,1\)^\{d\_\{v\}\}\.\(13\)The erase gate controls which transformed\-key channels are used to remove old content from memory, while the write gate controls which value channels are committed to memory\.
### 3\.1Softmax Attention
Standard causal softmax attention, defined in Eq\. \([1](https://arxiv.org/html/2607.07953#S2.E1)\), directly compares the current query with all previous keys and forms a normalized weighted average of the corresponding values\.
Softmax attention is highly expressive because each token can selectively retrieve information from the entire previous context\. However, this explicit pairwise comparison leads to quadratic cost in the sequence length during training\. Linear\-attention variants replace the explicit attention matrix with recurrent memory states that can be updated incrementally\.
### 3\.2DeltaNet
DeltaNet replaces naive additive linear attention with an error\-correcting delta rule\[[2](https://arxiv.org/html/2607.07953#bib.bib2)\]\. Using the memory predictionv¯\(i\)\\bar\{v\}^\{\(i\)\}and residualr\(i\)r^\{\(i\)\}from Eqs\. \([9](https://arxiv.org/html/2607.07953#S3.E9)\) and \([10](https://arxiv.org/html/2607.07953#S3.E10)\), the memory is updated by writing only this residual:
W\(i\)=W\(i−1\)\+β\(i\)r\(i\)⊗κ\(i\),W^\{\(i\)\}=W^\{\(i\-1\)\}\+\\beta^\{\(i\)\}r^\{\(i\)\}\\otimes\\kappa^\{\(i\)\},\(14\)and the output is computed using the shared read rule in Eq\. \([8](https://arxiv.org/html/2607.07953#S3.E8)\)\.
The key idea is that the model does not simply add the new value to memory\. Instead, it first asks what the memory already predicts for the current key, and then writes the correction needed to move the stored association towardv\(i\)v^\{\(i\)\}\. This makes the update selective and key\-specific\.
The main advantage of DeltaNet is that it improves over naive additive storage by using an error\-correcting update\. Its limitation is that it has no explicit mechanism for globally clearing stale information\. As interference accumulates, the model can correct individual associations, but it cannot directly decay the previous memory state in a coarse way\.
### 3\.3Gated DeltaNet
Gated DeltaNet augments DeltaNet with a learned scalar forgetting mechanism\[[3](https://arxiv.org/html/2607.07953#bib.bib3)\]\. Before computing the delta\-rule residual, the previous memory is decayed by the token\-dependent scalar gateα\(i\)\\alpha^\{\(i\)\}from Eq\. \([11](https://arxiv.org/html/2607.07953#S3.E11)\)\. The decay\-adjusted prediction is
v¯α\(i\)=α\(i\)W\(i−1\)κ\(i\)=α\(i\)v¯\(i\),\\bar\{v\}\_\{\\alpha\}^\{\(i\)\}=\\alpha^\{\(i\)\}W^\{\(i\-1\)\}\\kappa^\{\(i\)\}=\\alpha^\{\(i\)\}\\bar\{v\}^\{\(i\)\},\(15\)and the corresponding residual is
rα\(i\)=v\(i\)−v¯α\(i\)\.r\_\{\\alpha\}^\{\(i\)\}=v^\{\(i\)\}\-\\bar\{v\}\_\{\\alpha\}^\{\(i\)\}\.\(16\)The memory update is then
W\(i\)=α\(i\)W\(i−1\)\+β\(i\)rα\(i\)⊗κ\(i\),W^\{\(i\)\}=\\alpha^\{\(i\)\}W^\{\(i\-1\)\}\+\\beta^\{\(i\)\}r\_\{\\alpha\}^\{\(i\)\}\\otimes\\kappa^\{\(i\)\},\(17\)with output again computed by Eq\. \([8](https://arxiv.org/html/2607.07953#S3.E8)\)\.
Gated DeltaNet preserves the delta\-rule correction, but applies it relative to a decayed version of the previous memory\. This gives the architecture an explicit way to forget information, which can help reduce interference in long or cluttered contexts\.
The trade\-off is that the forgetting operation is state\-wide\. DeltaNet is purely selective in the sense that it updates memory through a key\-specific residual\. Gated DeltaNet keeps this corrective write, but also introduces the coarse decay termα\(i\)W\(i−1\)\\alpha^\{\(i\)\}W^\{\(i\-1\)\}\. As a result, it gains a direct mechanism for memory clearance, but partially sacrifices the strictly selective character of the original DeltaNet update\.
### 3\.4Kimi Delta Attention
Kimi Delta Attention keeps the gated delta\-rule structure but replaces scalar forgetting with channel\-wise forgetting\[[4](https://arxiv.org/html/2607.07953#bib.bib4)\]\. Instead of applying the same decay factor to the entire memory, the model uses the vector\-valued gate𝜶\(i\)\\bm\{\\alpha\}^\{\(i\)\}from Eq\. \([12](https://arxiv.org/html/2607.07953#S3.E12)\) to decay different transformed\-key dimensions at different rates\.
The channel\-wise decay\-adjusted prediction is
v¯𝜶\(i\)=W\(i−1\)Dα\(i\)κ\(i\),\\bar\{v\}\_\{\\bm\{\\alpha\}\}^\{\(i\)\}=W^\{\(i\-1\)\}D\_\{\\alpha\}^\{\(i\)\}\\kappa^\{\(i\)\},\(18\)with residual
r𝜶\(i\)=v\(i\)−v¯𝜶\(i\)\.r\_\{\\bm\{\\alpha\}\}^\{\(i\)\}=v^\{\(i\)\}\-\\bar\{v\}\_\{\\bm\{\\alpha\}\}^\{\(i\)\}\.\(19\)The update becomes
W\(i\)=W\(i−1\)Dα\(i\)\+β\(i\)r𝜶\(i\)⊗κ\(i\),W^\{\(i\)\}=W^\{\(i\-1\)\}D\_\{\\alpha\}^\{\(i\)\}\+\\beta^\{\(i\)\}r\_\{\\bm\{\\alpha\}\}^\{\(i\)\}\\otimes\\kappa^\{\(i\)\},\(20\)and the output is computed by Eq\. \([8](https://arxiv.org/html/2607.07953#S3.E8)\)\.
BecauseWWmaps transformed\-key features to values, the diagonal decay matrix multiplies on the right ofWWand acts along the key\-feature dimension\. This mechanism can be viewed as a more fine\-grained version of Gated DeltaNet\. Rather than decaying the whole memory with a single scalar, Kimi Delta Attention can preserve some feature dimensions while forgetting others more aggressively\. If the vector gate collapses to a scalar gate,
𝜶\(i\)=α\(i\)𝟏dϕ,\\bm\{\\alpha\}^\{\(i\)\}=\\alpha^\{\(i\)\}\\mathbf\{1\}\_\{d\_\{\\phi\}\},then the update reduces to the scalar\-forgetting form used by Gated DeltaNet\.
The advantage of this formulation is that it increases the granularity of memory control\. Relative to Gated DeltaNet, it can distinguish between feature subspaces that should be retained and feature subspaces that should be cleared\. However, the active delta\-rule edit is still controlled by a single scalarβ\(i\)\\beta^\{\(i\)\}\. The same gate controls how much old content is removed at the current key and how much new value is written into memory\.
### 3\.5Gated DeltaNet\-2
Gated DeltaNet\-2 extends Kimi Delta Attention by decoupling the scalar delta gate into the channel\-wise erase gate𝒃\(i\)\\bm\{b\}^\{\(i\)\}and channel\-wise write gate𝒘\(i\)\\bm\{w\}^\{\(i\)\}from Eq\. \([13](https://arxiv.org/html/2607.07953#S3.E13)\)\[[5](https://arxiv.org/html/2607.07953#bib.bib5)\]\. Instead of using the same scalarβ\(i\)\\beta^\{\(i\)\}to control both removal of old content and insertion of new content,𝒃\(i\)\\bm\{b\}^\{\(i\)\}controls the key\-side erase operation, while𝒘\(i\)\\bm\{w\}^\{\(i\)\}controls the value\-side write operation\.
Using the channel\-wise decay matrix from Eq\. \([12](https://arxiv.org/html/2607.07953#S3.E12)\), the decay\-adjusted memory is
W~\(i−1\)=W\(i−1\)Dα\(i\)\.\\widetilde\{W\}^\{\(i\-1\)\}=W^\{\(i\-1\)\}D\_\{\\alpha\}^\{\(i\)\}\.\(21\)Gated DeltaNet\-2 then defines a gated erase direction and a gated write target:
e\(i\)=𝒃\(i\)⊙κ\(i\),z\(i\)=𝒘\(i\)⊙v\(i\)\.e^\{\(i\)\}=\\bm\{b\}^\{\(i\)\}\\odot\\kappa^\{\(i\)\},\\qquad z^\{\(i\)\}=\\bm\{w\}^\{\(i\)\}\\odot v^\{\(i\)\}\.\(22\)The residual written to memory is
rGDN2\(i\)=z\(i\)−W~\(i−1\)e\(i\)\.r\_\{\\mathrm\{GDN2\}\}^\{\(i\)\}=z^\{\(i\)\}\-\\widetilde\{W\}^\{\(i\-1\)\}e^\{\(i\)\}\.\(23\)The memory update is
W\(i\)=W~\(i−1\)\+rGDN2\(i\)⊗κ\(i\),W^\{\(i\)\}=\\widetilde\{W\}^\{\(i\-1\)\}\+r\_\{\\mathrm\{GDN2\}\}^\{\(i\)\}\\otimes\\kappa^\{\(i\)\},\(24\)and the output is computed by Eq\. \([8](https://arxiv.org/html/2607.07953#S3.E8)\)\.
Thus, Gated DeltaNet\-2 preserves the channel\-wise decay of Kimi Delta Attention, but makes the active delta update more flexible\. The erase gate decides which key\-feature channels should be removed from the previous memory, while the write gate decides which value channels should be stored\. If
𝒃\(i\)=β\(i\)𝟏dϕ,𝒘\(i\)=β\(i\)𝟏dv,\\bm\{b\}^\{\(i\)\}=\\beta^\{\(i\)\}\\mathbf\{1\}\_\{d\_\{\\phi\}\},\\qquad\\bm\{w\}^\{\(i\)\}=\\beta^\{\(i\)\}\\mathbf\{1\}\_\{d\_\{v\}\},then the update reduces to Kimi Delta Attention\. If the decay gate also collapses to
𝜶\(i\)=α\(i\)𝟏dϕ,\\bm\{\\alpha\}^\{\(i\)\}=\\alpha^\{\(i\)\}\\mathbf\{1\}\_\{d\_\{\\phi\}\},then it further reduces to Gated DeltaNet\.
## 4Cross\-Layer Routing
Deep stacks can dilute information extracted at lower layers, and cross\-layer schemes such as Attention Residuals and Mixture\-of\-Depths Attention\[[6](https://arxiv.org/html/2607.07953#bib.bib6),[7](https://arxiv.org/html/2607.07953#bib.bib7)\]address this with explicit depth\-wise pathways\. We pursue the same goal for DeltaNet\-style associative memories, but without a new depth\-attention operator: we ask whether an internal signal already produced inside the delta\-rule update can be reused across depth as a lightweight, linear\-time pathway\. We explore two axes:*what*to route \(the layer’s write error or its write value\) and*where*to inject it \(the next layer’s value target or the shared residual stream\)\. We use three names throughout:*Cross\-Layer Error Residuals \(CLER\)*is the initial value\-target error\-routing formulation;*CLER\-H*routes the same write error into the shared hidden stream; and*Cross\-Layer Value Routing \(CLVR\)*routes the write value into the shared hidden stream\. This keeps the ablation order explicit: CLER changes neither signal nor space, CLER\-H fixes the injection space, and CLVR keeps the hidden\-stream injection while changing the routed signal\.
All routing variants share the same footprint\. They augment existing DeltaNet variants with a side\-channel that carries an internal signal from one recurrent\-memory layer to a higher one, and they leave the underlying mixer, the recurrent update rule, and the gating unchanged\. They differ only in which internal signal is carried and where it is added, which lets matched experiments isolate the effect of each choice rather than confounding it with a change to the host architecture\.
In this section, a*routing\-capable layer*means a DeltaNet\-style recurrent\-memory layer that both produces a delta\-rule write residual and can receive a routed residual from a lower such layer\. Softmax\-attention layers are not routing\-capable in the reported hybrid stacks: they may pass the side\-channel forward, but they do not produce or consume routed signals\. The notation in this section switches from the generic token indexiito\(l,t\)\(l,t\)in order to track both layer depthlland sequence positiontt\.
### 4\.1Delta\-rule write quantities
In all cross\-layer routing variants we implement, each DeltaNet or Gated DeltaNet layer maintains a recurrent associative memoryWl,t∈ℝdv×dϕW\_\{l,t\}\\in\\mathbb\{R\}^\{d\_\{v\}\\times d\_\{\\phi\}\}that predicts the current value from the current key\. For a layerlland sequence positiontt, the model first computes a feature\-mapped key and a prediction from the previous memory state:
κl,t=ϕ\(kl,t\)∈ℝdϕ,v¯l,t=Wl,t−1κl,t∈ℝdv,\\kappa\_\{l,t\}=\\phi\(k\_\{l,t\}\)\\in\\mathbb\{R\}^\{d\_\{\\phi\}\},\\qquad\\bar\{v\}\_\{l,t\}=W\_\{l,t\-1\}\\kappa\_\{l,t\}\\in\\mathbb\{R\}^\{d\_\{v\}\},\(25\)whereϕ\(⋅\)\\phi\(\\cdot\)denotes the feature map used by the linear attention rule\. The layer then computes a write residual:
rl,t=vl,t−v¯l,t∈ℝdv,r\_\{l,t\}=v\_\{l,t\}\-\\bar\{v\}\_\{l,t\}\\in\\mathbb\{R\}^\{d\_\{v\}\},\(26\)which represents the correction that must be written into memory\. These two quantities, the write residualrl,tr\_\{l,t\}\(the part of the value the memory has not yet absorbed\) and the write valuevl,tv\_\{l,t\}itself, are the internal signals carried across depth: CLER and CLER\-H use the residual, while CLVR uses the value\.
### 4\.2Cross\-Layer Error Residuals \(CLER\)
The direct way to share this correction across depth, inspired by DeltaNet\-style architectures, is to forward the write error itself\. This is the*Cross\-Layer Error Residuals \(CLER\)*formulation \([Figure˜1](https://arxiv.org/html/2607.07953#S4.F1)\): it injects the residual from the nearest lower routing\-capable layer into the current layer’s value target before the new residual is computed:
v~l,t=vl,t\+Γlρ\(rp\(l\),t\),\\tilde\{v\}\_\{l,t\}=v\_\{l,t\}\+\\Gamma\_\{l\}\\rho\(r\_\{p\(l\),t\}\),\(27\)wherep\(l\)p\(l\)denotes the nearest lower DeltaNet/Gated DeltaNet layer that produced a CLER residual,Γl∈ℝ\\Gamma\_\{l\}\\in\\mathbb\{R\}is a learned per\-layer scalar, andρ:ℝdv→ℝdv\\rho:\\mathbb\{R\}^\{d\_\{v\}\}\\to\\mathbb\{R\}^\{d\_\{v\}\}is a residual normalization taken as the identity in our main comparisons\. The current layer then computes:
rl,t=v~l,t−Wl,t−1κl,t\.r\_\{l,t\}=\\tilde\{v\}\_\{l,t\}\-W\_\{l,t\-1\}\\kappa\_\{l,t\}\.\(28\)
Figure 1:Cross\-Layer Error Residuals \(CLER\)\. A lower routing\-capable recurrent\-memory layer computes a DeltaNet\-style write residual\. If softmax\-attention layers intervene, the residual is carried forward as a side\-channel\. The next routing\-capable layer scales the routed residual and injects it into its value target before computing its own write residual\.CLER is a side\-channel rather than a new mixer: the recurrent update, gating, and output read are unchanged, and in hybrid stacks the residual is carried through intervening softmax layers to the next routing\-capable layer\.
### 4\.3Cross\-Layer Value Routing \(CLVR\)
As reported in Section[6](https://arxiv.org/html/2607.07953#S6), the value\-target injection in Eq\. \([27](https://arxiv.org/html/2607.07953#S4.E27)\) does not improve over matched baselines\. We attribute this to a space mismatch: the routed residual stays in the receiver’s independently\-learned value space, where it is generally misaligned with the higher layer’s value geometry and competes with the receiver’s own write target\. We therefore make two changes: route into the*shared hidden stream*rather than the per\-layer value space, and treat the routed quantity as a free choice between the delta\-rule error and the write value \([Figure˜2](https://arxiv.org/html/2607.07953#S4.F2)\)\.
Concretely, for a routing\-capable layerllwe project an internal signalsl,ts\_\{l,t\}to the model dimension and add it to the residual stream,
εl,t=Plsl,t,hl,t←hl,t\+εl,t,\\varepsilon\_\{l,t\}=P\_\{l\}\\,s\_\{l,t\},\\qquad h\_\{l,t\}\\leftarrow h\_\{l,t\}\+\\varepsilon\_\{l,t\},\(29\)wheresl,t∈ℝdvs\_\{l,t\}\\in\\mathbb\{R\}^\{d\_\{v\}\},hl,t∈ℝdmodelh\_\{l,t\}\\in\\mathbb\{R\}^\{d\_\{\\mathrm\{model\}\}\}is the ordinary residual\-stream state after layerll, andPl∈ℝdmodel×dvP\_\{l\}\\in\\mathbb\{R\}^\{d\_\{\\mathrm\{model\}\}\\times d\_\{v\}\}is a per\-layer projection that is*zero\-initialized*, optionally factored as a low\-rank productPl=UlDlP\_\{l\}=U\_\{l\}D\_\{l\}withUl∈ℝdmodel×dpU\_\{l\}\\in\\mathbb\{R\}^\{d\_\{\\mathrm\{model\}\}\\times d\_\{p\}\}andDl∈ℝdp×dvD\_\{l\}\\in\\mathbb\{R\}^\{d\_\{p\}\\times d\_\{v\}\}\. Zero initialization makes the routed contribution vanish at the start of training, so the model begins exactly at the host baseline and learns whether and how strongly to route; the injectedεl,t∈ℝdmodel\\varepsilon\_\{l,t\}\\in\\mathbb\{R\}^\{d\_\{\\mathrm\{model\}\}\}then reaches every later layer and the output head through the standard residual path\. Unlike Eq\. \([27](https://arxiv.org/html/2607.07953#S4.E27)\), the signal is added in a space that is shared, and therefore aligned, across depth\.
Figure 2:Cross\-Layer Value Routing \(CLVR\)\. Each routing\-capable layer projects its internal write value into the shared residual stream through a zero\-initialized projectionPlP\_\{l\}; the routed contribution starts at zero and is added in a depth\-aligned space, where it is read by all later layers and the output head\.We consider two choices for the routed signalsl,ts\_\{l,t\}\. The first is the write*error*sl,t=rl,ts\_\{l,t\}=r\_\{l,t\}, which preserves CLER’s original motivation of forwarding what a memory failed to store; we call this hidden\-stream ablation*CLER\-H*\. The second is the write*value*sl,t=vl,ts\_\{l,t\}=v\_\{l,t\}, the target the memory attempts to store before its own read is subtracted; we call this variant*Cross\-Layer Value Routing \(CLVR\)*\. Sincevl,t=rl,t\+v¯l,tv\_\{l,t\}=r\_\{l,t\}\+\\bar\{v\}\_\{l,t\}, the two signals differ only by the memory read and occupy the same value space, soPlP\_\{l\}has identical shape in both cases and the variants are exactly parameter\-matched at every rank\. The error\-versus\-value comparison therefore isolates the*content*of the routed signal rather than its parameter budget\.
CLVR is related to, but distinct from, two existing lines of work\. Value\-residual learning for softmax transformers\[[8](https://arxiv.org/html/2607.07953#bib.bib8)\]adds a fixed early\-layer value back into later layers’ attention values to counter attention concentration; CLVR instead routes*each*linear\-memory layer’s own internal write value into the aligned residual stream through a learned, zero\-initialized projection, and is defined for delta\-rule memories rather than softmax attention\. Attention Residuals\[[6](https://arxiv.org/html/2607.07953#bib.bib6)\]and related depth\-attention schemes route layer*outputs*by a learned softmax over depth and replace the residual sum; CLVR routes a signal taken from*inside*the recurrent operator and adds it rather than replacing the stream\. To our knowledge, routing the internal write value of a linear\-attention/delta\-rule memory into the shared residual stream has not been studied previously\.
## 5Experimental Setup
We keep the data, tokenizer, sequence length, global batch size, precision, and hardware fixed across the main 350M comparisons, while allowing architecture\-specific layer counts and stack patterns needed to target the 350M\-parameter class\. Under this setup, differences in validation loss and throughput can be interpreted primarily as architecture, optimizer, or stack\-composition effects rather than as changes in the data pipeline\. For the main quantitative claims, we require the corresponding run record to identify the model scale, token budget, stack composition, optimizer, learning rate, metric definition, and checkpoint semantics\. Exploratory variants with incomplete records are discussed separately as design directions rather than as settled quantitative findings\. Appendix[B](https://arxiv.org/html/2607.07953#A2)summarizes these inclusion criteria for the supplementary material\.
### 5\.1Shared Training Setup
Following the experimental setting used by Gated DeltaNet and Gated DeltaNet\-2\[[3](https://arxiv.org/html/2607.07953#bib.bib3),[5](https://arxiv.org/html/2607.07953#bib.bib5)\], the experiments use decoder\-only language models in the 350M\-parameter class, trained on FineWeb\-Edu\[[9](https://arxiv.org/html/2607.07953#bib.bib9)\]with the LLaMA2 tokenizer\[[10](https://arxiv.org/html/2607.07953#bib.bib10)\]\. The controlled routing ablations use a 20\-layer model with hidden size 1024, feed\-forward dimension 2816, sequence length 4096, global batch size 128, bf16 precision, and one GH200 node with four GPUs on the CSCS Alps system\. The broader architecture comparison keeps this data, tokenizer, sequence length, batch, precision, and hardware recipe fixed while allowing architecture\-specific layer counts, evaluating softmax attention, DeltaNet, Gated DeltaNet, Kimi Delta Attention, and Gated DeltaNet\-2 under matched training runs up to 15B tokens\.
We distinguish between*hybrid*stacks, where linear\-attention layers are interleaved with softmax\-attention layers, and*pure*stacks, where every mixer layer uses the corresponding linear\-attention rule\. The default DeltaNet, Gated DeltaNet, and Gated DeltaNet\-2 hybrid configurations set one softmax\-attention layer every three layers, giving a 2:1 linear\-to\-softmax pattern\.111For the 20\-layer runs, this corresponds to linear\-memory mixers at layers\{0,1,3,4,…,18,19\}\\\{0,1,3,4,\\ldots,18,19\\\}and softmax\-attention mixers at layers\{2,5,…,17\}\\\{2,5,\\ldots,17\\\}\. The 22\- and 24\-layer stacks follow the same every\-third\-layer rule\. The separate Gated DeltaNet\-2 full\-attention 3:1 variants set one softmax\-attention layer every four layers, giving repeated blocks of three Gated DeltaNet\-2 layers followed by one full self\-attention layer\.This distinction is central to the empirical results: hybrid stacks often improve loss, while pure stacks expose the throughput advantage of recurrent linear memories\.
Table 1:Core configuration information for the main experiments\. Values are reported from the local launchers and run records when available; entries marked as not fully recorded are not inferred\.#### Metric definitions\.
All validation losses are held\-out language\-model cross\-entropies, so lower is better\. In the 15B\-token WSD runs,*saturation loss*is the validation loss recorded near the end of the high\-learning\-rate/saturation phase before the final decay, while*final loss*is the validation loss at the recorded final checkpoint after the full schedule\.*Relative speed*is normalized training throughput within the 350M\-parameter, 15B\-token sweep, with pure Gated DeltaNet under AdamW set to 100%; it is not an inference\-throughput measurement\. In the supplementary 1B\-token table,*ktok/s/GPU*reports thousands of training tokens processed per second per GPU, without normalization\.
### 5\.2Optimizer, Learning\-Rate, and Iteration\-Time Scaling
The 350M\-parameter, 15B\-token comparison evaluates AdamW\[[11](https://arxiv.org/html/2607.07953#bib.bib11)\]and Muon\[[12](https://arxiv.org/html/2607.07953#bib.bib12)\]\. The learning\-rate ablation trains 350M hybrid models for 2000 steps, or approximately 1\.05B tokens, and shows that the best learning rate depends strongly on both optimizer and mixer\. In those runs, DeltaNet and Gated DeltaNet with AdamW prefer a learning rate near10−310^\{\-3\}, while the remaining AdamW/Muon combinations cluster around3×10−43\\times 10^\{\-4\}\. We therefore interpret optimizer effects together with learning\-rate sensitivity rather than as a single optimizer substitution\.
The sequence\-length scaling of iteration time is measured at 4k, 16k, and 32k tokens\. It is intended to characterize the expected asymptotic behavior of the sequence mixers rather than to replace validation loss as the primary quality metric\. Relative speeds in Table[2](https://arxiv.org/html/2607.07953#S6.T2)are normalized to the fastest 350M\-parameter, 15B\-token entry, Gated DeltaNet with AdamW in a pure stack\.
### 5\.3Cross\-Layer Routing: Experimental Setup
Every routing run is compared against a matched non\-routing baseline that shares its host architecture, optimizer, schedule, and seed: a routed DeltaNet against DeltaNet, and a routed Gated DeltaNet against Gated DeltaNet\. The initial CLER ablations \(Table[6](https://arxiv.org/html/2607.07953#S6.T6)\) use a learned per\-layer scalar coefficientΓ\\Gammain Eq\. \([27](https://arxiv.org/html/2607.07953#S4.E27)\) and raw routed residuals, i\.e\.ρ\\rhois the identity map\. Residual\-normalized configurations exist as exploratory variants but are not used as main quantitative evidence unless they meet the same matched\-run record standard as the reported ablations\.
For the refined routing study of Section[4\.3](https://arxiv.org/html/2607.07953#S4.SS3)we keep the same 350M recipe and add a larger Gated DeltaNet configuration, so that the reported comparison spans both model and token scale without implying a complete larger\-scale sweep\. The matched routing comparisons in Table[7](https://arxiv.org/html/2607.07953#S6.T7)include both DeltaNet and Gated DeltaNet at 350M parameters / 1B tokens and at 350M parameters / 15B tokens\. At 1\.3B parameters / 40B tokens, the reported hidden\-stream routing comparison is available only for the Gated DeltaNet host\. Each routing run is paired with a non\-routing baseline that shares its architecture, optimizer, schedule, and seed\. We did not perform repeated\-seed runs for these comparisons, so we report no standard deviations\. Within this study we evaluate three routing pathways under the Muon optimizer: the initial CLER value\-target injection \(Eq\. \([27](https://arxiv.org/html/2607.07953#S4.E27)\)\), CLER\-H \(sl,t=rl,ts\_\{l,t\}=r\_\{l,t\}in Eq\. \([29](https://arxiv.org/html/2607.07953#S4.E29)\)\), and CLVR \(sl,t=vl,ts\_\{l,t\}=v\_\{l,t\}\)\. Because the routing projections are zero\-initialized, each routing run starts from exactly its baseline; consistent with the rest of the report we compare the*final*validation loss at the end of training rather than the best intermediate value, since the latter is dominated by evaluation noise\. We note that the 15B\- and 40B\-token runs read from a token\-count \(\_tc\) slice of FineWeb\-Edu that differs from the slice used for the 1B\-token ablations; absolute losses are therefore comparable only within a fixed token budget and host architecture, and all reported routing gains are differences against the matched baseline at the same scale\.
We evaluate cross\-layer routing only on DeltaNet and Gated DeltaNet hosts, so no empirical claim is made about these routing variants on top of Kimi Delta Attention or Gated DeltaNet\-2\.
## 6Results
This section reports the empirical findings in increasing order of scope: 350M\-parameter models trained for 15B tokens, optimizer and learning\-rate behavior, sequence\-length scaling of training iteration time, larger DeltaNet runs, downstream evaluations, and finally validation and residual analysis for cross\-layer routing\.
All reported architecture and routing comparisons are single\-run comparisons; we did not perform repeated\-seed runs and therefore do not report standard deviations\. We therefore abstain from drawing strong conclusions from small validation\-loss differences, especially when differences are below roughly10−310^\{\-3\}to10−210^\{\-2\}and are not supported by matched settings, consistent optimizer trends, or downstream agreement\. Larger effects, such as the 32k iteration\-time gap between softmax attention and pure recurrent stacks, the consistent Muon improvements in matched 350M\-parameter, 15B\-token rows, and the best final\-loss entry in Table[2](https://arxiv.org/html/2607.07953#S6.T2), remain meaningful within the reported configuration\.
### 6\.1350M\-Parameter, 15B\-Token Validation\-Loss/Throughput Frontier
Figure[3](https://arxiv.org/html/2607.07953#S6.F3)and Table[2](https://arxiv.org/html/2607.07953#S6.T2)summarize the 350M\-parameter runs trained for 15B tokens\. The best final validation loss among these reported runs is Kimi Delta Attention with Muon in a hybrid stack, which reaches 2\.273\. DeltaNet with Muon in a hybrid stack is the strongest non\-KDA entry at 2\.299, while Gated DeltaNet\-2 with Muon in a hybrid stack reaches 2\.345\. The fastest entry is pure Gated DeltaNet with AdamW, normalized to 100% relative speed, but this setting has a substantially higher final loss of 2\.433 than the best hybrid Muon runs\.
Two patterns are consistent across the table\. First, Muon improves the matched final loss for every architecture family shown here\. Second, hybrid stacks usually improve loss relative to pure stacks, but they reduce part of the speed advantage that motivates linear attention\. The resulting trade\-off is visible in the reported sweep: the lowest loss comes from a slower KDA hybrid run, while the fastest pure recurrent stack is not the strongest model by validation loss\.
Figure 3:Validation\-loss scaling for the 350M\-parameter runs trained to 15B tokens\. Lower is better\. In this sweep, Kimi Delta Attention with Muon in a hybrid stack reaches the lowest final loss, while DeltaNet and Gated DeltaNet remain competitive with simpler recurrent\-memory updates\.Table 2:350M\-parameter, 15B\-token architecture comparison\. Saturation loss is measured before the final WSD decay, final loss at the final recorded checkpoint, and relative speed is normalized training throughput against the fastest entry in this sweep, pure Gated DeltaNet with AdamW\. Lower losses are better; higher relative speed is faster\. Values are directly comparable within this sweep, but each row is a single run rather than a repeated\-seed estimate\.
### 6\.2Learning\-Rate and Optimizer Effects
Figure[4](https://arxiv.org/html/2607.07953#S6.F4)shows that the optimizer comparison is not separable from learning\-rate choice\. At the 350M\-parameter hybrid setting, Muon prefers a lower learning rate around3×10−43\\times 10^\{\-4\}\(with10−410^\{\-4\}being close/catching up\), whereas all AdamW linear attention variants prefer10−310^\{\-3\}\. The exception is standard softmax attention, which matches its Muon counterpart with3×10−43\\times 10^\{\-4\}\. This shows that a single default learning rate can distort architecture comparisons: a variant may underperform because its optimizer/learning\-rate pair is poorly matched, rather than because the recurrent rule itself is inferior\. The iso plots in[Figure˜5](https://arxiv.org/html/2607.07953#S6.F5)visualize the same runs, showing the final\-loss interpolation line for different learning rates\. It demonstrates the plateau length in which performance is close to optimal: For Muon \(the solid lines\), the plateau is slightly longer and on lower learning rates\. Additionally, softmax is the most dependent on a good learning rate choice, whereas linear attention architectures are more forgiving, especially on higher learning rates\.
These results indicate that Muon should be treated as part of the architecture evaluation protocol rather than as an optimizer detail independent of the model family\. In Table[2](https://arxiv.org/html/2607.07953#S6.T2), Muon improves final loss for every matched architecture/stack pair, but the magnitude of the improvement varies across softmax attention, DeltaNet, Gated DeltaNet, Kimi Delta Attention, and Gated DeltaNet\-2\.
Figure 4:Learning\-rate ablation for 350M\-parameter hybrid models trained for 2000 steps, approximately 1\.05B tokens\. The preferred learning rate varies by optimizer and mixer, making learning\-rate tuning necessary before drawing architectural conclusions\.Figure 5:Learning\-rate ablation iso plots for 350M\-parameter hybrid models trained for 2000 steps\. The iso plots show the final loss interpolation line for different learning rates\. The loss increasing on the left and right ends for all attention/optimizer combination implies the covered range is sufficient, reaching from a too\-low learning rate over the optimum up to a too\-high learning rate\. The optimum lies between3×10−43\\times 10^\{\-4\}and10−310^\{\-3\}depending on attention/optimizer\.
### 6\.3Sequence\-Length Iteration\-Time Scaling
Figure[6](https://arxiv.org/html/2607.07953#S6.F6)measures iteration time as the sequence length increases from 4k to 32k tokens\. At 32k tokens, the timing measurement is 3\.37 seconds per iteration for softmax attention, 1\.56 seconds for a Gated DeltaNet hybrid stack, and 0\.96 seconds for a pure Gated DeltaNet stack\. From 4k to 32k tokens, the corresponding growth factors are approximately2\.9×2\.9\\times,1\.7×1\.7\\times, and1\.1×1\.1\\times\. This provides direct empirical evidence for the expected scaling advantage of recurrent linear\-memory mixers, especially when the stack is pure\.
The timing result also clarifies the cost of hybridization\. Hybrid stacks often recover validation loss relative to pure linear stacks, but the retained softmax layers reintroduce part of the sequence\-length cost\. The preferred stack therefore depends on whether the target setting prioritizes lower loss at moderate context length or flatter iteration\-time scaling at longer context length\.
Figure 6:Iteration\-time scaling from 4k to 32k tokens\. Softmax attention time grows much faster with sequence length \(\+192%\+192\\%\) than Gated DeltaNet hybrid \(\+65%\+65\\%\) and pure stacks \(\+8%\+8\\%\)\. Note that hybrid iteration time increase is very close to1/31/3of softmax, matching the fact that hybrid has1/31/3softmax layers\. Error bars \(step\-to\-step variance\) are very low\.
### 6\.4Larger DeltaNet Runs and Downstream Evaluation
Table[3](https://arxiv.org/html/2607.07953#S6.T3)extends the DeltaNet comparison beyond the 350M\-parameter scale\. At 1\.3B parameters and 40B tokens, the best listed final loss is 2\.063 for a pure DeltaNet run at learning rate1\.5×10−41\.5\\times 10^\{\-4\}, while the corresponding pure CLER\-DeltaNet run at the same learning rate reaches 2\.112\. At 3B parameters and 60B tokens, the two better DeltaNet hybrid runs reach 1\.955 at learning rates3×10−43\\times 10^\{\-4\}and1\.5×10−41\.5\\times 10^\{\-4\}, while the run at5×10−45\\times 10^\{\-4\}has a substantially higher final loss of 2\.332\. These runs reinforce the learning\-rate sensitivity seen at smaller scale\.
Table 3:Larger DeltaNet\-only runs at 1\.3B\-parameter/40B\-token and 3B\-parameter/60B\-token scale\. Lower final loss is better\. These rows illustrate larger\-scale DeltaNet sensitivity to learning rate and stack pattern, but they are not a full architecture sweep across all mechanisms\.RunLearning rateScaleFinal lossdn\-hybrid3\.6e\-41\.3B/40B2\.066dn\-pure3\.6e\-41\.3B/40B2\.085dn\-hybrid1\.5e\-41\.3B/40B2\.109dn\-pure1\.5e\-41\.3B/40B2\.063dn\-pure\-cler1\.5e\-41\.3B/40B2\.112dn\-hybrid5e\-43B/60B2\.332dn\-hybrid3e\-43B/60B1\.955dn\-hybrid1\.5e\-43B/60B1\.955Table 4:Downstream checks on HellaSwag, PIQA, and WinoGrande using the local lm\-eval\-harness API path\. Accuracy is exact\-match multiple\-choice accuracy; normalized accuracy is the harness’s length\-normalized likelihood accuracy where produced\.HellaSwagPIQAWinoGrandeSizeArch\.VariantOptim\.Iter\.Tokensaccacc\_normaccacc\_normacc350MGDNhybridMuon2861015\.0B0\.34800\.41580\.66760\.67410\.5414350MGDNpureMuon2861015\.0B0\.33370\.39780\.67630\.67410\.4870350MDeltaNethybridMuon2861015\.0B0\.35410\.43050\.66810\.67790\.5051350MDeltaNetpureMuon2861015\.0B0\.34260\.41330\.67410\.68010\.5170350MGDNhybridAdamW2861015\.0B0\.33690\.39920\.66160\.66160\.5099350MGDNpureAdamW2861015\.0B0\.32650\.38170\.65610\.65720\.5217350MDeltaNethybridAdamW2861015\.0B0\.33610\.39870\.66540\.65130\.5249350MDeltaNetpureAdamW2861015\.0B0\.32930\.39180\.64800\.65510\.51541\.3BDeltaNethybridMuon1907340\.0B0\.42770\.54840\.72090\.73010\.55721\.3BDeltaNetpure, no CLERMuon1907340\.0B0\.41190\.52340\.71060\.71930\.56121\.3BCLER\-DeltaNetpure,γ=0\.1\\gamma\{=\}0\.1Muon1907340\.0B0\.40490\.52310\.70890\.72030\.54461\.3BCLER\-DeltaNetpure,γ=0\.0\\gamma\{=\}0\.0Muon1907340\.0B0\.40780\.52200\.70950\.71600\.56273BDeltaNethybridMuon715260\.0B0\.46170\.60630\.73340\.74100\.5848
Table[5](https://arxiv.org/html/2607.07953#S6.T5)reports the matched 350M\-parameter, 15B\-token Muon downstream evaluations for the CLER\-H and CLVR checkpoints\. The routing variants are broadly comparable to their matched Gated DeltaNet and DeltaNet baselines on HellaSwag and PIQA, with small mixed differences across accuracy and normalized accuracy\. Although the WinoGrande point estimates are higher for the routed variants, these task\-specific single\-checkpoint differences are not sufficiently reliable to serve as evidence of a routing effect\. We therefore use the downstream evaluations only as a check for clear degradation, which we do not observe on the evaluated tasks\.
Table 5:Matched 350M\-parameter, 15B\-token Muon downstream evaluations for Gated DeltaNet and DeltaNet routing checkpoints\. All rows use the scoring\-server lm\-eval\-harness path with full\-sequence forward passes, the LLaMA2 tokenizer, batch size 8, maximum length 4096, and zero\-shot evaluation\.
### 6\.5CLER Validation Loss
Table[6](https://arxiv.org/html/2607.07953#S6.T6)summarizes the matched CLER comparisons at the 350M\-parameter, 1B\-token scale\. Under Muon, CLER\-Gated has slightly higher final loss than Gated DeltaNet, while CLER\-DeltaNet obtains a very small improvement over DeltaNet; the latter difference is below10−310^\{\-3\}and comes from a single run, so we do not interpret it as a robust gain\. Under AdamW, CLER does not improve either host\. The pathway is not inactive \(the routed residuals are nonzero at every routing\-capable layer\), but the matched trajectories are nearly overlapping, consistent with the space\-mismatch argument of Section[4\.3](https://arxiv.org/html/2607.07953#S4.SS3): the error routed into the per\-layer value target appears either insufficiently influential or poorly aligned with the receiver’s learned dynamics\.
Table 6:Matched CLER comparisons at 350M parameters and approximately 1B training tokens\.Δ\\Deltais computed relative to the corresponding non\-CLER baseline under the same optimizer; negative values indicate lower final validation loss\. The small Muon CLER\-DeltaNet gain is below10−310^\{\-3\}and is not interpreted as a robust improvement\.
### 6\.6Cross\-Layer Value Routing Results
The negative result above is consistent with the original injection adding a small, and possibly misaligned, signal in the wrong space\. Section[4\.3](https://arxiv.org/html/2607.07953#S4.SS3)addresses both concerns at once: it routes into the shared hidden stream through a zero\-initialized projection \(Eq\. \([29](https://arxiv.org/html/2607.07953#S4.E29)\)\) rather than into the per\-layer value target, and it treats the routed quantity as a choice between the write error and the write value\. Table[7](https://arxiv.org/html/2607.07953#S6.T7)reports the resulting matched comparisons for the available host/scale settings\.
Table 7:Matched hidden\-stream routing comparisons under Muon\. CLER\-H routes the write residual \(sl,t=rl,ts\_\{l,t\}=r\_\{l,t\}\), while CLVR routes the write value \(sl,t=vl,ts\_\{l,t\}=v\_\{l,t\}\) in Eq\. \([29](https://arxiv.org/html/2607.07953#S4.E29)\)\. Baseline denotes the final validation loss of the matched non\-routing model, andΔ\\Deltais measured relative to that baseline at the same host and scale; negative values indicate lower loss\. All entries are single matched runs; no standard deviations are available\. The 1\.3B/40B hidden\-stream routing comparison is reported for Gated DeltaNet only\. Absolute losses are comparable only within a row because the token\-count slices differ across scales \(Section[5](https://arxiv.org/html/2607.07953#S5)\)\.Two findings are consistent across every reported row\. First, moving the routing target from the per\-layer value space, which does not improve over baseline in Table[6](https://arxiv.org/html/2607.07953#S6.T6), to the aligned hidden stream turns the comparison from neutral\-or\-negative into a small positive effect\. Second, CLVR is uniformly better than CLER\-H, even though the two signals occupy the same space and use an identically shaped projection\. Because the two variants differ only by the receiver’s own memory read, this comparison isolates the routed*content*as the useful ingredient and shows that the delta\-rule error, the signal that motivated CLER, is not the right quantity to forward\. Sincevl,t=rl,t\+v¯l,tv\_\{l,t\}=r\_\{l,t\}\+\\bar\{v\}\_\{l,t\}, the write error is the write value minus the memory’s own read, so routing the error still carries the part of the value that the lower memory has not yet absorbed; this is consistent with error routing helping, but consistently less than routing the full value\.
The gains are small and point in the same direction across the reported single\-run rows\. At 350M parameters and 1B tokens CLVR lowers final validation loss by about0\.0100\.010to0\.0120\.012on*both*hosts \(−0\.0103\-0\.0103for Gated DeltaNet,−0\.0119\-0\.0119for DeltaNet\)\. The gain then shrinks in the longer 350M/15B runs on both hosts and in the available larger Gated DeltaNet row, reaching−0\.0059\-0\.0059and−0\.0016\-0\.0016at 350M/15B and−0\.0019\-0\.0019at 1\.3B/40B\. It does not reverse in any reported row, and the value\-over\-error ordering is preserved throughout Table[7](https://arxiv.org/html/2607.07953#S6.T7); however, the table does not establish larger\-scale behavior for DeltaNet because the 1\.3B/40B DeltaNet hidden\-stream routing comparison is not reported\. This pattern is consistent with cross\-layer routing supplying information that a smaller or less\-trained recurrent memory has not yet captured on its own, with diminishing headroom as the host is strengthened\. For the runs where we log it, the zero\-initialized projectionPlP\_\{l\}grows to a comparable, nonzero norm across all routing\-capable layers, confirming that the trained model actively uses the pathway rather than leaving it a negligible perturbation\.
We also tested two natural extensions of value routing at 350M/1B \(single seed\): gating the routed value by a local surprise signal‖rl,t‖/‖vl,t‖\\\|r\_\{l,t\}\\\|/\\\|v\_\{l,t\}\\\|, and routing the concatenation\[rl,t;vl,t\]\[r\_\{l,t\};v\_\{l,t\}\]of error and value\. Neither improved on plain value routing with final losses of2\.82202\.8220and2\.81942\.8194respectively, against2\.81522\.8152for CLVR on the same seed with the surprise gate being slightly worse\. These controls are single\-seed and their differences are within run\-to\-run noise, but they reinforce the main finding that the unmodified write value is the signal worth routing, and that re\-introducing the error or modulating the value does not help\. Two further single\-seed controls attribute the gain to the routed signal rather than to the added parameters\. Routing the layer’s own hidden state through the same zero\-initialized projection, instead of an internal mixer signal, is essentially flat \(Δ≈−0\.0014\\Delta\\approx\-0\.0014\), and enlarging the Gated DeltaNet baseline by the same parameter count does not help either \(\+0\.0021\+0\.0021\); both fall well short of the routing gains, so the improvement comes from*what*is routed rather than from the extra capacity\.
For context, we also evaluated a depth\-attention baseline, Attention Residuals\[[6](https://arxiv.org/html/2607.07953#bib.bib6)\], which combines sub\-layer outputs through a learned softmax over depth rather than additively routing an internal signal\. Its effect is host\-dependent in our 350M/15B runs: it improves DeltaNet \(final loss2\.32462\.3246,−0\.0101\-0\.0101versus the DeltaNet baseline\) but slightly worsens Gated DeltaNet \(\+0\.0030\+0\.0030\)\. CLVR is more uniform across the two hosts in these runs, though its gains are smaller than the DeltaNet\-only Attention Residuals gain\. We do not pursue Attention Residuals further here, because it replaces the residual sum and routes layer outputs rather than an internal write value, and because its benefit does not transfer across the two linear\-memory hosts we study\.
### 6\.7Reproducibility
The implementation and experiment records are available at[![[Uncaptioned image]](https://arxiv.org/html/2607.07953v1/assets/github-mark.png)tommasocerruti/linear\-attention\-architectures](https://github.com/tommasocerruti/linear-attention-architectures)\. The released repository is a focused Megatron\-LM fork containing the linear\-attention implementations, final training launchers, FineWeb\-Edu/LLaMA2 data\-preparation path, smoke checks, and the scoring\-server evaluation wrapper used for the downstream results\. For reproduction, the launchers are the executable source of truth: they pin the architecture, optimizer, token budget, checkpoint format, and evaluation cadence, while site\-specific paths such as the tokenizer, Megatron data prefix, W&B credentials, and output directories are supplied through environment variables\. Table[1](https://arxiv.org/html/2607.07953#S5.T1)summarizes the shared configuration, and the run\-specific scripts provide the remaining details needed to rerun the 350M, larger\-scale, and routing comparisons\. Exploratory routing controls should be treated as reproducible paper claims only when they are accompanied by complete run records with the same metric definitions and checkpoint semantics used for the main tables\.
## 7Discussion: Validation Loss, Throughput, and Architectural Trade\-offs
The common notation separates the main design choices in the DeltaNet family: how the memory is read, how old content is forgotten or erased, and how new value information is written\. The empirical results show that these choices should not be interpreted as a one\-dimensional architecture ranking\. They instead define a multi\-objective frontier involving validation loss, training throughput, sequence\-length scaling, and implementation complexity\.
In this frontier, no single setting dominates every objective\. Kimi Delta Attention with Muon reaches the best 350M\-parameter, 15B\-token validation loss in our experiments, but it is also among the slower entries\. Pure recurrent stacks give the strongest long\-context iteration\-time scaling, while hybrid stacks usually recover part of the validation\-loss gap at the cost of some throughput\. The practical question is therefore not simply whether to use softmax attention or linear attention, but which combination of mixer, stack pattern, optimizer, and learning rate best matches the target operating point\.
#### Evidence limitations\.
The empirical comparisons are an audited run set rather than a full statistical study\. The reported architecture and routing rows are single runs, and the available records do not support seed\-averaged rankings or standard deviations\. Hyperparameter coverage is also uneven across variants, so optimizer and learning\-rate effects should be read together\. The downstream evaluation is limited to HellaSwag, PIQA, and WinoGrande, and may miss behaviors that matter for long\-context or memory\-intensive use cases\. Appendix[B](https://arxiv.org/html/2607.07953#A2)states the resulting inclusion rule: incomplete branch records are treated as exploratory design evidence, not as main numerical claims\.
The cross\-layer routing results are mixed but coherent\. In its DeltaRule\-inspired formulation, where a lower layer’s delta\-rule write error is injected into the next layer’s value target \(Eq\. \([27](https://arxiv.org/html/2607.07953#S4.E27)\)\), CLER is active but does not yield a reliable validation\-loss improvement in the controlled settings reported here: routed residuals are nonzero and the receiver\-side coefficients are trainable, yet the matched comparisons are flat \(Table[6](https://arxiv.org/html/2607.07953#S6.T6)\)\. We trace this to a basis mismatch\. The delta\-rule residual is a local correction defined by one layer’s memory state, key representation, value target, and feature map; once routed into another layer’s independently learned value space it need not be aligned, and it competes directly with the receiver’s own write target\. Two adaptations resolve the issue \(Section[4\.3](https://arxiv.org/html/2607.07953#S4.SS3)\): routing into the shared, and therefore aligned, residual stream rather than the per\-layer value space, and forwarding the write*value*rather than the write error\. The resulting method, CLVR, gives a small improvement over the reported matched DeltaNet and Gated DeltaNet baselines \(Table[7](https://arxiv.org/html/2607.07953#S6.T7)\), with the larger\-scale hidden\-stream routing comparison available for Gated DeltaNet only\. The matched downstream evaluations are more limited: HellaSwag and PIQA are broadly comparable to the baselines, while WinoGrande moves in the favorable direction for both CLER\-H and CLVR \(Table[5](https://arxiv.org/html/2607.07953#S6.T5)\), these differences are not reliable enough here to serve as an independent signal\. We interpret these evaluations as showing no clear downstream degradation on the tested tasks, not as proof of broad downstream improvement\. That value beats error at fixed parameter count indicates the useful cross\-layer signal is the layer’s write target, not the delta\-rule correction that originally motivated CLER\. The choice is left open for richer hosts such as Gated DeltaNet\-2, where the write value and the separate erase and write residuals are decoupled\.
## 8Future Work
Although CLVR helps in the reported DeltaNet and Gated DeltaNet rows, the effect is small and leaves several questions open\. The clearest is whether the gain persists, or grows, on hosts we did not evaluate under routing: Kimi Delta Attention and Gated DeltaNet\-2\. Gated DeltaNet\-2 in particular decouples the write value from separate erase and write residuals, so it is not obvious which internal signal should be routed; a matched comparison there would test whether “route the write value” generalizes beyond DeltaNet and Gated DeltaNet\. It also remains open whether the diminishing return we observe with longer training and in the larger Gated DeltaNet row continues until CLVR becomes neutral, or settles at a small positive floor, which additional repeated runs at larger token budgets could resolve\.
Future routing runs should also log the quantities needed to measure influence directly: the routed signal after projection, the receiver’s residual stream before and after injection, and the learned projection norms at matched checkpoints\. These logs would make it possible to separate cases that validation loss alone cannot distinguish, namely whether a remaining gap is because the routed signal is too small, expressed in the wrong basis, or aligned but redundant with the receiver’s local update\.
Finally, CLVR should be tested on tasks where cross\-layer recovery has a clearer role than average next\-token prediction\. Synthetic key\-value retrieval, passkey\-style tasks, long\-context QA, many\-distractor in\-context learning, and other associative\-recall settings are natural candidates\. The throughput study should also be extended from training iteration time to measured inference throughput, decoding memory footprint, and quality at long context length\.
## 9Conclusion
We compared softmax attention with DeltaNet, Gated DeltaNet, Kimi Delta Attention, and Gated DeltaNet\-2 in a common recurrent\-memory notation\. The empirical picture is a multi\-objective trade\-off among validation loss, throughput, and sequence\-length scaling rather than a single architecture ranking\.
Within the reported 350M\-parameter, 15B\-token sweep, Kimi Delta Attention with Muon in a hybrid stack reaches the best validation loss, while pure Gated DeltaNet with AdamW gives the fastest normalized training speed\. Hybrid stacks often improve loss, but pure recurrent stacks preserve the strongest long\-context iteration\-time scaling\. The larger DeltaNet and downstream results reinforce the same conclusion: architecture comparisons are meaningful only when scale, optimizer, learning rate, stack pattern, and metric semantics are recorded together\.
We also studied a lightweight, linear\-time pathway for sharing information across depth in DeltaNet\-style stacks\. Starting from Cross\-Layer Error Residuals, which forward a lower layer’s delta\-rule write error into the next layer’s value target, we found that this natural formulation does not improve over matched DeltaNet or Gated DeltaNet baselines\. Diagnosing the cause as a space mismatch, we instead route the signal into the aligned hidden stream through a zero\-initialized projection and find that the layer’s write*value*, not its write error, is the useful signal to forward\. The resulting method, Cross\-Layer Value Routing \(CLVR\), gives a small reduction in final validation loss in the matched single runs reported for both DeltaNet and Gated DeltaNet at 350M parameters, and in the larger Gated DeltaNet run, while preserving the linear\-time structure of the host\. The gain diminishes as the base model is trained longer or made larger, which leaves its behavior at much larger scale, and on other linear\-memory hosts such as Kimi Delta Attention and Gated DeltaNet\-2, as the main open question\.
## 10Acknowledgments
Thanks to the Projects in Machine Learning Research \(PMLR\) course at ETH Zurich, and to CSCS for compute on Alps\.
## References
- \[1\]Vaswani, A\., Shazeer, N\., Parmar, N\., Uszkoreit, J\., Jones, L\., Gomez, A\. N\., Kaiser, L\., and Polosukhin, I\.Attention is all you need\.In*Advances in Neural Information Processing Systems*, volume 30, pages 5998–6008, 2017\.
- \[2\]Schlag, I\., Irie, K\., and Schmidhuber, J\.Linear transformers are secretly fast weight programmers\.In*Proceedings of the 38th International Conference on Machine Learning*, pages 9355–9366\. PMLR, 2021\.
- \[3\]Yang, S\., Kautz, J\., and Hatamizadeh, A\.Gated Delta Networks: Improving Mamba2 with Delta Rule\.In*The Thirteenth International Conference on Learning Representations*, 2025\.
- \[4\]Kimi Team\.Kimi Linear: An Expressive, Efficient Attention Architecture\.*arXiv preprint arXiv:2510\.26692*, 2025\.
- \[5\]Hatamizadeh, A\., Choi, Y\., and Kautz, J\.Gated DeltaNet\-2: Decoupling Erase and Write in Linear Attention\.*arXiv preprint arXiv:2605\.22791*, 2026\.
- \[6\]Kimi Team\.Attention Residuals\.*arXiv preprint arXiv:2603\.15031*, 2026\.
- \[7\]Zhu, L\., Fang, Y\., Liao, B\., Wang, S\., Cheng, T\., Huang, Z\., Chen, C\., Wei, L\., Zeng, Y\., Wang, Y\., Lin, Y\., Li, Y\., and Wang, X\.Mixture\-of\-Depths Attention\.*arXiv preprint arXiv:2603\.15619*, 2026\.
- \[8\]Zhou, Z\., Wu, T\., Jiang, Z\., Obeid, F\., and Lan, Z\.Value Residual Learning\.In*Proceedings of the 63rd Annual Meeting of the Association for Computational Linguistics \(Volume 1: Long Papers\)*, pages 28341–28356\. Association for Computational Linguistics, 2025\.
- \[9\]Penedo, G\., Kydlíček, H\., Ben Allal, L\., Lozhkov, A\., Mitchell, M\., Raffel, C\., von Werra, L\., and Wolf, T\.The FineWeb Datasets: Decanting the Web for the Finest Text Data at Scale\.*arXiv preprint arXiv:2406\.17557*, 2024\.
- \[10\]Touvron, H\., Martin, L\., Stone, K\., Albert, P\., Almahairi, A\., Babaei, Y\., Bashlykov, N\., Batra, S\., Bhargava, P\., Bhosale, S\., and others\.Llama 2: Open Foundation and Fine\-Tuned Chat Models\.*arXiv preprint arXiv:2307\.09288*, 2023\.
- \[11\]Loshchilov, I\. and Hutter, F\.Decoupled Weight Decay Regularization\.In*International Conference on Learning Representations*, 2019\.
- \[12\]Jordan, K\., Jin, Y\., Boza, V\., You, J\., Cesista, F\., Newhouse, L\., and Bernstein, J\.Muon: An optimizer for hidden layers in neural networks\.2024\.[https://kellerjordan\.github\.io/posts/muon/](https://kellerjordan.github.io/posts/muon/)\.
## Appendix
## Appendix ASupplementary 1B\-Token Baselines
The main text emphasizes the 15B\-token architecture sweep and the larger 1\.3B\- and 3B\-parameter DeltaNet runs because they provide the clearest validation\-loss/throughput comparison\. Table[8](https://arxiv.org/html/2607.07953#A1.T8)reports an earlier 350M\-parameter baseline sweep trained for approximately 1B tokens under the same FineWeb\-Edu/LLaMA2 setup, sequence length 4096, and one\-node GH200 configuration as the controlled routing experiments\.
Table 8:Supplementary 350M baseline runs trained for approximately 1B tokens\. These shorter\-budget runs provide context before the 15B\-token sweep but are not used to determine the main validation\-loss/throughput frontier\.
## Appendix BResult Inclusion Criteria
This appendix summarizes how results are separated between the main empirical analysis, supplementary context, and exploratory discussion\. Main quantitative claims must identify the model family, stack pattern, scale, token budget, optimizer, learning rate, metric definition, and checkpoint semantics\.
#### Statistical scope\.
The inclusion standard is an auditability standard, not a variance estimate\. The reported entries are single runs, so no standard deviations are available; small validation\-loss gaps are treated as suggestive unless the same direction appears across matched settings or downstream checks\.
#### Main empirical results\.
The main tables and figures include the 350M\-parameter, 15B\-token architecture sweep, the learning\-rate ablation, sequence\-length timing, larger DeltaNet runs, downstream evaluations, the matched 350M\-parameter CLER ablations, and the cross\-layer routing comparisons \(Table[7](https://arxiv.org/html/2607.07953#S6.T7)\)\. These results specify the relevant training scale and metric semantics and are comparable within their experimental groups\.
#### Supplementary context\.
The 350M\-parameter, 1B\-token baselines in Table[8](https://arxiv.org/html/2607.07953#A1.T8)use a shorter training budget than the main architecture sweep\. They provide early baseline and optimizer context but do not determine the final 350M\-parameter validation\-loss/throughput frontier\.
#### Cross\-layer routing results\.
The matched hidden\-stream routing comparisons, CLER\-H and CLVR, together with the Attention Residuals baseline, are reported in the main text \(Table[7](https://arxiv.org/html/2607.07953#S6.T7)\) with matched baselines and final validation losses, and meet the same standard as the other main results\. The remaining controls, namely surprise\-gated value routing, error\-plus\-value concatenation, output injection, and low\-rank projection variants, are reported only as indicative checks; they would need seed\-averaged matched runs to enter the main comparison\.Similar Articles
You Could Have Come Up with Kimi Delta Attention
This blog post derives Kimi Delta Attention step by step from standard softmax attention through linear attention and DeltaNet variants, explaining the state update equations used by recent Qwen and Kimi models.
Kimi Linear: An Expressive, Efficient Attention Architecture
Kimi Linear proposes a new linear attention architecture designed to enhance both expressiveness and efficiency in Transformer models, with contributions from the Kimi Team at Moonshot AI.
Delta Attention Residuals
Delta Attention Residuals improve layer-wise routing in transformer models by attending to feature changes (deltas) rather than cumulative hidden states, achieving 1.7–8.2% validation perplexity gains across scales from 220M to 7.6B parameters.
Learning to Skip Blocks: Self-Discovered Ultrametric Routing for Hardware-Accelerated Sparse Attention
This paper introduces Dynamic Ultrametric Attention, a framework where Transformers learn per-head block-sparse routing topologies during training, which are then offloaded to a custom Triton block-sparse kernel at inference time, achieving up to 28x speedup and 98.4% memory reduction over dense attention.
Exact Linear Attention
This paper introduces Exact Linear Attention (ELA), a mechanism that achieves linear computational complexity for Transformer attention without approximation error by leveraging kernel decomposition, and addresses gradient explosion and token dilution through constrained kernel functions. It also presents engineering innovations including Hyper Link, Memory Lobe, and a routing bias for Mixture of Experts.