Denoising Surface: Modeling and Predicting Inference Cost for Diffusion LLM Serving
cs.LG
Submitted: 2026-09-30
Updated: 2026-09-30
Code: https://github.com/tatsu-lab/stanford_alpaca
License: http://creativecommons.org/licenses/by/4.0/
The gist: As diffusion large language models (dLLMs) become more capable, they are moving from research settings to real-world serving, where request management (such as scheduling and resource allocation)
Abstract
As diffusion large language models (dLLMs) become more capable, they are moving from research settings to real-world serving, where request management (such as scheduling and resource allocation) relies on accurate estimation of per-request inference cost. However, common cost proxies fall short for dLLMs: output length ignores that one forward pass can unmask multiple tokens, and denoising-step count ignores the heterogeneous per-step costs. We observe that the block-autoregressive generation mechanism induces a two-dimensional execution structure over output blocks and within-block denoising steps, whereas these proxies collapse it into a scalar, discarding information essential for characterizing the cost. Motivated by this insight, we propose the Denoising Workload Surface (DWS), which preserves this two-dimensional block-step structure as a probability surface to weight the heterogeneous per-step costs. We then design a coarse-to-fine training scheme that enables a lightweight prompt-only predictor to accurately predict the complex DWS. This predictor runs efficiently even on a single CPU core, avoiding GPU contention with the serving model. Since DWS decouples request-dependent execution behavior from deployment-specific cost factors, the predictor transfers across hardware configurations without retraining. In real-world serving experiments, DWS reduces cost-prediction error by up to 2.50 times over scalar-based predictors, while the DWS-guided shortest-job-first scheduler reduces end-to-end latency by up to 1.92 times for online chatbots.
Sources
- LLaDA2.0: Scaling Up Diffusion Language Models to 100B
- TeDiServe: High SLO Attainment Serving for Diffusion Language Models
- TIDE: Efficient and Lossless MoE Diffusion LLM Inference with I/O-aware Expert Offload
- DiffusionGemma Technical Report
- dInfer: An Efficient Inference Framework for Diffusion Language Models
- Seed Diffusion: A Large-Scale Diffusion Language Model with High-Speed Inference
- Robust Length Prediction: A Perspective from Heavy-Tailed Prompt-Conditioned Distributions
- Optimus: Elastic Decoding for Efficient Diffusion LLM Serving
- Qwen3 Technical Report
- Dream 7B: Diffusion Large Language Models
- BlockServe: Block-Grained Continuous Batching for High-Throughput Diffusion LLM Serving
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: The Predictive Cost of ELBO Binding
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks