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

STaR-Quant: State-Time Consistent Post-Training Quantization for Diffusion Large Language Models

About

Diffusion large language models (DLLMs) have recently emerged as a promising alternative to autoregressive LLMs by generating text through iterative masked denoising with bidirectional context. However, their large model sizes and iterative denoising process introduce substantial memory and computational overhead, motivating post-training quantization for efficient deployment. In this paper, we identify two key challenges for low-bit DLLM quantization: state-dependent activation disparity and temporal error accumulation. Masked and unmasked tokens exhibit different activation distributions within each denoising step, while quantization errors can accumulate across steps during iterative decoding. To address these challenges, we propose STaR-Quant, a state-time consistent PTQ framework for DLLMs. STaR-Quant introduces State-Guided Activation Transformation (SGAT) to assign masked and unmasked tokens to different activation transformation spaces with a unified static weight-side transformation. It further introduces Temporal Attention Compensation (TAC) to correct the quantized attention representation via a lightweight block-diagonal affine mapping. Experiments on representative DLLMs demonstrate that STaR-Quant consistently improves low-bit weight-activation quantization over strong PTQ baselines, while delivering up to 1.69x speedup and 3.14x memory saving over FP16 deployment.

Xin Yan, Aqiang Wang, Zhenglin Wan, Xingrui Yu, Ivor Tsang• 2026

Related benchmarks

TaskDatasetResultRank
Commonsense ReasoningWinoGrande
Accuracy72.98
1581
Commonsense ReasoningHellaSwag
HellaSwag Accuracy74.31
897
Question AnsweringPIQA
Accuracy74.34
589
Sentence CompletionHellaSwag
Accuracy72.43
440
Code GenerationHumanEval
pass@151.76
329
Commonsense ReasoningWinoGrande
Accuracy73.98
94
Science Question AnsweringARC
ARC Accuracy86.5
82
Language UnderstandingCEval
Accuracy64.56
67
Question AnsweringARC
Accuracy85.12
45
Truthful Question AnsweringTruthfulQA
Accuracy (TruthfulQA)49.72
37
Showing 10 of 12 rows

Other info

Follow for update