Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Hybrid Linear Attention Done Right: Efficient Distillation and Effective Architectures for Extremely Long Contexts

About

Hybrid Transformer architectures, which combine softmax attention blocks and recurrent neural networks (RNNs), have shown a desirable performance-throughput tradeoff for long-context modeling, but their adoption and studies are hindered by the prohibitive cost of large-scale pre-training from scratch. Some recent studies have shown that pre-trained softmax attention blocks can be converted into RNN blocks through parameter transfer and knowledge distillation. However, these transfer methods require substantial amounts of training data (more than 10B tokens), and the resulting hybrid models also exhibit poor long-context performance, which is the scenario where hybrid models enjoy significant inference speedups over Transformer-based models. In this paper, we present HALO (Hybrid Attention via Layer Optimization), a pipeline for distilling Transformer models into RNN-attention hybrid models. We then present HypeNet, a hybrid architecture with superior length generalization enabled by a novel position encoding scheme (named HyPE) and various architectural modifications. We convert the Qwen3 series into HypeNet using HALO, achieving performance comparable to the original Transformer models while enjoying superior long-context performance and efficiency. The conversion requires just 2.3B tokens, less than 0.01% of their pre-training data

Yingfa Chen, Zhen Leng Thai, Zihan Zhou, Zhu Zhang, Xingyu Shen, Shuo Wang, Chaojun Xiao, Xu Han, Zhiyuan Liu• 2026

Related benchmarks

TaskDatasetResultRank
Long-context recallLong-context Recall-intensive Tasks Suite SQuAD, FDA, SWDE
SQuAD Recall Accuracy54.5
48
Commonsense ReasoningCommonsense Reasoning Suite (PIQA, ARC-e, ARC-c, HellaSwag, WinoGrande)
PIQA Accuracy73.2
35
Long-context ReasoningRULER
RULER Performance (8K Context)36.4
35
Commonsense ReasoningARC-e, ARC-c, PIQA, WinoG., HellaS.
ARC-e Accuracy82.5
25
Needle-In-A-Haystack RetrievalNIAH-Single-1 128K context
Accuracy100
23
Needle-In-A-Haystack RetrievalNIAH-Single-2 256K context
Accuracy95.2
23
Needle-In-A-Haystack RetrievalNIAH-Single-1 256K context
Accuracy100
23
Needle-In-A-Haystack RetrievalNIAH-Single-1 64K context
Accuracy100
23
Needle-In-A-Haystack RetrievalNIAH-Single 128K context 2
Accuracy95
23
Needle-In-A-Haystack RetrievalNIAH-Single-3 128K context
Accuracy75
23
Showing 10 of 40 rows

Other info

GitHub

Follow for update