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

FLARE: Diffusion for Hybrid Language Model

About

Autoregressive (AR) large language models (LLMs) have achieved broad practical success, but sequential decoding remains a key bottleneck for low-latency deployment. Recent efficient-inference work has progressed along two axes: reducing the cost of each model invocation through efficient architectures, and reducing serial decoding steps through parallel generation. Hybrid attention backbones address the former, while diffusion language models (dLLMs) pursue the latter via iterative parallel denoising. Combining these advantages remains challenging: AR-to-dLLM conversion often fails to preserve seed-checkpoint capability, and hybrid-attention recurrent states and masking constraints make diffusion training and serving nontrivial. We present FLARE, a systematic conversion framework for hybrid-attention LLMs. Our analysis identifies transfer data quality as the primary determinant of capability preservation, outweighing loss formulation and attention-mask design. The resulting framework combines a token-equal AR-and-diffusion objective, hardware-aware kernels, and unified inference, enabling one checkpoint to support both AR-style verified decoding and diffusion-style parallel denoising. Starting from strong AR checkpoints with limited post-training data, FLARE is competitive with leading open-source dLLMs across model scales and delivers consistent throughput gains over open-source dLLM baselines in single-GPU concurrent serving. Our results further suggest that practical dLLMs are limited not only by decoding algorithms, but also by transfer data quality and the training inefficiency of current block-diffusion objectives, motivating joint design of data, objectives, architectures, and inference systems.

Yuchen Zhu, Jing Shi, Chongjian Ge, Hao Tan, Yiran Xu, Wanrong Zhu, Jason Kuen, Koustava Goswami, Rajiv Jain, Yongxin Chen, Molei Tao, Jiuxiang Gu• 2026

Related benchmarks

TaskDatasetResultRank
MathMATH 500
Accuracy95.2
120
CodeHumanEval
HumanEval Accuracy93.29
118
Instruction FollowingIFEval
IFEval Score73.2
87
CodeMBPP
Pass@191.05
73
MathAIME24
Accuracy58.89
57
MathematicsAIME 2024
Accuracy63.33
40
Decoding ThroughputHumanEval
Throughput (tok/s)1.76e+3
27
Decoding ThroughputGSM8K
Throughput (tok/s)2.09e+3
27
Throughput MeasurementGPQA Diamond
Throughput (tokens/s)1.44e+3
27
Knowledge & Instruction FollowingMMLU
Score84.8
24
Showing 10 of 26 rows

Other info

Follow for update