Different Tasks, Different Timesteps: Temporal Task-Aware Quantization for Diffusion Language Models
Abstract
Domain fit: AI-core
Diffusion Language Models (DLMs) introduce unique quantization challenges due to their iterative denoising process. In this work, we reveal that quantization sensitivity is both taskand time-dependent, with different downstream tasks exhibiting distinct precision requirements across denoising steps. We further show that these patterns can be predicted without groundtruth labels or task-specific evaluation metrics. Building on these findings, we introduce T 2Quant, a task and time-aware mixedprecision quantization framework that uses a label-free sensitivity signal to optimally allocate activation precision across denoising steps under a fixed bit-width budget. Extensive experiments across three DLMs and six benchmarks demonstrate that T 2Quant consistently improves the average accuracy of existing quantization methods, achieving gains of up to 6.7 percentage points under the same budget.
Results and benchmarks
Diffusion Language Models (DLMs) introduce unique quantization challenges due to their iterative denoising process.
| Task | Dataset | Metric | Value | Source |
|---|---|---|---|---|
| Diffusion | GSM8K | Precision | 90 | paper-derived |
| Diffusion | MMLU-Pro | Precision | 100 | paper-derived |
Audit each benchmark finding before selecting an implementation path. Evidence refs map to the disclosure below.
Evidence graph: 3 refs, 3 links.
Utility signals: depth 100/100, grounding 85/100, status high.
Implementation
No direct implementation yet
Maintained implementation evidence is not confirmed for this paper yet.
Use the implementation status and reproduction sections for the current action plan.
No verified maintained repo yet
There is no verified maintained implementation yet. Use this baseline plan to decide whether to prototype now or defer.
- No maintained paper-verified implementation was found; start with the closest related repositories below.
- Compare repo methods against the paper equations/algorithm before trusting metrics.
- Create a minimal baseline implementation from the paper and use adjacent repos as references.
Time to first repro: a few days
AI-Efficiency/Awesome-Model-Quantization is the closest maintained adjacent implementation (Matches contextual method/domain keyword: quantization). It is not paper-verified; validate algorithm and evaluation setup against the paper before trusting reported metrics. Community adoption signal: 2455 GitHub stars.
- Adjacent implementations are not paper-verified
- Recommended repository is adjacent and not paper-verified.
Reproduction readiness
No repo
No verified implementation available
- No maintained repository has been identified for this paper. Check adjacent implementations or HF artifacts below.
Hardware requirements
- Expect multi-day setup/compute for meaningful reproduction based on current guidance.
Framework baselines
- Hugging Face Diffusers training guide
Practical baseline for diffusion model reproduction.
Repositories and ecosystem
Closest related implementations
These are not paper-verified. Use them as reference points when no direct implementation is available.
- AI-Efficiency/Awesome-Model-Quantization Adjacent · Confidence: Medium · 2,455 stars
Matches contextual method/domain keyword: quantization
Official
No additional official repositories detected.
Community
- OfekAvivi/T2QuantConfidence: Medium
T²Quant (T2Quant): task- and time-aware activation quantization for diffusion language models (LLaDA, Dream). Label-free calibration + exact DP precision schedule. arXiv:2610.31198
3 stars · 0 forks · Last push Oct 23, 2026 · MIT license
Hugging Face artifacts
No trustworthy direct or curated related Hugging Face artifacts were found yet. Use targeted searches to quickly locate candidate models, datasets, and demos.
Tip: start with models, then check datasets and spaces if you need evaluation data or demos.
Research context
Tasks
Diffusion
Methods
Diffusion, Quantization
Domains
Natural Language Processing
Open this paper in HFEPX to review benchmark signals, evaluation modes, and human-feedback protocol context.
Open in HFEPXJump to Paper2Code search queries derived from this paper's research context.