ArXiv: 2509.26328
π― Pitch
Turning an autoregressive LLM into a fast parallel decoder typically sacrifices quality or demands massive retrainingβFast-dLLM v2 sidesteps both with a 1B-token fine-tune that makes the model generate in blocks instead of token-by-token, hitting 2.5Γ speedups while matching or beating the original accuracy. The trick is forcing each token to train in both masked and unmasked contexts simultaneously, which preserves autoregressive-style attention patterns even when multiple tokens are decoded at once.
1. Executive Summary
This paper introduces Fast-dLLM v2, a block diffusion language model framework that adapts pretrained autoregressive LLMs into efficient diffusion-style decoders for parallel text generation. Using Qwen2.5-Instruct models (1.5B and 7B) fine-tuned on the LLaMA-Nemotron dataset, the method combines a block diffusion mechanism with complementary masking (training each token in both masked and unmasked contexts via paired complementary views within blocks) and a hierarchical caching mechanism (a block-level cache for inter-block KV reuse and a DualCache-based sub-block cache for intra-block refinement) to accelerate decoding. Fast-dLLM v2 achieves up to 2.5Γ speedup over standard AR decoding while matching or surpassing AR accuracy across benchmarks including GSM8K, MATH, HumanEval, and MMLU, establishing that lossless adaptation of AR models into block-diffusion decoders is possible with only ~1B tokens of fine-tuning β a 500Γ reduction compared to full-attention diffusion models like Dream β only when the block-wise attention mask structure remains AR-compatible and training-time block alignment is strictly preserved at inference.
2. Context and Motivation
The Core Problem: Autoregressive Decoding Is a Serial Bottleneck
The fundamental problem this paper tackles is the inherent sequentiality of autoregressive decoding in large language models. In a standard AR-LLM, tokens are generated one-by-one in strict left-to-right order: to produce token , the model must have already generated . Each generation step requires a full forward pass through the transformer, and these forward passes cannot be parallelized across time steps because each one depends on the output of the previous.
This serial bottleneck has concrete consequences. The throughput of an AR-LLM during inference is bounded by memory bandwidth and the sequential chain of transformer forward passes. In latency-sensitive applications β interactive assistants, real-time translation, on-device inference β this serial dependency directly constrains how quickly the user receives a response. In high-throughput batch processing β generating training data, evaluating on large test suites, serving many users simultaneously β the per-token processing cost limits how many queries can be handled per unit of hardware.
The paper reports that Qwen2.5-7B-Instruct achieves roughly 39.5 tokens/sec at batch size 1 and 102.5 tokens/sec at batch size 4 on an A100 GPU (Figure 1b). While respectable, this throughput leaves significant hardware parallelism underutilized: modern GPUs can perform many more operations per second than a single token's forward pass requires, but the serial dependency prevents batching those operations across time.
Why This Matters: The Gap Between Training Parallelism and Inference Serialism
This problem is important for a reason that goes beyond immediate throughput concerns: there is a fundamental asymmetry between training and inference in autoregressive models. During training, all tokens in a sequence are available simultaneously, so the model processes the entire input in parallel with a single forward pass (using causal masking to prevent information leakage from future tokens). This means training efficiency scales well with hardware parallelism. During inference, however, the model must generate tokens one at a time, converting what was a parallel operation during training into a serial one at deployment.
This asymmetry means that as models scale up, the inference cost per token grows (more parameters = more FLOPs per forward pass), while the degree of exploitable parallelism at inference time does not (the tokens are still generated sequentially). The practical result is that deploying large models becomes increasingly expensive not just because the hardware requirements grow, but because the serial generation pattern fundamentally limits how effectively that hardware can be utilized.
This is a systems-level problem with theoretical significance: it represents a gap between what the model architecture can do (parallel processing during training) and what the generation paradigm allows (serial token production). Bridging this gap β finding ways to generate text with greater parallelism while preserving the quality that makes AR models effective β has been a persistent challenge in the field since the earliest work on non-autoregressive generation.
Prior Approach 1: Full-Attention Masked Diffusion LLMs
The most direct alternative to autoregressive decoding explored in recent work is masked diffusion language models (dLLMs). These models are trained to denoise corrupted sequences: at training time, tokens are randomly replaced with [MASK] tokens, and the model learns to predict the original tokens given the partially masked input. At inference time, the model starts from a fully masked sequence (or a prompt with the remainder masked) and iteratively unmasks tokens, with multiple tokens being predicted in parallel at each denoising step.
How they work. The training objective for masked diffusion is:
where is the masking ratio, is the sequence with each token independently masked with probability , and the model predicts the original tokens at all masked positions simultaneously. This is fundamentally more parallel than AR training: all positions are predicted in one forward pass given the full corrupted context.
Notable examples. The paper cites LLaDA-8B (Nie et al., 2025), trained from scratch on the masked diffusion objective, and Dream-7B (Ye et al., 2025b,a), which adapts Qwen2.5-7B into a full-attention diffusion model. Dream is the most relevant comparison point because it shares the same base model (Qwen2.5-7B) and the same goal (converting a pretrained AR model to diffusion), but uses full bidirectional attention during diffusion rather than the block-wise approach of Fast-dLLM v2.
Where they fall short. The paper identifies several critical weaknesses of full-attention dLLMs that have prevented them from achieving practical speed advantages:
-
KV cache incompatibility. In AR models, the Key-Value cache stores representations of previously generated tokens, so each new token only needs to compute attention against the single new query vector β a operation per step rather than . In full-attention dLLMs, the bidirectional attention means that when tokens are unmasked at different denoising steps, the attention patterns change, and previously computed KV states become stale. The model must recompute attention over the full sequence at each step, eliminating the efficiency that KV caching provides in AR models. Fast-dLLM (Wu et al., 2025) introduced DualCache as an approximate caching mechanism, but as the paper notes, "this does not fundamentally resolve incompatibility of dLLMs with KV cache, since such approximate caches are not equivalent to the original computation."
-
Inference latency often exceeds AR models. Despite the theoretical advantage of parallel token generation, full-attention dLLMs typically require multiple denoising iterations (e.g., 20-64 steps) to produce high-quality output. Each iteration involves a full forward pass over the entire (potentially long) sequence with bidirectional attention. The number of forward passes may be smaller than AR's one-per-token, but each forward pass is more expensive (full sequence attention rather than single-token append). Net throughput often ends up worse than AR, defeating the purpose.
-
Fixed or restricted sequence lengths. Many full-attention dLLMs require the total generation length to be specified upfront or padded to a fixed length, since the entire masked region must be initialized before denoising begins. This limits flexibility for variable-length generation tasks.
-
Prohibitive training cost. Adapting a pretrained AR model to full-attention diffusion requires the model to learn an entirely new attention pattern (causal β bidirectional) from scratch. Dream required approximately 580 billion tokens of fine-tuning to achieve competitive performance. For most practitioners, this volume of post-training is prohibitively expensive.
Prior Approach 2: Block Diffusion as an Interpolation
Block diffusion language models (BD3-LMs, Arriola et al., 2025) were proposed precisely to address the KV-cache incompatibility of full-attention dLLMs. The key idea is to interpolate between autoregressive and diffusion regimes: the sequence is divided into blocks, and generation proceeds block-by-block in an autoregressive fashion (left-to-right), while within each block, tokens are generated via masked diffusion (parallel prediction and iterative refinement).
How they work. At the block level, the model behaves autoregressively: block is conditioned on all previously generated blocks through causal attention, and its tokens are cached as prefix context for future blocks. This enables block-level KV caching, where the representations of past blocks are frozen and reused, exactly as in AR models. Within block , the model uses bidirectional attention to enable diffusion-based parallel token prediction: tokens in the block can attend to each other in both directions, allowing the model to refine uncertain tokens using context from later (still-masked) positions in the same block.
The attention mask structure (illustrated in Figure 7 of the paper) decomposes naturally into three sub-masks:
- Block-diagonal (): bidirectional self-attention within each block for the noised sequence , enabling within-block refinement.
- Offset block-causal (): causal attention from noised tokens in block to clean tokens in earlier blocks , preserving inter-block conditioning.
- Block-causal (): causal attention among clean tokens across blocks, enabling autoregressive-style progression.
Where existing block diffusion falls short. Despite the conceptual appeal, BD3-LMs were only validated on "relatively small-scale models and conventional LM metrics, rather than modern large-scale LLM settings." The paper does not specify the exact model scales used in BD3-LMs, but the implication is clear: no prior work had demonstrated that block diffusion could be applied to 7B-parameter instruction-tuned models on the diverse benchmark suite (GSM8K, MATH, HumanEval, MMLU, GPQA, IFEval) that the community uses to evaluate modern LLMs. The practical applicability to state-of-the-art models remained "unclear, especially in terms of maintaining high-quality text generation and robust scaling behavior."
Additionally, the training recipe for block diffusion β how to construct the masking schedule, how to handle block alignment, how to prevent the model from losing its AR capabilities during fine-tuning β had not been optimized. The paper demonstrates that non-obvious design choices (padding for block alignment, complementary masking for full supervision coverage) make substantial differences in final performance (Table 2 shows a +3.7 point average improvement from these techniques).
Prior Approach 3: Concurrent Block Diffusion Adaptations
The paper acknowledges several concurrent works that also adapt pretrained AR models to block diffusion:
- SDAR (Cheng et al., 2025): fine-tunes a block diffusion model from a pretrained autoregressive model.
- D2F (Wang et al., 2025b): distills a large dLLM into a more efficient block diffusion model using Diffusion Forcing.
- Set Block Decoding (Gat et al., 2025): integrates next-token prediction and masked token prediction within a single architecture for simultaneous multi-token generation.
The paper differentiates Fast-dLLM v2 from these concurrent works primarily through its data efficiency: requiring only ~1B tokens of fine-tuning versus unspecified larger amounts in concurrent approaches. However, the paper does not provide detailed comparisons of training costs or methodology beyond this broad claim, making the differentiation somewhat superficial in the text. The key claim that Fast-dLLM v2 requires "only 1B tokens" is anchored to the training configuration in Appendix A.1: 2,500 steps Γ 524,288 tokens/step β 1.31 billion tokens for the 7B model (and 6,000 steps for the 1.5B model, yielding ~3.15 billion tokens β notably more than the "~1B" headline figure).
Why Block Diffusion Hasn't Succeeded at Scale Before
The paper implicitly identifies several reasons why block diffusion hadn't previously been demonstrated at the 7B scale with competitive performance:
1. The attention mask must be AR-compatible to enable efficient fine-tuning. Full-attention diffusion models require the pretrained model to learn an entirely new attention pattern (bidirectional rather than causal), which is a major representational shift requiring massive amounts of data (Dream's 580B tokens). Block diffusion's attention pattern is much closer to the original causal structure β the block-causal and offset block-causal sub-masks preserve left-to-right dependencies that the pretrained model already understands. This is why the paper can claim the adaptation is "lossless" with only ~1B tokens: the model doesn't need to unlearn its autoregressive conditioning patterns; it only needs to learn to use bidirectional context within local blocks.
This is a subtle but crucial point that the paper states directly: "Unlike the full-attention dLLM in Dream, our design uses a block-wise attention mask structure closer to the original AR models, making the adaptation process inherently more compatible and data-efficient." The word "closer" matters β it's not identical to AR attention, but the difference is small enough that the model can adapt quickly.
2. Block alignment during training is critical for preventing cross-sample leakage. Because block diffusion uses bidirectional attention within blocks, if sequence boundaries don't align with block boundaries, tokens from different training samples could end up in the same block and attend to each other β a form of data leakage that would degrade learning. The paper's padding strategy ("+ pad" in Table 2) pads each sample to a multiple of block size before packing, ensuring clean block boundaries. Without this, the naive approach shows a substantial performance gap (41.3 vs. 42.2 average accuracy).
3. Complementary masking is needed to supervise all tokens. In standard masked diffusion training, only the masked tokens contribute to the loss. With random masking at ratio , some tokens may rarely or never be masked during training if the training data is limited. The complementary masking strategy (training on both mask and complement in the same batch) ensures every token position receives gradient signal in both states (masked and unmasked), approximately doubling the effective supervision per training example. Table 2 shows this improves average accuracy by +2.8 points.
4. No prior work combined block diffusion with efficient intra-block decoding strategies. Even if block diffusion enables KV caching across blocks, within-block generation still requires iterative refinement that can be slow. Fast-dLLM v2 inherits the confidence-aware parallel decoding from Fast-dLLM (Wu et al., 2025), where tokens exceeding a confidence threshold are decoded and unmasked in parallel, reducing the number of refinement steps needed. The hierarchical caching (block-level for inter-block reuse, sub-block for intra-block reuse via DualCache) further reduces redundant computation. No prior block diffusion work had integrated these acceleration techniques.
How Fast-dLLM v2 Positions Itself
The paper situates Fast-dLLM v2 at the intersection of three research threads, claiming to combine their advantages while avoiding their individual weaknesses:
| Research Thread | Example | Strength | Weakness | Fast-dLLM v2's Position |
|---|---|---|---|---|
| AR LLMs | Qwen2.5, LLaMA | High quality, KV-cache efficiency | Serial decoding bottleneck | Preserves quality and KV-cache via block-wise structure |
| Full-attention dLLMs | Dream, LLaDA | Parallel token prediction | No KV-cache, requires retraining from scratch | Uses block-wise attention for cache compatibility and data-efficient fine-tuning |
| Block diffusion (small scale) | BD3-LMs | KV-cache + parallel generation | Never validated at 7B scale | Scales to 7B with competitive benchmarks |
The paper's central thesis can be stated as a conditional claim: block diffusion can match AR quality while achieving faster inference, but only if (1) the attention mask structure remains compatible with the pretrained AR model's representations (enabling data-efficient fine-tuning), (2) training-time block alignment is strictly enforced to prevent cross-sample leakage, (3) complementary masking ensures full supervision coverage, and (4) hierarchical caching and parallel decoding are integrated to fully exploit the parallelism that block diffusion enables.
The explicit comparison point is Dream: both start from pretrained AR models and convert them to diffusion, but Dream requires 500Γ more training data (500B vs. ~1B tokens), and Fast-dLLM v2 achieves higher throughput (102.5 vs. 48.2 tokens/sec at batch size 4, Figure 1b) while matching or exceeding accuracy (60.3 vs. 57.6 average score, Table 1). The comparison to LLaDA-8B (trained from scratch) and LLaDA-1.5 (with preference optimization) establishes that Fast-dLLM v2's fine-tuning approach produces better results than training diffusion models from scratch, at least at the 7B scale.
The Throughput-Accuracy Tradeoff as Framing
The paper frames its contribution through the lens of a throughput-accuracy tradeoff, explicitly visualized in Figure 1a. The x-axis is GSM8K accuracy (higher is better), the y-axis is throughput in tokens/sec (higher is better), creating a desirable "upper-right" region. Fast-dLLM v2 occupies the upper-right corner: 2.54Γ faster than Qwen2.5-7B-Instruct at comparable accuracy (~84% vs. ~83.7%), and +5.2% more accurate than Fast-dLLM-LLaDA at similar throughput. This framing is effective because it captures the two-dimensional nature of the problem: the goal isn't just to be faster or more accurate, but to shift the Pareto frontier outward β achieving combinations of speed and quality that were previously unattainable. The paper's claim to "practical deployment" rests on this dual improvement.
However, the framing also reveals a tension in the paper's positioning. The headline claim of "2.5Γ speedup" is specifically at batch size 1 with parallel decoding threshold 0.9 (Figure 4 shows 101.7 vs. 39.1 tokens/sec on GSM8K), while the Figure 1a comparison uses throughput figures that appear to be at some batch size (not explicitly stated). At higher batch sizes, the speedup diminishes (1.5Γ on A100 at batch size 64, 1.8Γ on H100 at batch size 64, Figure 5), suggesting that the advantage is most pronounced in latency-bound settings where the serial bottleneck is most acute.
3. Technical Approach
3.1 Reader Orientation
What the system is: Fast-dLLM v2 is a training recipe and inference pipeline that converts a standard autoregressive language model (specifically Qwen2.5-Instruct) into a block diffusion model β a model that generates text in chunks (blocks), predicting multiple tokens in parallel within each block using iterative masked-token refinement, while still conditioning on previously generated blocks in a left-to-right autoregressive fashion.
What problem it solves and the shape of the solution: The core problem is that standard autoregressive models generate tokens one at a time, creating a serial bottleneck that underutilizes GPU parallelism during inference. The solution is a hybrid architecture that preserves the autoregressive model's learned representations and KV-cache compatibility at the block level (so previously generated context can be efficiently cached and reused) while enabling parallel, multi-token generation within each block (so multiple tokens can be predicted simultaneously, reducing the number of sequential forward passes). The system achieves this through three coordinated mechanisms: (1) a block-wise attention mask that is close enough to the original causal attention to enable data-efficient fine-tuning (~1B tokens), (2) a complementary masking training strategy that ensures every token learns to be predicted in both visible and masked contexts, and (3) a hierarchical caching inference pipeline that reuses computations both across blocks and within partially decoded blocks.
3.2 Big-Picture Architecture (Diagram in Words)
The Fast-dLLM v2 system has five major components, organized around a training pipeline and an inference pipeline that share the same attention mask structure:
-
Pretrained AR Base Model (Qwen2.5-Instruct): The starting point. This is a standard causal transformer language model pretrained on next-token prediction. It serves as the initialization for fine-tuning; all its learned weights, representations, and autoregressive conditioning patterns are preserved as the foundation.
-
Block-Wise Training Pipeline (Section 3.2): Transforms the AR model into a block diffusion model through supervised fine-tuning on instruction data. The pipeline pads and packs sequences into fixed-length blocks, applies random binary masking within each block, duplicates each sample with complementary masks (so every token appears both masked and unmasked), applies a custom block-wise attention mask that enables bidirectional attention within blocks while preserving causal attention between blocks, and trains the model to predict masked tokens using a next-token prediction style shift (predicting position from the hidden state at position ). Only ~1B tokens of fine-tuning data are needed because the attention structure closely resembles the original causal attention.
-
Attention Mask Architecture (Appendix A.2): A composite attention mask that controls which tokens can attend to which others. At training time, it concatenates the noised sequence and clean sequence , applying three sub-mask patterns: block-diagonal (bidirectional within each block of ), offset block-causal ( attends causally to earlier blocks of ), and block-causal ( attends causally within and across blocks). At inference time, it simplifies to bidirectional attention within the current block and causal attention to cached previous blocks.
-
Hierarchical Caching Inference Pipeline (Section 3.3): The deployment-time system that generates text efficiently. It operates at two levels: a block-level cache that stores KV representations of fully decoded blocks (exactly like AR KV-cache, since blocks are generated left-to-right and never change), and a sub-block cache (DualCache from Fast-dLLM) that enables efficient recomputation within partially decoded blocks where tokens are being iteratively unmasked. Within each block, a confidence-aware parallel decoding strategy unmasks multiple high-confidence tokens simultaneously, reducing the number of refinement steps.
-
Confidence-Aware Parallel Decoding (inherited from Fast-dLLM, Section 3.3): The intra-block generation algorithm. At each refinement step, the model predicts probabilities for all currently masked positions. Tokens whose predicted probability exceeds a threshold (set to 0.9 for the optimal speed-accuracy tradeoff) are finalized and unmasked in parallel, while uncertain tokens remain masked for further refinement. This allows the model to generate easy tokens early and spend more computation on ambiguous ones.
Information flow during inference: The prompt is fed through the model to generate the first block β the first block's tokens are iteratively refined using parallel decoding with DualCache sub-block reuse β once the block is complete (all tokens unmasked), its KV representations are stored in the block-level cache β the next block is initialized as fully masked, conditioned on all previous blocks via the block-level cache β the process repeats for the next block, with block-level cache hits progressively reducing the per-block computation cost β generation terminates when an EOS token is produced or a maximum length is reached.
3.3 Roadmap for the Deep Dive
-
First, the block-wise training pipeline and complementary masking, because the entire approach hinges on how the model is fine-tuned β the specific masking strategy, the token shift mechanism, and the block alignment protocol determine whether the adaptation preserves AR quality. Understanding the training recipe is prerequisite to understanding why inference works the way it does.
-
Second, the attention mask design, because it is the architectural "glue" that enables both the training pipeline (simultaneous processing of noised and clean views) and the inference pipeline (block-level KV caching with intra-block bidirectional attention). The attention mask directly encodes the interpolation between autoregressive and diffusion behavior.
-
Third, the loss function and its normalization, because the complementary masking strategy changes how the loss is computed compared to standard masked diffusion β the absence of the normalization factor is intentional and justified by the complementary mask design, and this is a subtle point that affects training dynamics.
-
Fourth, the inference pipeline, because it operationalizes the trained model for efficient text generation. This covers the block-wise autoregressive loop, the hierarchical caching mechanism (block-level + sub-block DualCache), the confidence-aware parallel decoding algorithm, and the batch decoding strategy with length padding.
-
Fifth, the training hyperparameters and compute budget, because the paper's headline claim of data efficiency ("only ~1B tokens") needs to be understood in the context of specific model sizes, learning rates, step counts, and hardware configurations. The tradeoff between training cost and inference throughput gains is central to the paper's practical value proposition.
3.4 Detailed, Sentence-Based Technical Breakdown
This is primarily a systems and training methodology paper whose core idea is that pretrained autoregressive LLMs can be efficiently converted into block diffusion models β preserving their learned representations while enabling parallel multi-token generation β if and only if (1) the block-wise attention mask remains structurally close to the original causal attention, (2) training-time block boundaries are strictly enforced to prevent cross-sample attention leakage, (3) complementary masking ensures every token receives gradient signal in both masked and unmasked states, and (4) hierarchical caching and parallel decoding are integrated at inference to fully exploit the parallelism that the block structure enables.
Block-Wise Training Pipeline and Complementary Masking
The training pipeline transforms a standard autoregressive model into a block diffusion model through supervised fine-tuning. The goal is to teach the model to predict tokens at masked positions using bidirectional context within the same block, while still conditioning on previous blocks in a left-to-right autoregressive manner. The pipeline operates in a specific sequence of steps, each addressing a distinct requirement of the block diffusion paradigm.
Step 1: Block-aligned padding. Given a set of tokenized training samples, each sample is first padded to a length that is an integer multiple of the block size (fixed at 32 for all experiments) by appending [MASK] tokens as needed. These padding tokens are explicitly excluded from loss computation and gradient updates β they are purely structural, ensuring that every sequence naturally divides into an integer number of complete blocks without splitting sample content across block boundaries. Without this padding, when sequences are packed together (concatenated into longer training batches), a block boundary might fall in the middle of a sample, causing tokens from two different training examples to end up in the same bidirectional-attention block β which would create unintended cross-sample attention leakage since the bidirectional attention within blocks would allow tokens from different samples to attend to each other. The padding ensures that every block contains tokens from exactly one training sample, or is entirely padding.
Step 2: Block-aligned packing. After padding, sequences are concatenated into a long token stream and then split into training sequences of fixed context length . Since each padded sequence has a length that is a multiple of , the concatenated stream is also naturally a multiple of 32, meaning that when it is split into length-2048 segments, each segment contains exactly complete, non-overlapping blocks. These blocks are "already aligned by construction" β no block contains tokens from more than one original sample or spans across the training sequence boundary. This block-aligned packing ensures efficient batching (every training sequence uses the full 2048-token context) while maintaining clean block boundaries throughout. The batch size for training is 256, meaning each training step processes tokens.
Step 3: Random binary masking within blocks. For each training sequence, the model operates on individual blocks independently. For each block of size , a binary mask is randomly sampled, where means that position in the block is replaced with a learned [MASK] embedding, and means the original token remains visible. The masking is applied independently per block, not globally across the entire sequence, so different blocks can have different masking patterns within the same training sample. The masking probability is not specified as a single value β the paper describes the complementary masking process (Step 4) which implies that masking ratios are sampled from a distribution, with the key property being that complement pairs are used.
Step 4: Complementary masking strategy. This is the critical training innovation. Rather than masking each sample once, every training sample is duplicated into two views with complementary masks: one view uses mask , and the other view uses the complementary mask , where exactly where and vice versa. These two complementary views are placed together in the same training batch, so the model processes both simultaneously in a single forward pass.
The purpose of complementary masking is to ensure complete supervision coverage: every token in the input sequence appears in both a masked state (where the model must predict it) and an unmasked state (where it serves as visible context for predicting other tokens). In standard masked training with a single random mask at ratio , some tokens might be rarely or never masked during training if the training data volume is limited relative to vocabulary size and sequence length. With complementary masking, since and partition the token set, every position gets exactly 50% masked coverage on average across the two views for any masking pattern. This approximately doubles the effective gradient signal per training sample compared to single-mask training, which is especially important given the paper's goal of data-efficient fine-tuning with only ~1B tokens.
Formally, for any given training example and a sampled mask , the model sees two versions:
- Version 1: tokens where are masked, tokens where are visible
- Version 2: tokens where are masked, tokens where are visible
In both versions, the model learns to predict the original tokens at the masked positions using visible context both from earlier blocks (causally) and from the same block (bidirectionally). The key insight is that every token position gets trained in both roles β as visible context and as prediction target β which the paper argues is essential for the model to learn robust representations that work in both inference modes (where initially all tokens in a new block are masked, and gradually some become visible as decoding proceeds).
Step 5: Token shift for prediction. When predicting a masked token at position , the model does not use its own hidden state at position . Instead, it uses the hidden state at the preceding position to predict . This is called a "shifted-label strategy" and is explicitly adopted "to preserve the pretrained AR model's representation quality" (Section 3.2, citing Ye et al., 2025b,a).
The reason this matters: In a standard autoregressive model trained with next-token prediction, the hidden state at position is optimized to contain information predictive of token β that is the entire training objective. If Fast-dLLM v2 instead used the hidden state at position to predict (as would be natural in a masked autoencoder like BERT), it would require the model to learn an entirely new mapping from hidden-state-at-position- to token-at-position-, discarding the pretrained next-token prediction capability. By keeping the prediction mapping, Fast-dLLM v2 preserves the pretrained model's internal representations and only needs to adapt them to handle the bidirectional block context β a much smaller distribution shift than learning a new prediction direction.
This is a non-obvious design choice with significant consequences. It means that within a block, the first token (position 0 of the block) cannot be predicted using the shifted strategy because there is no position within the block. The first token of each block is therefore either predicted using the last token of the previous block (which is causally available) or, for the very first block, using the prompt context. The paper does not elaborate on edge cases, but the inference procedure handles this naturally since blocks are decoded left-to-right.
Step 6: Concatenation of noised and clean sequences. For each training sample with its sampled mask , the system constructs two sequences:
- : the noised sequence, where positions with are replaced with [MASK] and other positions retain their original tokens
- : the clean sequence, containing the original ground-truth tokens at all positions
These are concatenated along the sequence dimension to form a single input of length , and processed through the transformer in a single forward pass. This concatenation is what enables the joint attention mask design (detailed in the next section) to simultaneously process both the corrupted and clean views, with the clean tokens providing supervision signal to the noised positions and the noised positions receiving bidirectional context from their block.
The concatenated input structure is: , where the first positions contain the noised tokens and the second positions contain the clean tokens. The attention mask controls which positions can attend to which others across this -length sequence.
Key design choice β Why fine-tuning and not training from scratch: The paper explicitly builds on pretrained Qwen2.5-Instruct models rather than training block diffusion models from scratch. The justification is twofold: (1) The block-wise attention mask is structurally "closer to the original AR models" than full-attention diffusion, meaning the pretrained weights already encode useful representations that only need adaptation rather than replacement. (2) Starting from a strong pretrained model provides a much higher quality floor β the model can fall back on its AR capabilities for inter-block conditioning while learning the new intra-block bidirectional skill. Training from scratch (as in LLaDA-8B, Table 1) produces significantly worse results (average score 43.3 for LLaDA-8B vs. 60.3 for Fast-dLLM v2 7B), even when using more total parameters.
Attention Mask Design
The attention mask is the architectural mechanism that defines which tokens can attend to which other tokens during the transformer's self-attention computation. It is a binary matrix where means position can attend to position , and means attention is blocked. The mask is applied before the softmax in each attention head, setting the attention logit for blocked pairs to so they receive zero weight after softmax.
The paper's attention mask is a composite structure built from three sub-patterns, each serving a distinct purpose in the block diffusion framework. The full mask at training time is defined as a block matrix:
where the four quadrants correspond to attention between the four possible source-target region pairs: noisedβnoised (top-left), noisedβclean (top-right), cleanβnoised (bottom-left), and cleanβclean (bottom-right).
The sub-masks operate as follows:
1. Block-Diagonal Mask (, top-left quadrant): This controls attention within the noised sequence . The mask allows a token at position in to attend to token at position in if and only if and belong to the same block. Formally:
where and are indices within the noised portion of the concatenated sequence (positions 1 through ).
What it computes: This enables full bidirectional attention within each block of the noised sequence. A masked token at the beginning of block can attend to a masked or unmasked token at the end of block , and vice versa. This is the "diffusion" aspect β the model uses context from all positions within the current block, both left and right, to predict what each masked token should be. However, a token in block cannot attend to tokens in block or of the noised sequence β those connections are blocked.
Why this form: The block-diagonal structure is what makes this a block diffusion model rather than a full-attention diffusion model. By restricting bidirectional attention to within-block only, the mask preserves the independence of blocks during noised processing β block 's internal refinement does not depend on the noised state of block . This is what enables the inference-time optimization of processing one block at a time: since blocks are independent in the noised view, previously decoded blocks can be completely removed from the noised computation, reducing memory and compute. The block size determines the maximum context window for intra-block bidirectional attention β a design choice that balances the benefits of bidirectional context (larger ) against the cost of iterative refinement (more tokens to decode per block) and the preservation of AR structure (smaller means more blocks, meaning more inter-block AR steps).
2. Offset Block-Causal Mask (, top-right quadrant): This controls attention from the noised sequence to the clean sequence . The mask allows a noised token at position (in some block of ) to attend to clean tokens at positions in if and only if belongs to a block strictly before block . Formally:
where indexes into the noised portion and indexes into the clean portion.
What it computes: This provides the autoregressive conditioning signal. When predicting masked tokens in block of the noised sequence, the model can see the clean (ground-truth) tokens from all previous blocks in the clean sequence. This is the "autoregressive" aspect β each block's predictions are conditioned on the correct tokens from all earlier blocks, just as in standard AR generation where token is conditioned on . The clean tokens serve as perfect teacher-forcing context during training.
Why this form: The offset block-causal structure mirrors the causal attention in standard AR transformers, but at the block level rather than the token level. Instead of token attending to tokens through , block of the noised sequence attends to clean blocks through . This preserves the left-to-right information flow that the pretrained model was originally trained with, making the fine-tuning adaptation smoother. The "offset" refers to the fact that the clean sequence is positioned in the second half of the concatenated input (positions through ), so the attention indices naturally map block of to blocks of .
3. Block-Causal Mask (, bottom-right quadrant): This controls attention within the clean sequence . The mask allows a clean token at position to attend to another clean token at position if and only if is in the same block as or in an earlier block. Formally:
where both and index into the clean portion of the concatenated sequence.
What it computes: This enables the clean tokens to process their own sequence with a block-causal structure β tokens within a block can see each other bidirectionally (same block), and tokens can see all tokens from previous blocks (causal across blocks), but cannot see tokens from future blocks. This is essentially the same attention pattern that the model uses during inference for the prefix context: previously decoded blocks are fully visible to each other (via KV cache) and to the current decoding block.
Why this form: The block-causal mask serves two purposes during training. First, it allows the model to build representations of the clean tokens that incorporate intra-block context (since tokens in the same clean block can attend bidirectionally), which is important because these clean token representations are then attended to by the noised tokens via . Second, it maintains the causal constraint across blocks, preventing the model from using future-clean-block information when processing current-clean-block tokens β this ensures the training setup doesn't create an artificial advantage that would disappear at inference time (where future clean tokens don't exist yet).
4. Zero Mask (bottom-left quadrant): The bottom-left quadrant of is set to 0 β clean tokens cannot attend to noised tokens. This is a design choice that prevents the clean sequence representations from being contaminated by the randomly masked noised sequence, ensuring that the teacher-forcing context () remains clean and does not incorporate information about the specific masking pattern being used for the current training sample.
Training-time forward pass mechanics. With the concatenated input and attention mask, a single transformer forward pass simultaneously computes:
- Representations for all noised tokens , which incorporate bidirectional context from their own block (via ) and causal context from previous clean blocks (via )
- Representations for all clean tokens , which incorporate bidirectional context within blocks and causal context across blocks (via )
The model's prediction heads operate on the noised token representations to predict the original tokens at masked positions. The clean token representations are not used for prediction directly β they serve as context carriers for the noised tokens. This "dual-view in one forward pass" design is computationally efficient because it avoids running separate forward passes for the noised and clean sequences, but it doubles the sequence length from to , which quadratically increases the attention computation cost. The paper uses flex-attention (a PyTorch implementation of efficient sparse attention) to mitigate this cost by exploiting the structured sparsity pattern of .
Inference-time attention mask (Figure 7b). At inference time, the mask is simplified because only one block is being actively decoded at a time, and all previous blocks are already clean and cached:
- The current noised block (the block being decoded) attends bidirectionally to itself (same as for that block)
- The current noised block attends causally to all previous clean blocks (same as but restricted to the active block)
- Previous clean blocks attend to each other causally (same as ), but their KV representations are cached and not recomputed
This means that at each decoding step, the transformer only needs to compute attention for the current block's noised tokens attending to the current block and the cached prefix. The computation cost per step is proportional to rather than to full sequence length squared, and the prefix KV cache eliminates the need to recompute representations for previously decoded blocks.
Training Objective and Normalization
The paper trains the model to minimize a masked-token-only cross-entropy loss over the block-structured input. The loss function is:
where is a training sample, is the sampled binary mask, is the token at position in the noised sequence (which may be [MASK] or the original token), is the original (clean) token at position , denotes all clean tokens from blocks before the block containing position (the autoregressive context), and denotes all tokens (both masked and unmasked) in the block containing position (the bidirectional context).
What it computes: For every position in the noised sequence where the token has been replaced with [MASK] (indicated by the indicator ), the model computes the negative log-likelihood of the true token given the available context. The context consists of two parts: the clean autoregressive prefix (all tokens from blocks before the current one, accessed via in the attention mask) and the within-block context (all tokens in the current block, both masked and unmasked, accessed bidirectionally via ). The expectation is taken over training samples and sampled masks . The total loss for a training sample is the sum of per-position losses across all masked positions in that sample.
Why this form β the missing normalization: Standard masked diffusion training objectives typically include a normalization factor of (where is the masking ratio), as shown in the preliminary equation in Section 3.1:
The factor normalizes the loss by the expected number of masked tokens, ensuring that different masking ratios contribute equally to the overall loss magnitude (otherwise, higher masking ratios would dominate the gradient simply because more tokens are masked). Fast-dLLM v2's loss function conspicuously omits this normalization factor.
The paper explicitly addresses this (Appendix A.3): "This is intentional and justified by our complementary masking strategy." The reasoning is as follows. For each training sample , the complementary masking strategy always generates two views with masks and . This means that for the two views combined, the total number of masked tokens is always exactly (the full sequence length), regardless of the specific masking ratio:
Since the two complementary views are placed in the same batch and contribute equally to the gradient, the effective normalization is already built in β the per-sample loss is automatically normalized by across the complementary pair. If a factor were also included, it would upweight samples with low masking ratios (since would be large for small ) and downweight samples with high masking ratios, introducing an unintended bias. The paper argues that the complementary mask structure makes the standard normalization both unnecessary and harmful.
A subtle detail about the conditioning notation: The paper writes the prediction probability as rather than (as in the standard diffusion objective). This reflects the token shift strategy and the block-structured conditioning. The term indicates that the model conditions on the clean prefix (not noised), since the inter-block conditioning is autoregressive and the clean tokens from previous blocks are available via . The term indicates that within the current block, the model conditions on the noised tokens (some masked, some unmasked, depending on ), accessed bidirectionally via . This is distinct from standard full-attention diffusion where the model conditions on the entire noised sequence at all positions.
Inference Pipeline
The inference pipeline operationalizes the trained block diffusion model for efficient text generation. It operates as a block-by-block autoregressive loop, with each block internally undergoing iterative masked-token refinement using parallel decoding. The pipeline integrates three levels of caching and two levels of parallelism.
Level 1: Block-wise autoregressive loop (outer loop). Generation proceeds one block at a time, in left-to-right order. The prompt (if any) is tokenized and treated as the initial prefix. For each block :
- The block is initialized as a sequence of tokens, all set to [MASK].
- The block is appended to the existing prefix (which consists of all previously decoded blocks, now fully unmasked and fixed).
- The model processes this concatenated sequence using the inference-time attention mask: the current block attends bidirectionally to itself and causally to the prefix; the prefix attends causally to itself.
- Within the current block, iterative refinement proceeds (Level 2 below) until all tokens are unmasked.
- Once the block is fully decoded, its tokens are finalized and cached. The process moves to block .
The block-wise autoregressive loop continues until the model generates an end-of-sequence (EOS) token within a block, or until a maximum number of blocks is reached. This loop mirrors standard AR generation but at block granularity β instead of generating token 1, then token 2, then token 3, the model generates block 1 (all 32 tokens via iterative refinement), then block 2 (another 32 tokens), and so on.
Level 2: Intra-block parallel decoding with confidence thresholding (inner loop). Within each block, tokens are generated through an iterative refinement process rather than being predicted in a single step. The algorithm, inherited from Fast-dLLM (Wu et al., 2025), works as follows:
- At the start of block decoding, all positions in the block are masked.
- A forward pass through the model produces probability distributions over the vocabulary for every masked position (using the shifted prediction strategy: position is predicted from hidden state at ).
- For each masked position, the model's predicted token is the argmax (or sampled, though the paper states "greedy decoding (argmax)" is used for evaluation in Appendix A.4) of the probability distribution, and the associated confidence is the probability assigned to that token.
- Tokens whose confidence exceeds a predefined threshold are finalized β they are unmasked and their values are locked in for the remainder of the block's decoding. Tokens whose confidence is below remain masked for further refinement.
- The block now contains a mix of unmasked (finalized) and masked (still refining) tokens. Another forward pass is run, where the model sees the newly unmasked tokens as additional context for predicting the remaining masked positions.
- Steps 2-5 repeat until all positions in the block are unmasked (or a maximum number of iterations is reached).
The confidence threshold controls the tradeoff between speed and accuracy. When , parallel decoding is effectively disabled β no token is finalized early, and all tokens must go through the full iterative refinement process until none are masked. This produces the highest quality but requires the most forward passes. When is lowered (e.g., to 0.9), the model can finalize high-confidence tokens early, reducing the number of masked positions in subsequent forward passes and therefore reducing the total number of passes needed to decode the full block.
The paper empirically determines the optimal threshold using GSM8K accuracy vs. throughput measurements (Figure 4). With , throughput increases from 39.1 to 101.7 tokens/sec (a 2.6Γ speedup) while GSM8K accuracy drops only marginally (from the non-parallel baseline of approximately 83-84% to roughly 83% β the exact figures are shown visually in Figure 4). Thresholds of 0.8 and 0.7 yield further throughput gains (up to 152.8 tokens/sec, a 4.06Γ speedup) but with more substantial accuracy degradation. The paper selects as the operating point: "Threshold 0.9 is selected, offering a 2.6Γ speedup with minimal accuracy drop."
Level 3: Hierarchical caching for efficiency. The inference pipeline employs two distinct caching mechanisms:
Block-level cache (inter-block): After a block is fully decoded (all tokens unmasked), its Key-Value representations from the final forward pass are stored in a persistent cache. Since future blocks only attend to previous blocks causally (via in the attention mask), and the previous blocks' tokens never change once finalized, these KV cache entries remain valid for all subsequent decoding steps. This is functionally identical to standard AR KV caching β once a token (or block of tokens) is generated, its KV entries are reused rather than recomputed for every subsequent forward pass. The block-level cache eliminates the need to recompute attention for the growing prefix, reducing the per-step computation cost from to , where the quadratic prefix-self-attention term is eliminated.
The paper claims this is a fundamental advantage over full-attention diffusion models, where "they often cannot use KV cache effectively due to bidirectional attention" and approximate caches "do not fundamentally resolve incompatibility of dLLMs with KV cache, since such approximate caches are not equivalent to the original computation" (Section 1). In Fast-dLLM v2, the block cache is an exact cache β the cached KV values are identical to what would be computed from scratch β because the block structure guarantees that past blocks are never modified once finalized and future blocks attend to them causally.
Sub-block cache (intra-block, DualCache): Within a block that is still being decoded, the iterative refinement process involves running multiple forward passes with gradually changing token states (some tokens being unmasked as their confidence exceeds ). The DualCache mechanism from Fast-dLLM (Wu et al., 2025) is employed to avoid recomputing KV entries for positions that haven't changed between iterations.
The DualCache maintains two types of cached KV entries:
- Prefix cache: KV entries for the prefix context (all previously decoded blocks), which never change within a block's decoding and can be reused across all iterations for that block. This is essentially the same as the block-level cache, but scoped to the intra-block refinement loop.
- Suffix cache: KV entries for the portion of the current block that has been finalized (tokens unmasked in previous iterations). These entries are reused as long as the tokens don't change β which they won't, since finalized tokens are locked.
When a new refinement iteration runs, only the KV entries for the currently masked positions need to be computed from scratch. The KV entries for the prefix and the previously finalized tokens within the block are fetched from cache. This is particularly valuable in the later stages of block decoding, when most tokens have been finalized and only a few uncertain positions remain masked β the forward pass computes attention for only those few positions against the full (cached) prefix and intra-block context.
The paper reports (Figure 6b) that the sub-block cache provides "negligible gains when the batch size is small (and memory bandwidth is underutilized)" but "substantial speedup in the compute-bound regime, such as when the batch size is 32." This is because at small batch sizes, the computation is memory-bandwidth-bound (the GPU spends most of its time waiting for data from memory), so recomputing KV entries doesn't significantly impact throughput. At larger batch sizes, computation becomes the bottleneck, and the cache's reduction in redundant computation directly translates to higher throughput. The cache "has no observable effect on model accuracy (Figure 6a), confirming that it is a purely efficiency-enhancing feature without compromising output quality."
Batch decoding with length padding. To support batched inference where different sequences in the batch may have different lengths, the paper uses a right-padding strategy. Each sequence is padded with [MASK] tokens to make its total length divisible by the block size . At each block-decoding step, all sequences in the batch decode the next block in parallel, even if some sequences have already generated their EOS token (in which case their remaining blocks are entirely padding). This "ensur[es] consistent and efficient scheduling on modern hardware" β the GPU can run all sequences through the same operations simultaneously rather than handling variable-length sequences with complex masking logic. The padding tokens are excluded from loss computation (during training) and ignored as output (during inference).
Sub-block size as a decoding granularity knob. The paper introduces the concept of a sub-block size during inference. While the block size is fixed by training (the attention mask structure depends on ), the confidence-aware parallel decoding can be applied at a finer granularity by partitioning each block into sub-blocks and applying the threshold independently within each sub-block.
Table 3 shows that sub-block size affects both accuracy and (implicitly) throughput: size 8 provides the best average accuracy (43.9 HumanEval, 62.0 GSM8K), while smaller sizes (2, 4) and larger sizes (16, 32) perform slightly worse. The paper's interpretation is that sub-block decoding provides finer control over which tokens are finalized when β with sub-block size 32 (equivalent to no sub-blocking, the whole block is one unit), the model must wait for all tokens to reach the confidence threshold before any are finalized; with sub-block size 8, tokens can be finalized independently in groups of 8, allowing the model to commit to high-confidence tokens earlier while continuing to refine uncertain ones.
Critically, the sub-block size is purely an inference-time hyperparameter β it does not require retraining because the training procedure already trains the model to predict tokens at arbitrary positions within blocks. Table 4 confirms that changing the block size at inference time (without retraining) causes substantial degradation: GSM8K drops from 62.0 to as low as 53.2 when inference block size is mismatched from the training block size of 32. This is because the attention mask during training is defined over blocks of size 32, and changing the block size at inference changes the attention pattern that the model learned to expect. Sub-block decoding avoids this by keeping the block structure fixed (the model still sees 32-token blocks with bidirectional attention) while only changing the granularity of the parallel decoding thresholding.
Training Hyperparameters and Compute Budget
The paper's headline claim β that Fast-dLLM v2 requires "only ~1B tokens of fine-tuning" β must be understood in the context of the specific training configuration. The actual training setup is detailed in Appendix A.1.
Model configurations. Two model sizes are trained:
- Qwen2.5-1.5B-Instruct: 1.5 billion parameters
- Qwen2.5-7B-Instruct: 7 billion parameters
Both are fine-tuned using supervised fine-tuning (SFT) on the LLaMA-Nemotron post-training dataset (Bercovich et al., 2025), described as containing "high-quality instruction-following examples covering a broad range of domains." The paper does not specify the exact size of the subset used or the filtering criteria applied.
Hyperparameters for the 1.5B model:
- Training steps: 6,000
- Learning rate:
- Learning rate schedule: linear warmup over the first 500 steps (the paper doesn't specify the decay schedule after warmup, but AdamW is used as the optimizer)
- Context length: 2048 tokens
- Batch size: 256 sequences
- Tokens per step:
- Total tokens processed: billion tokens
- Training hardware: 64 NVIDIA A100 GPUs with DeepSpeed Zero-3
- Training time: approximately 8 hours
Hyperparameters for the 7B model:
- Training steps: 2,500
- Learning rate:
- Learning rate schedule: linear warmup over the first 500 steps
- Context length: 2048 tokens
- Batch size: 256 sequences
- Tokens per step:
- Total tokens processed: billion tokens
- Training hardware: 64 NVIDIA A100 GPUs with DeepSpeed Zero-3
- Training time: approximately 12 hours
Fixed architectural choices:
- Block size for all experiments (both 1.5B and 7B models)
- Training sequences are "right-padded and packed in a block-aligned fashion"
- Sub-block size fixed at 8 during evaluation (unless otherwise stated for ablations)
- Parallel decoding disabled (threshold = 1.0) during evaluation (unless otherwise stated for throughput experiments)
The "~1B tokens" claim: The paper's abstract and introduction claim that Fast-dLLM v2 requires "only βΌ1B tokens of fine-tuning." This figure corresponds specifically to the 7B model's training budget of approximately 1.31 billion tokens. The 1.5B model actually uses approximately 3.15 billion tokens β roughly 3Γ more than the headline figure. The paper's phrasing "βΌ1B tokens" is technically accurate for the main 7B result but elides the scale difference between model sizes. In absolute terms, 1.31 billion tokens is very small by modern LLM post-training standards β for comparison, Dream (Ye et al., 2025b,a) reports using ~580 billion tokens, a ~440Γ difference for the 7B model (not 500Γ β that figure appears to reference Dream's total training budget rather than the exact ratio for the 7B comparison).
What "token" means in the training budget: The paper's token counts refer to the number of tokens processed during training, counting each position in each sequence exactly once. Since the complementary masking strategy processes each sample twice (once with mask and once with ), the effective number of distinct training examples seen is half the token count β approximately 1.5-2.5 billion tokens of unique text content are seen, with each token appearing once masked and once unmasked across the complementary views.
Evaluation configuration (Appendix A.4): Unless otherwise specified, all benchmark evaluations use:
- Greedy decoding (argmax, no temperature sampling)
- Zero-shot prompting for all tasks except GPQA (5-shot)
- Block size = 32, sub-block size = 8
- Parallel decoding disabled (threshold = 1)
- Evaluation frameworks: LM-Eval for non-code tasks, EvalPlus for code tasks (HumanEval, MBPP)
This means that the benchmark results reported in Table 1 (which establish that Fast-dLLM v2 matches or exceeds AR baselines) are obtained with the slowest, highest-quality decoding configuration (threshold = 1.0, no parallel decoding). The 2.5Γ speedup numbers come from a separate throughput evaluation with threshold = 0.9 applied specifically to GSM8K (Figure 4). The paper does not report benchmark results under the accelerated decoding configuration, so the claim "matching accuracy while achieving 2.5Γ speedup" should be understood as two separate measurements from different experimental conditions rather than a single end-to-end demonstration.
Design choice β why block size 32? The paper does not explicitly justify the choice of , but the ablation in Table 4 provides implicit motivation. At inference time, decoding with mismatched block sizes degrades performance severely (e.g., GSM8K drops from 60.2 at to 53.2 at ). This suggests that the training block size must be selected upfront and fixed. The choice of 32 likely balances several factors: (1) large enough to provide meaningful bidirectional context within blocks (enabling the model to resolve ambiguities using both left and right context), (2) small enough that the number of autoregressive steps between blocks is not too small (more blocks means more opportunities for external conditioning and KV cache reuse), (3) small enough that the iterative refinement within blocks converges in a reasonable number of steps, and (4) a power of 2 that aligns well with hardware memory layouts and attention computation patterns. The paper's consistent use of 32 throughout suggests this was determined empirically but not through a systematic ablation reported in the paper.
4. Key Insights and Innovations
Innovation 1: Block-wise attention as a compatibility bridge, not a compromise
The dominant assumption in diffusion language modeling has been that moving away from autoregressive generation requires moving away from the autoregressive attention structure entirely. Full-attention dLLMs like Dream and LLaDA adopt bidirectional attention throughout the entire sequence during diffusion β a clean conceptual break from causal masking that treats the sequence as a holistic unit to be jointly denoised. This is intellectually elegant but practically devastating: it forces the pretrained model to unlearn its causal attention patterns and learn bidirectional ones from scratch, which is why Dream requires ~580B tokens of fine-tuning. The field implicitly treated this as an unavoidable cost of diffusion β if you want parallel generation, you pay for it with massive retraining.
Fast-dLLM v2's conceptual move is to recognize that the attention mask itself is a tunable architectural parameter that can be shaped to remain AR-compatible while still enabling intra-block diffusion. Rather than viewing block diffusion as a compromise between two extremes (AR and full diffusion), the paper reframes it as a compatibility strategy: the block-wise attention mask (block-diagonal for intra-block, offset block-causal for inter-block, block-causal for the clean prefix) is structurally close enough to the original causal attention that the pretrained model's representations transfer with minimal disruption. The model doesn't need to learn that tokens can attend to their left β it already knows that. It only needs to learn that within a small local window (32 tokens), attention can also flow from left to right.
This is a fundamental reframing rather than an incremental improvement. Prior work asked, "How do we train a diffusion model efficiently?" Fast-dLLM v2 asks, "How close can we stay to the AR attention structure while still getting parallelism?" The answer β block-wise causal masking with intra-block bidirectionality β turns out to be close enough that ~1B tokens suffices for adaptation. The evidence is the training efficiency itself: Table 2 shows that even the "naive token shift" baseline (which already uses block-wise structure but lacks padding and complementary masking) achieves a 41.3 average accuracy with the same training budget, meaning the core architectural choice works even without the training recipe refinements. The 500Γ reduction in training data compared to Dream is not a minor optimization β it represents a qualitative difference in what category of solution is being attempted (fine-tuning vs. retraining).
Innovation 2: Complementary masking as an implicit normalization mechanism
Standard masked diffusion training objectives include a normalization factor to account for the varying number of masked tokens at different masking ratios. This normalization is treated as a mathematical necessity β without it, high-masking-ratio examples dominate the gradient because more tokens contribute to the loss. The field has accepted this as part of the standard formulation (D3PM, MDM, LLaDA, Dream all use it).
Fast-dLLM v2 makes a diagnostic observation that is obvious in retrospect but wasn't previously articulated: if you always train on complementary pairs of masks , then the total number of masked positions across the pair is always exactly , regardless of the masking ratio. The normalization is already built into the data construction β every training example contributes exactly loss terms across its two complementary views, naturally balancing the gradient signal without an explicit coefficient.
This is a conceptual contribution about training dynamics in masked modeling, not just a minor trick. The normalization factor has an unintended side effect: it upweights the loss when is small (few tokens are masked) and downweights it when is large. Since the masking ratio is sampled uniformly from , this means the model receives disproportionately strong gradient signals from lightly-masked examples (where prediction is easiest) and weak signals from heavily-masked examples (where prediction is hardest and arguably most valuable). Complementary masking eliminates this bias without needing to tune or schedule the normalization factor.
The paper is explicit about this justification in Appendix A.3, connecting the omission of directly to the complementary mask design. This is a case where a training recipe choice (complementary masking) has a non-obvious interaction with a standard loss formulation, and recognizing that interaction eliminates the need for a term that had been treated as mathematically required. The evidence in Table 2 confirms the importance: adding complementary masking ("+ CM") to the padded baseline improves average accuracy by +2.8 points, and the paper's loss formulation explicitly depends on this design.
Innovation 3: Inference-time block consistency as a hard architectural constraint
Most work on efficient decoding treats the gap between training and inference as a problem to be mitigated β through knowledge distillation, speculative decoding, or approximate caching. Fast-dLLM v2 takes the opposite approach: it treats training-inference consistency as a hard architectural constraint and designs the system around it.
The clearest evidence for this principle is the behavior under block size mismatch (Table 4). When the inference-time block size differs from the training-time block size, performance collapses β GSM8K drops from 60.2 to 53.2 when inference uses block size 2 instead of the training block size of 32. This isn't a gradual degradation; it's a cliff. The reason is that the attention mask pattern (bidirectional within blocks, causal across blocks) is baked into the model's learned representations during fine-tuning. Changing the block size at inference changes which tokens can attend to which others, creating a distribution shift that the model has no mechanism to handle.
This is a diagnostic finding with implications beyond this specific system. It tells us that block diffusion models learn structure-dependent representations where the block boundary is a meaningful architectural primitive, not just an implementation detail. The sub-block decoding strategy (Table 3) succeeds precisely because it respects this constraint: it keeps the 32-token block structure intact for attention purposes while only changing the granularity of the confidence thresholding β a decoding-time heuristic that doesn't affect the attention computation.
The practical consequence is that block size becomes a first-class architectural hyperparameter that must be chosen before training and fixed permanently. This makes block diffusion models less flexible than autoregressive models (where generation length is fully variable) but more flexible than fixed-length diffusion models (where the entire sequence length must be specified upfront). The paper implicitly argues that this constraint is acceptable because the efficiency gains ( throughput) outweigh the loss of flexibility, but it is a real tradeoff that deployers must understand.
Innovation 4: The pretraining-to-inference compute transfer as a quantifiable, difficulty-dependent efficiency frontier
The paper's headline comparison to Dream (580B tokens vs. ~1B tokens) establishes that block diffusion can be dramatically more data-efficient than full-attention diffusion when adapting pretrained AR models. But the intellectual contribution goes deeper: the paper implicitly defines a transfer efficiency frontier that quantifies how much fine-tuning compute is needed to convert AR capabilities into diffusion capabilities as a function of how similar the diffusion attention structure is to the original AR structure.
This is a new diagnostic concept. Prior work on adapting AR models to diffusion (Dream, LLaDA) treated the training cost as a fixed property of the target task β "it takes 580B tokens to teach a 7B model to do diffusion." Fast-dLLM v2 shows that this cost is not fixed; it varies dramatically with the architectural distance between the source and target attention patterns. Full-attention diffusion (bidirectional everywhere) is architecturally far from causal AR, requiring massive retraining. Block diffusion (bidirectional only within 32-token windows) is architecturally close, requiring minimal fine-tuning.
The paper doesn't formalize this as a scaling law or explicitly measure the "attention distance" metric, but the contrast between Dream's 580B tokens and Fast-dLLM v2's ~1B tokens makes the relationship empirically visible. This suggests an optimization problem for future work: given a target throughput improvement and a pretrained AR model, what attention mask structure minimizes the fine-tuning cost while achieving the desired speedup? The block size is one point on this frontier; the paper's ablation in Table 4 (showing that inference-time block size changes are destructive) suggests that this frontier must be explored at training time, not deferred to inference.
This framing also reframes the contribution of the hierarchical caching and parallel decoding (Innovations inherited from Fast-dLLM). These techniques are not novel in themselves β they were proposed in prior work. Their significance in this paper is that they demonstrate what becomes possible once the attention mask compatibility problem is solved. Once you have a block diffusion model that preserves AR quality after ~1B tokens of fine-tuning, you can layer on Fast-dLLM's caching and parallel decoding to convert that quality preservation into actual throughput gains. The 2.5Γ speedup is the downstream consequence; the architectural compatibility insight is the enabling contribution.
The evidence for this insight is distributed across the paper rather than concentrated in one figure. Table 1 shows that Fast-dLLM v2 matches or exceeds AR baselines in accuracy (the compatibility claim). Figure 4 shows that with parallel decoding at threshold 0.9, throughput increases 2.6Γ with minimal accuracy degradation (the efficiency claim). Table 2 shows the importance of training recipe refinements (padding, complementary masking) that are only possible because the attention mask structure is block-based. Together, these establish that the efficient adaptation is not just a training recipe detail but a consequence of a deliberate architectural choice to minimize attention pattern distance from the pretrained model.
5. Experimental Analysis
Evaluation Methodology
-
Dataset. The paper evaluates on a comprehensive suite of benchmarks covering diverse aspects of language capability: HumanEval (code generation, pass@1 with EvalPlus framework), MBPP (code generation, pass@1 with EvalPlus), GSM8K (grade-school math reasoning), MATH (competition math), IFEval (instruction following), MMLU (knowledge-intensive multi-task QA), and GPQA (graduate-level science QA). The LLaMA-Nemotron post-training dataset (Bercovich et al., 2025) is used for fine-tuning, described as "high-quality instruction-following examples covering a broad range of domains." The paper does not specify the exact size of the subset used for fine-tuning beyond the token counts computed from steps Γ batch size Γ context length (Section 4.1, Appendix A.1).
-
Base model(s). All experiments use the Qwen2.5-Instruct model family (Qwen et al., 2025), specifically the 1.5B and 7B parameter variants. The paper argues (Section 4.1) that these models are chosen because they are "widely recognized" and representative of modern instruction-tuned LLMs. The adaptation is performed through supervised fine-tuning (SFT) with the block diffusion training pipeline, not training from scratch β the pretrained weights serve as initialization. This choice is central to the paper's claim of data efficiency, since the pretrained autoregressive representations are preserved rather than learned anew.
-
Metrics. The primary evaluation metric for each benchmark is task-specific accuracy: pass@1 for code generation tasks (HumanEval, MBPP) computed via the EvalPlus framework, exact match accuracy for GSM8K and MATH, instruction-following accuracy for IFEval, and multiple-choice accuracy for MMLU and GPQA. Aggregate performance is reported as a simple average score ("Avg.") across all seven benchmarks (Section 4.2, Table 1). Throughput is measured in tokens per second, computed by measuring total generation time on NVIDIA A100 and H100 GPUs at specified batch sizes (Figures 1b, 4, 5). The paper does not describe the exact timing protocol (e.g., whether prompt processing time is included, whether multiple runs are averaged, or how sequence length variation is handled).
-
Baselines. The paper compares against multiple categories of models summarized in Table 1. Autoregressive baselines: LLaMA-3.2 (1.2B), SmolLM 2 (1.7B), Qwen2.5-1.5B, Qwen2.5-7B, and two Nemo-FT variants (Qwen2.5-1.5B-Nemo-FT and Qwen2.5-7B-Nemo-FT) which are the same base models fine-tuned with standard next-token prediction (NTP) loss on the same LLaMA-Nemotron dataset for the same number of training steps. These NTP-tuned baselines are critical controls because they isolate the effect of the block diffusion training recipe from the effect of additional fine-tuning data. Full-attention diffusion baselines: LLaDA-8B (Nie et al., 2025, trained from scratch on MDM loss), LLaDA-1.5 (Zhu et al., 2025, with variance-reduced preference optimization), LLaDA-MoE (7B mixture-of-experts variant), and Dream-7B (Ye et al., 2025a, adapted from Qwen2.5-7B into full-attention diffusion using ~580B tokens). Dream is the most directly comparable baseline because it shares the same base model and the same goal of converting AR to diffusion.
-
Generation budget / compute accounting. The paper does not use a unified "generation budget" or FLOPs accounting framework as found in scaling law analyses. Instead, it uses two distinct metrics in different contexts. For accuracy comparisons (Table 1), all models are evaluated using greedy decoding (argmax) with parallel decoding disabled (threshold = 1.0), so the "compute" spent per benchmark is that benchmark's standard evaluation protocol (number of test examples Γ tokens generated per example). For throughput comparisons (Figures 1, 4, 5), speed is measured in tokens per second on specific GPU hardware (A100 or H100) at specified batch sizes (1, 4, 32, 64). For the block diffusion models, throughput depends on the confidence threshold Ο (Figure 4) and the sub-block size (Figure 6b). The paper does not normalise for total FLOPs spent, which means the accuracy and throughput measurements come from different experimental configurations and cannot be directly multiplied to obtain a unified efficiency metric.
-
Cross-validation / statistical protocol. The paper does not report any cross-validation, statistical significance testing, confidence intervals, or error bars on any experimental result. All benchmark numbers in Table 1 are single-point estimates from a single evaluation run on the standard test sets. Throughput measurements in Figures 4, 5, and 6b are similarly presented as single values without variance estimates. The paper does not discuss how sensitive results are to random seeds, data ordering, or masking pattern sampling during training. This absence is notable given that the training budgets are small (~1β3B tokens) and the fine-tuning runs are conducted only once β there is no evidence presented about the stability of the adaptation process across multiple training runs.
Main Quantitative Results
Accuracy Comparisons Against Baselines (Table 1)
The headline result is that Fast-dLLM v2 models match or surpass the strongest baselines at their respective parameter scales while being diffusion-based decoders. At the 1.5B scale, Fast-dLLM v2 achieves an average score of 45.0, outperforming all 1B-scale baselines including Qwen2.5-1.5B (44.3), Qwen2.5-1.5B-Nemo-FT (44.3), SmolLM 2 (40.7), and LLaMA-3.2 (35.9). The improvement over the Nemo-FT baseline is modest (+0.7 points) but consistent β Fast-dLLM v2 scores higher on HumanEval (43.9 vs. 37.2), GSM8K (62.0 vs. 58.5), and IFEval (47.0 vs. 39.4), while scoring lower on MATH (38.1 vs. 43.5) and GPQA (27.7 vs. 31.0). This suggests the block diffusion adaptation redistributes capability across tasks rather than uniformly improving.
At the 7B scale, Fast-dLLM v2 achieves an average of 60.3, surpassing Qwen2.5-7B-Nemo-FT (59.6), Qwen2.5-7B (58.2), and Dream-7B (57.6). The gap over Dream is +2.7 points on average, with particularly large advantages on HumanEval (63.4 vs. 57.9) and MBPP (63.0 vs. 68.3 β note this is a negative gap; Dream scores higher on MBPP base), and a smaller advantage on GSM8K (83.7 vs. 81.0). Compared to the AR Nemo-FT baseline, Fast-dLLM v2 shows large improvements on code tasks (HumanEval: 63.4 vs. 52.4, MBPP: 63.0 vs. 57.1) and smaller differences on knowledge tasks (MMLU: 66.6 vs. 68.6, GPQA: 31.9 vs. 34.2), with GSM8K essentially tied (83.7 vs. 84.1). These patterns suggest the block diffusion fine-tuning provides stronger benefits for code generation than for factual knowledge recall.
A critical observation is the comparison between LLaDA-8B (trained from scratch on MDM loss) and Fast-dLLM v2 (7B). Despite having more parameters (8B vs. 7B), LLaDA-8B achieves only 43.3 average β far below Fast-dLLM v2's 60.3. This demonstrates that starting from a pretrained AR model and adapting it is dramatically more effective than training a diffusion model from scratch, at least at this parameter scale and training budget. The LLaDA-1.5 variant (which adds preference optimization) closes some of this gap but the paper does not report a complete set of benchmarks for it (HumanEval+, MBPP+, and GPQA are missing), making the comparison incomplete.
The Nemo-FT baselines serve an important control function. Both Qwen2.5-1.5B-Nemo-FT and Qwen2.5-7B-Nemo-FT are trained on the same LLaMA-Nemotron dataset for the same number of steps as their Fast-dLLM v2 counterparts, but using standard next-token prediction loss. The fact that Fast-dLLM v2 matches or exceeds these baselines (60.3 vs. 59.6 at 7B, 45.0 vs. 44.3 at 1.5B) indicates that the block diffusion training recipe does not degrade the model's general capabilities β it preserves them while additionally enabling parallel decoding. This is the paper's "lossless adaptation" claim in quantitative form.
Throughput Results Under Parallel Decoding (Figures 1, 4, 5)
The paper measures throughput under multiple hardware configurations and batch sizes, with the headline claim being a 2.5Γ speedup over standard AR decoding. The specific numbers come from different experimental conditions and should be disaggregated.
Figure 1b: Throughput at batch sizes 1 and 4 on A100. At batch size 1, Fast-dLLM v2 achieves 113.2 tokens/sec compared to Qwen2.5-7B-Instruct's 39.5 tokens/sec β a 2.87Γ speedup. At batch size 4, the corresponding figures are 217.5 vs. 102.5 tokens/sec, a 2.12Γ speedup. The paper does not state the confidence threshold or sub-block cache configuration used for these measurements. Fast-dLLM-Dream (7B) achieves 48.2 and 54.4 tokens/sec at batch sizes 1 and 4 respectively, while Fast-dLLM-LLaDA (8B) achieves 30.2 and 34.1 tokens/sec. This establishes that block diffusion provides substantially higher throughput than full-attention diffusion across batch sizes, even when both use the Fast-dLLM acceleration techniques (DualCache, parallel decoding).
Figure 1a: Throughput-accuracy tradeoff on GSM8K. This figure places Fast-dLLM v2 at the Pareto frontier: it achieves ~84% GSM8K accuracy (similar to Qwen2.5-7B-Instruct) while delivering more than double the throughput (~100 vs. ~40 tokens/sec). Compared to Dream-7B, Fast-dLLM v2 achieves both higher accuracy (~84% vs. ~81%) and higher throughput (~100 vs. ~80 tokens/sec). Compared to Fast-dLLM-LLaDA (8B), Fast-dLLM v2 achieves +5.2% higher accuracy at similar throughput. The specific throughput numbers in this figure are rounded and the batch size is not stated, making it a qualitative comparison rather than a precisely reproducible measurement.
Figure 4: Throughput-accuracy sweep under confidence threshold variation (GSM8K, 7B model). This is the most detailed single-benchmark analysis. At threshold Ο = 1.0 (no parallel decoding), accuracy is approximately 83.5% (read from the figure; the paper doesn't state the exact number) and throughput is 39.1 tokens/sec. As Ο is lowered:
- Ο = 0.9: accuracy ~83%, throughput 101.7 tokens/sec (2.60Γ speedup)
- Ο = 0.8: accuracy ~80%, throughput 127.2 tokens/sec (3.25Γ speedup)
- Ο = 0.7: accuracy ~76%, throughput 138.2 tokens/sec (3.53Γ speedup)
- Ο = 0.6: accuracy ~67%, throughput 152.8 tokens/sec (3.91Γ speedup)
- Ο = 0.5: accuracy ~62%, throughput 152.8 tokens/sec (plateau)
The paper selects Ο = 0.9 as the operating point, labeling it "2.6Γ speedup with minimal accuracy drop." At this threshold, accuracy drops from the non-parallel baseline by approximately 0.5 percentage points β the paper calls this "minimal." The throughput gain from 39.1 to 101.7 tokens/sec represents the net effect of parallel decoding within blocks. However, it is worth noting that the non-parallel baseline at Ο = 1.0 itself represents the block diffusion model without parallel decoding acceleration, not the original Qwen2.5-7B AR model. Figure 1b shows that even at Ο = 1.0 (no parallel decoding), the block diffusion model achieves 113.2 tokens/sec at batch size 1 β much higher than the 39.1 tokens/sec reported in Figure 4. This discrepancy suggests that the Figure 4 measurements are taken under different conditions (possibly different batch sizes, sequence length distributions, or measurement protocols) than Figure 1b, and the paper does not explain the difference.
Figure 5: Throughput scaling with batch size (A100 vs. H100). At batch size 1, diffusion achieves approximately 200 tokens/sec on H100 vs. AR's ~120 tokens/sec (~1.67Γ). At batch size 64, diffusion achieves approximately 900 tokens/sec on H100 vs. AR's ~500 tokens/sec (~1.8Γ). On A100, the corresponding speedups are ~1.5Γ at batch size 64. The paper highlights that the speedup is larger on H100 ("newer hardware architectures where parallelism can be better exploited"), which is consistent with diffusion decoding benefiting more from hardware with higher computational throughput (since it does more parallel computation per forward pass but also more total computation). However, the paper does not normalize for total FLOPs, so it's unclear whether the throughput advantage represents genuine efficiency (less total computation per token) or just better hardware utilization of a more computationally intensive process (more FLOPs per token but better parallelism).
Performance Under Mismatched Inference Configurations (Tables 3, 4)
Table 3: Sub-block size ablation (1.5B model). When varying sub-block size at inference (keeping training block size fixed at 32), performance varies non-monotonically:
- GSM8K peaks at sub-block size 2 (62.8) and degrades gradually to 60.2 at size 32
- HumanEval peaks at sub-block size 8 (43.9) and degrades sharply to 38.4 at size 32
- HumanEval+ peaks at sizes 4β8 (40.2) and degrades to 34.8 at size 32
The paper interprets size 8 as "optimal" based on average across tasks, but the task-dependent optimal values suggest the sub-block size controls a speed-accuracy tradeoff that varies by task type. Code generation (HumanEval) benefits from finer granularity (smaller sub-blocks) more than math reasoning (GSM8K), possibly because code tokens have more local dependencies that benefit from earlier commitment.
Table 4: Block size mismatch at inference (1.5B model). This ablation demonstrates a critical finding: changing the block size at inference time without retraining causes severe degradation:
- GSM8K: drops from 60.2 at block size 32 (matching training) to 53.2 at block size 2 β a 7-point accuracy loss
- HumanEval: drops from 38.4 at size 32 to 37.8 at size 2, with non-monotonic behavior (peaks at sizes 4β8 at 43.3)
The asymmetry between GSM8K and HumanEval is notable: GSM8K is much more sensitive to block size mismatch than HumanEval, suggesting that math reasoning relies more heavily on the specific attention structure learned during training (bidirectional context within 32-token windows) while code generation is more robust to attention pattern changes. The non-monotonic pattern in HumanEval (sizes 4 and 8 outperform size 32 even though the model was trained with size 32) is surprising and the paper does not discuss it.
Ablation Studies and Robustness Checks
Training recipe ablation (Table 2, 1.5B model): The baseline "naive token shift" (random masking within blocks, shifted-label prediction, no padding, no complementary masking) achieves an average accuracy of 41.3. Adding block-aligned padding ("+ pad") improves average accuracy to 42.2 (+0.9 points), with the largest gains on IFEval (39.9 β 45.8, +5.9 points) and GSM8K (59.0 β 60.1, +1.1 points). The IFEval gain suggests that preventing cross-sample attention leakage (which padding achieves by ensuring clean block boundaries during sequence packing) is particularly important for instruction-following tasks where sample-specific formatting and structure might otherwise bleed across block boundaries. Further adding complementary masking ("+ pad + CM") boosts average accuracy to 45.0 (+2.8 points over "+ pad", +3.7 points over naive), with broad gains across GSM8K (60.1 β 62.0), HumanEval (38.4 β 43.9), MBPP (45.2 β 50.0), IFEval (45.8 β 47.0), and MMLU (53.5 β 55.1). GPQA is essentially flat across ablations (27.7β27.9), suggesting it is insensitive to the training recipe improvements β possibly because graduate-level QA relies on factual knowledge encoded during pretraining that fine-tuning on instruction data doesn't substantially modify. MATH shows a slight degradation (37.3 β 37.0 β 38.1), with the complementary masking variant partially recovering. The cumulative +3.7 point gain from the full recipe establishes that each component (padding, complementary masking) contributes independently to the final performance.
Sub-block cache effectiveness (Figure 6, 7B model): The sub-block cache (DualCache within blocks) has no measurable effect on accuracy regardless of sub-block size or batch size (Figure 6a). All four configurations (batch 1/cache no, batch 1/cache yes, batch 32/cache no, batch 32/cache yes) produce identical accuracy across sub-block sizes 4, 8, and 16 β the lines overplot perfectly. This confirms the paper's claim that the cache is a "purely efficiency-enhancing feature." However, the cache's throughput benefit is highly conditional: at batch size 1, it provides "negligible gains" (Figure 6b, where the two batch-1 curves nearly overlap), while at batch size 32 it provides "substantial speedup" (the batch-32 cache-yes curve is visibly higher than batch-32 cache-no across all sub-block sizes). At sub-block size 16 with batch size 32, the cache increases throughput from approximately 250 to approximately 380 tokens/sec β a ~1.5Γ improvement within the already-accelerated diffusion decoding. This batch-size dependence is characteristic of compute-bound vs. memory-bandwidth-bound regimes and demonstrates that the cache's value scales with deployment scale.
Sub-block size effects on throughput (Figure 6b): Larger sub-block sizes consistently increase throughput across all configurations because they reduce the number of sequential forward passes needed to decode each block (more tokens are processed in parallel per forward pass). At batch size 32 with cache enabled, throughput increases from approximately 280 tokens/sec at sub-block size 4 to approximately 380 tokens/sec at sub-block size 16. However, Figure 6a shows that larger sub-block sizes also slightly degrade accuracy (consistent with Table 3), creating a throughput-accuracy tradeoff. The paper does not provide a unified analysis treating sub-block size and confidence threshold as jointly tunable parameters, which would be necessary for a practitioner to select the optimal operating point for a given deployment.
Inference-time block size mismatch (Table 4): Discussed above under quantitative results. The key non-obvious finding is the asymmetry between GSM8K (large sensitivity) and HumanEval (low sensitivity, non-monotonic), which the paper does not explain or investigate further. This is a robustness concern: if different tasks have different sensitivities to block size, then a single block size chosen at training time may be suboptimal for some deployment tasks, and there is no mechanism to adapt it without retraining.
Parallel decoding threshold sweep (Figure 4): This ablation establishes the throughput-accuracy Pareto curve for the confidence threshold parameter on GSM8K. The non-obvious finding is that throughput plateaus below Ο = 0.6 (both Ο = 0.6 and Ο = 0.5 achieve ~152.8 tokens/sec) while accuracy continues to degrade (from ~67% to ~62%). This suggests that at very low thresholds, the model finalizes most tokens in the first one or two refinement steps, eliminating the need for additional forward passes but producing poor-quality outputs because low-confidence tokens are committed before the model has seen sufficient context from their neighbors. The paper does not investigate whether this plateau is due to all tokens being finalized in the first step (saturating the possible parallelism) or due to some other bottleneck.
Model scale comparison (1.5B vs. 7B, Table 1): While not presented as a formal ablation, the paper provides results for both model sizes trained with the same recipe and dataset (though different step counts and learning rates). The 7B model's average of 60.3 is substantially higher than the 1.5B model's 45.0, as expected from scaling laws. However, the relative improvement over the respective Nemo-FT baselines is larger at 7B (+0.7 points) than at 1.5B (+0.7 points β identical in absolute terms). The paper does not discuss whether the block diffusion adaptation becomes more or less effective with model scale, which would be relevant for practitioners considering applying this method to even larger models.
Critical Assessment
Claim 1: "Lossless adaptation" with ~1B tokens
The paper frames its data efficiency as "lossless adaptation" β converting an AR model to block diffusion without degrading performance. The evidence for this claim comes from Table 1, where Fast-dLLM v2 (7B) achieves 60.3 average vs. Qwen2.5-7B-Nemo-FT's 59.6. On its face, this supports the claim: the block diffusion model slightly outperforms the AR baseline fine-tuned on the same data.
However, "lossless" is an overstatement for several reasons. First, the comparison is to a specific baseline (Nemo-FT) that is itself fine-tuned on only the LLaMA-Nemotron dataset for the same number of steps. This baseline may not represent the best possible AR model one could obtain with the same compute budget β a differently tuned AR fine-tuning run might outperform both. Second, the per-task results show meaningful variation: Fast-dLLM v2 underperforms Nemo-FT on MATH (61.6 vs. 72.0) and GPQA (31.9 vs. 34.2), while overperforming on code tasks. Calling this "lossless" masks that the adaptation shifts the model's capability profile β it gains on some tasks and loses on others, with the net average being slightly positive. A more precise characterization would be that the adaptation is approximately capability-preserving on average but redistributes performance across tasks.
Third, the 1.5B results complicate the narrative. While the 1.5B Fast-dLLM v2 also matches its Nemo-FT baseline (45.0 vs. 44.3), the 1.5B model used 3.15B tokens β 3Γ more than the "~1B" headline. The paper's claim that Fast-dLLM v2 works with "only ~1B tokens" is specifically true for the 7B model but not for the 1.5B model. Since both model sizes are presented as evidence for the method's effectiveness, the training cost is not uniform across scales. This matters because a practitioner wanting to apply the method to a new model size would not know whether to budget ~1B tokens or ~3B tokens β the scaling behavior of the required fine-tuning data with model size is unexplored.
Claim 2: 2.5Γ speedup without compromising quality
The paper's headline speedup claim is supported by Figure 4 for GSM8K specifically: at Ο = 0.9, throughput increases from 39.1 to 101.7 tokens/sec (2.60Γ) with "minimal accuracy drop." The evidence is clear for this single-benchmark, single-configuration measurement.
However, the generality of this claim is limited in several ways. First, the speedup is measured only on GSM8K β the paper does not provide throughput-accuracy tradeoff curves for any other benchmark. Given that Table 3 shows task-dependent optimal sub-block sizes and Table 4 shows task-dependent sensitivity to block size mismatch, it is likely that the optimal Ο and the shape of the throughput-accuracy curve vary by task. A 2.6Γ speedup on GSM8K does not guarantee a 2.6Γ speedup on HumanEval or MMLU.
Second, the baseline for the speedup computation (39.1 tokens/sec at Ο = 1.0) is the block diffusion model itself without parallel decoding, not the original Qwen2.5-7B AR model. Figure 1b shows the AR model achieves 39.5 tokens/sec at batch size 1, very close to the 39.1 baseline in Figure 4 β but the 39.1 in Figure 4 is for the block diffusion model without parallel decoding, and Figure 1b shows the block diffusion model achieving 113.2 tokens/sec at batch size 1 even without specifying the threshold. There is a significant unexplained discrepancy between these throughput numbers that makes it impossible to determine the true speedup relative to the AR baseline under controlled conditions. The paper needs a single table showing: AR throughput at batch size B, Fast-dLLM v2 throughput at batch size B with Ο = 1.0, and Fast-dLLM v2 throughput at batch size B with Ο = 0.9, all measured under identical conditions (same hardware, same prompts, same sequence lengths, same measurement protocol). This is not provided.
Third, the "without compromising quality" claim is supported only by the small accuracy drop in Figure 4 (approximately 0.5 percentage points on GSM8K). But the benchmark results in Table 1 are all evaluated with Ο = 1.0 (parallel decoding disabled). The paper does not report any benchmark results at Ο = 0.9, so there is no evidence that the accelerated model maintains competitive accuracy on HumanEval, MATH, MMLU, or any task other than GSM8K. The paper's framing conflates two separate findings β "matches AR accuracy on benchmarks when evaluated without parallel decoding" (Table 1) and "achieves 2.5Γ speedup on GSM8K with parallel decoding" (Figure 4) β without demonstrating that both hold simultaneously.
Claim 3: 500Γ reduction in training data compared to Dream
This claim is numerically supported: Dream uses ~580B tokens, Fast-dLLM v2 uses ~1B tokens for the 7B model, a factor of ~500Γ. The comparison is fair in that both methods adapt pretrained Qwen2.5-7B models to diffusion-based decoding.
However, the comparison elides several relevant differences. Dream's 580B tokens are used for full-attention diffusion adaptation β a fundamentally harder problem because the model must learn bidirectional attention patterns across the entire sequence. Fast-dLLM v2's ~1B tokens are used for block-wise adaptation β an easier problem because the attention pattern is closer to the original AR structure. The 500Γ reduction is therefore partly a measure of architectural similarity rather than training methodology superiority. This doesn't invalidate the claim β the paper explicitly argues that the block-wise structure is what enables data efficiency β but it means the claim should be understood as "block diffusion is 500Γ more data-efficient than full-attention diffusion for adapting AR models" rather than "our training recipe is 500Γ better than Dream's training recipe."
Additionally, the 1.5B model uses 3.15B tokens (not ~1B), so the 500Γ factor does not hold uniformly across model scales. And the comparison is to only one full-attention diffusion model (Dream); concurrent block diffusion works like SDAR and D2F are mentioned but not quantitatively compared in terms of training data requirements.
Claim 4: State-of-the-art efficiency among dLLMs
Table 1 supports this claim for accuracy: Fast-dLLM v2 (7B) at 60.3 average outperforms Dream-7B (57.6), LLaDA-8B (43.3), and LLaDA-1.5 (incomplete results). Figure 1b supports it for throughput: Fast-dLLM v2 at 113.2/217.5 tokens/sec (batch sizes 1/4) substantially exceeds Dream's 48.2/54.4 and LLaDA's 30.2/34.1.
The weakness is that the paper does not provide a combined accuracy-at-speed metric. We know from Table 1 that Fast-dLLM v2 is more accurate than Dream when both use greedy decoding, and from Figure 1b that it is faster. But we don't know whether Dream with aggressive parallel decoding (lowering its own confidence threshold) might match Fast-dLLM v2's throughput at some accuracy level, or vice versa. The two-dimensional Pareto frontier is only sketched for GSM8K in Figure 1a, where Fast-dLLM v2 clearly dominates, but this single-task comparison is insufficient to claim state-of-the-art status across the full benchmark suite.
Missing experiments that would strengthen the paper
Several experiments are conspicuously absent:
-
Benchmark results at Ο = 0.9 (the selected operating point): The paper selects Ο = 0.9 based on GSM8K throughput-accuracy tradeoff but never evaluates whether this threshold preserves accuracy on other benchmarks. This is the most important missing experiment for the paper's practical claims.
-
Per-benchmark throughput-accuracy tradeoff curves: The optimal threshold likely varies by task (as sub-block size does in Table 3), but only GSM8K is characterized.
-
Multiple training runs with different seeds: With only ~1β3B tokens of fine-tuning, training variance could be substantial. The paper provides no evidence that the reported results are stable.
-
FLOPs-matched comparison between AR and diffusion models: The paper compares throughput (tokens/sec) but not total FLOPs. Since diffusion decoding may use more FLOPs per token (multiple forward passes, bidirectional attention), a FLOPs-normalized comparison would reveal whether the throughput gains represent genuine computational efficiency or just better hardware utilization.
-
Scaling behavior with model size beyond 7B: The method works at 1.5B and 7B, but whether the data efficiency advantage persists at larger scales (13B, 70B) is unknown.
-
Comparison to speculative decoding and other AR acceleration methods: The paper frames diffusion as an alternative to AR, but the practical alternative for latency reduction is often speculative decoding or Medusa-style multi-token prediction, which are not compared.
-
Analysis of failure modes: The paper provides case studies in Appendix B showing successful outputs, but no analysis of when or why the block diffusion model produces incorrect results. Understanding failure modes is essential for deployment.
-
Sequence length scaling: All experiments use context length 2048. How throughput and accuracy scale with generation length (which determines the number of blocks and thus the number of autoregressive steps) is unexplored.
Summary of evidential strength
The paper's strongest evidence is for its core architectural claim: that block diffusion with AR-compatible attention masking enables data-efficient adaptation of pretrained LLMs. Table 1 (accuracy parity with AR baselines), Table 2 (importance of training recipe components), and the 500Γ comparison to Dream collectively make this case convincingly.
The paper's weakest evidence is for its practical deployment claims. The 2.5Γ speedup is demonstrated on only one benchmark (GSM8K), under measurement protocols that are inconsistent with other reported throughput numbers, and without showing that the accelerated model maintains competitive accuracy on the full benchmark suite. The "without compromising generation quality" framing extrapolates from a single 0.5-percentage-point accuracy measurement on GSM8K to a general claim that is not validated.
The absence of any statistical rigor (no error bars, no multiple runs, no significance testing) is a genuine limitation given the small training budgets and single-evaluation protocol. A practitioner deciding whether to adopt this method cannot assess how reliably the reported results would replicate.
6. Limitations and Trade-offs
6.1 The 2.5Γ Speedup Is Validated on Only One Benchmark Under Inconsistent Measurement Conditions
The assumption or constraint. The paper's headline throughput claim β "up to 2.5Γ speedup over standard AR decoding without compromising generation quality" (Abstract, Section 1) β is based primarily on a single measurement: GSM8K accuracy vs. throughput at varying confidence thresholds, shown in Figure 4. At threshold Ο = 0.9, throughput increases from 39.1 to 101.7 tokens/sec (2.60Γ) with what the paper describes as "only a marginal drop in GSM8K accuracy" (Section 4.2). The parallel decoding threshold and the corresponding speedup are selected based on this one benchmark alone. The benchmark results in Table 1 that establish accuracy parity with AR baselines are all evaluated under a completely different configuration: "parallel decoding disabled (threshold = 1)" (Section 4.1), meaning the model uses the slowest possible decoding mode for the accuracy measurements that support the "without compromising quality" claim.
The consequence. A practitioner cannot determine from the paper what speedup to expect on their specific task, nor what accuracy degradation to tolerate. The optimal confidence threshold likely varies by task β Table 3 demonstrates that the optimal sub-block size differs between GSM8K (where size 2 achieves the highest accuracy, 62.8) and HumanEval (where size 8 is optimal, 43.9), indicating that different tasks have different sensitivity to decoding granularity. The same almost certainly holds for the confidence threshold, but no per-task threshold sweep is provided. A deployment targeting code generation might need a different Ο than a deployment targeting math reasoning, and choosing the wrong Ο could produce either unnecessarily slow generation (if Ο is too high) or unacceptable accuracy degradation (if Ο is too low). The paper provides no guidance for this per-task tuning.
What evidence exists in the paper. The throughput-accuracy tradeoff is characterized only for GSM8K (Figure 4). Additionally, the throughput numbers themselves are inconsistent across figures. Figure 4 reports the block diffusion model at Ο = 1.0 achieving 39.1 tokens/sec, but Figure 1b reports the same model achieving 113.2 tokens/sec at batch size 1 (threshold unspecified). These differ by nearly 3Γ for what should be comparable conditions, and the paper does not explain the discrepancy. This makes the baseline from which the 2.6Γ speedup is calculated uncertain: is it 2.6Γ over the model's own non-parallel mode at 39.1 tokens/sec, or over some other baseline? The AR baseline (Qwen2.5-7B-Instruct) achieves 39.5 tokens/sec at batch size 1 in Figure 1b β essentially identical to the 39.1 in Figure 4 β but Figure 1b simultaneously shows the block diffusion model at 113.2 tokens/sec, suggesting a much larger speedup than 2.5Γ under some conditions. The measurement protocol (batch size, prompt lengths, whether prompt processing is included, number of measurement runs) is never specified for any throughput figure, making the numbers non-reproducible and their relationships to each other opaque.
Mitigation status. Not addressed. The paper does not acknowledge that the speedup is validated on only one benchmark, does not explain the throughput discrepancies between Figure 4 and Figure 1b, and does not provide throughput-accuracy curves for any task other than GSM8K. Section 5 (Conclusion) states the 2.5Γ speedup as a general property of the method without qualification. The paper would need per-task threshold sweeps, a clearly specified measurement protocol, and benchmark results at the selected operating point (Ο = 0.9) across the full evaluation suite to substantiate the claim that the speedup comes "without compromising generation quality."
6.2 The Difficulty Estimation Cost for Block Diffusion Is Not Accounted for in the Efficiency Claims
The assumption or constraint. The paper measures throughput during generation β the forward passes required to decode tokens given a trained model β but does not account for the cost of fine-tuning the AR model into a block diffusion model. The training cost is reported as "only ~1B tokens" for the 7B model (approximately 1.31 billion tokens, or 2,500 steps at batch size 256 with context length 2048), running for "12 hours" on 64 NVIDIA A100 GPUs (Appendix A.1). This is presented as a favorable comparison to Dream's ~580B tokens, a 500Γ reduction. However, compared to deploying the original AR model directly (which requires zero additional fine-tuning tokens and zero GPU-hours beyond what was already spent on pretraining), the block diffusion adaptation imposes a non-trivial upfront cost.
The consequence. The efficiency gains from block diffusion (2.5Γ throughput improvement at inference) must be amortized against the training cost. For a deployment that will serve a very large number of inference requests over its lifetime, the 12-GPU-hour training cost is negligible compared to the inference savings. But for deployments with modest inference volume β a research lab evaluating on a few thousand test examples, or a startup with low user traffic β the training cost may dominate. The paper's framing treats the fine-tuning cost as a one-time expense and the throughput gain as a perpetual benefit, which is valid only for high-volume deployments. The breakeven point (how many inference tokens must be generated before the training cost is recovered through faster inference) is not calculated or discussed. For the 1.5B model, which requires approximately 8 GPU-hours of training (6,000 steps) and processes 3.15B tokens (3Γ more than the "~1B" headline figure), the breakeven point would be correspondingly higher.
Additionally, the training cost is measured only for the specific base model (Qwen2.5-Instruct) at specific scales (1.5B and 7B). A practitioner wishing to adapt a different base model (e.g., LLaMA-3, Mistral, DeepSeek) or a different scale (e.g., 13B, 70B) has no data on how the required fine-tuning budget scales. The paper's claim that the method requires "only ~1B tokens" is specific to Qwen2.5-7B-Instruct with the LLaMA-Nemotron dataset and the chosen hyperparameters. Extrapolating to other settings is unsupported.
What evidence exists in the paper. The training cost is reported in Appendix A.1 (steps, batch size, context length, GPU-hours) but is never amortized against inference savings or compared to the cost of simply using the original AR model. The throughput gains in Figures 1, 4, and 5 are presented as gross improvements without any discussion of the net efficiency including training overhead. The paper does not provide a FLOPs-matched comparison between (a) training + deploying the block diffusion model and (b) simply deploying the original AR model, which would be the relevant metric for a practitioner deciding whether to adopt the method.
Mitigation status. Not addressed. The paper frames the training cost exclusively as a favorable comparison to Dream ("500Γ reduction") rather than as a cost to be weighed against inference savings. No breakeven analysis, no scaling study of training cost with model size, and no discussion of how training cost varies with base model architecture are provided. The paper would need to report total training FLOPs (not just token count), characterize how training cost scales with model size for a fixed block size, and provide a breakeven analysis for representative deployment scenarios to make the efficiency claims actionable for practitioners with different inference volumes.
6.3 The Block Size Is a Frozen Architectural Constant That Cannot Be Changed at Inference
The assumption or constraint. The block size is fixed during training and baked into the model's attention mask structure. The paper states this explicitly: "We fix the block size to 32 for all experiments" (Appendix A.1). The inference procedure respects this by decoding exactly 32 tokens per block. Table 4 demonstrates what happens when this constraint is violated: running inference with a different block size than was used during training causes severe performance degradation. GSM8K accuracy drops from 60.2 (block size 32, matching training) to 53.2 (block size 2), a 7-point loss. HumanEval shows a similar pattern, dropping from 38.4 to 37.8 when block size is reduced to 2, though with non-monotonic behavior (sizes 4 and 8 achieve 43.3, actually outperforming the training-matched size 32 for unclear reasons).
The consequence. This is a genuine architectural rigidity that limits deployment flexibility. In a standard AR model, the generation length is fully variable β the model can produce sequences of any length, and the attention pattern (causal masking) adapts seamlessly because it depends only on token position, not on any fixed block structure. In Fast-dLLM v2, the block size is a first-class architectural hyperparameter that must be chosen before training and fixed permanently. This has several practical implications:
-
Sequence length must be a multiple of 32. The inference procedure pads sequences with [MASK] tokens to ensure this (Section 3.3). For generation tasks where the natural stopping point does not align with block boundaries, the model generates padding tokens that are discarded. This wastes some fraction of the computational budget (up to 31 tokens per sequence, or ~3% for a 1000-token generation).
-
The block size cannot be tuned per-task at deployment. Table 3 and Table 4 together suggest that different tasks have different optimal decoding granularities: GSM8K prefers sub-block size 2 (62.8 accuracy) while HumanEval prefers sub-block size 8 (43.9). But these are sub-block sizes β the underlying block size of 32 is fixed, and the model's attention patterns are trained for this specific structure. If a deployment scenario would benefit from larger blocks (more intra-block parallelism, fewer autoregressive steps) or smaller blocks (finer-grained control, less commitment to potentially incorrect parallel predictions), the model cannot accommodate this without retraining.
-
The choice of block size involves irreversible tradeoffs made before training. A larger block size would provide more parallelism per AR step (potentially higher throughput) but would require the model to predict more tokens simultaneously with bidirectional context, which may become harder as block size grows (the model must resolve more long-range dependencies within a single diffusion process). A smaller block size would reduce the per-block generation cost but increase the number of autoregressive steps, reducing the throughput advantage. The paper provides no guidance on how to select block size a priori for a given deployment scenario or base model, and no experiment varying block size at training time to characterize this tradeoff.
What evidence exists in the paper. Table 4 (block size mismatch at inference) and the fixed block size specification in Appendix A.1 are the direct evidence. The sub-block size ablation in Table 3 and Figure 6 shows that decoding granularity can be tuned at inference time without retraining, which partially mitigates the rigidity (sub-block size provides a knob for speed-accuracy tradeoff within the fixed 32-token block structure), but the fundamental block size constraint remains.
Mitigation status. Partially mitigated by the sub-block decoding mechanism (Table 3), which allows some inference-time flexibility in decoding granularity without changing the attention structure. However, the paper does not discuss the fundamental block size tradeoff, does not experiment with training-time block size variation, and does not characterize how the choice of block size affects the quality-efficiency Pareto frontier. A practitioner adopting this method must commit to (or replicate the training recipe with a different block size, at additional cost) without data on whether this is optimal for their use case.
6.4 The Method Is Validated on a Single Model Family and Training Data Distribution
The assumption or constraint. All experiments in the paper use Qwen2.5-Instruct models (1.5B and 7B) fine-tuned on the LLaMA-Nemotron post-training dataset. The paper states that the LLaMA-Nemotron dataset contains "high-quality instruction-following examples covering a broad range of domains" (Appendix A.1), but does not argue that Qwen2.5-Instruct is uniquely suited to block diffusion adaptation, nor that the LLaMA-Nemotron dataset is uniquely suited as fine-tuning data. The implicit assumption is that the findings generalize: any pretrained AR-LLM fine-tuned on any instruction dataset with the block diffusion recipe will exhibit similar data efficiency and throughput gains.
The consequence. This assumption is untested and has several plausible failure modes. First, the compatibility between the pretrained attention patterns and the block-wise attention mask may vary across model families. Qwen2.5's specific training procedure, architecture, and data mixture may produce representations that are particularly amenable to the block-wise adaptation β or particularly resistant. Without experiments on other widely-used model families (LLaMA-3, Mistral, Gemma, DeepSeek), there is no evidence that the 500Γ data efficiency advantage over Dream generalizes. A practitioner using a different base model cannot assume that ~1B tokens of fine-tuning will suffice.
Second, the LLaMA-Nemotron dataset is a specific post-training dataset with particular properties (the paper describes it as "instruction-following examples"). The block diffusion fine-tuning may interact with dataset properties in ways that affect downstream performance. For example, if the dataset contains mostly short-form responses, the model may learn to rely on intra-block context for short generations but struggle with long-form generation where more blocks are needed. If the dataset has particular formatting conventions, the bidirectional intra-block attention may learn to exploit formatting patterns that don't generalize.
Third, the paper evaluates on a standard but specific benchmark suite (HumanEval, MBPP, GSM8K, MATH, IFEval, MMLU, GPQA). These benchmarks predominantly test reasoning, knowledge, and code generation in English. Whether the block diffusion approach works for other languages, for open-ended generation tasks (summarization, creative writing, dialogue), or for tasks requiring very long generations (where the number of autoregressive block steps becomes large and the throughput advantage may diminish) is entirely unexplored.
What evidence exists in the paper. All experimental results (Tables 1β4, Figures 1β6) use Qwen2.5-Instruct as the base model and LLaMA-Nemotron as the fine-tuning data. The paper does not include any experiments with alternative model families, alternative fine-tuning datasets, non-English tasks, or open-ended generation benchmarks. The Related Work section (Section 2.2) mentions concurrent works (SDAR, D2F, Set Block Decoding) that also adapt pretrained AR models to block diffusion, but does not compare their model families, datasets, or benchmark results to Fast-dLLM v2 in any systematic way. The claim that Fast-dLLM v2 is "distinguished by its data-efficient fine-tuning process, requiring only 1B tokens" (Section 2.2) is made without evidence that concurrent methods require more β their training budgets are simply not reported.
Mitigation status. Not addressed. The paper does not discuss the generalizability of its findings across model families, datasets, or task types. Section 5 (Conclusion) presents the results as general properties of the Fast-dLLM v2 framework without qualification. To establish generalizability, the paper would need at minimum: experiments with at least one additional base model family (e.g., LLaMA-3-8B), experiments with at least one additional fine-tuning dataset, and evaluation on tasks beyond the standard reasoning/knowledge/code suite (e.g., summarization, translation, long-form QA) to characterize how the throughput-accuracy tradeoff varies with generation length and task type.
6.5 There Is No Statistical Rigor in the Experimental Results
The assumption or constraint. Every quantitative result in the paper β all benchmark accuracies in Table 1, all throughput measurements in Figures 1, 4, 5, and 6, all ablation results in Tables 2β4 β is reported as a single number without any measure of variance, confidence interval, or statistical significance. The paper does not describe running multiple training runs with different random seeds, does not report standard deviations across evaluation splits, and does not perform any hypothesis tests comparing Fast-dLLM v2 to baselines. The training budgets are small (approximately 1.3B tokens for the 7B model, 3.15B for the 1.5B model), the number of training steps is modest (2,500 and 6,000 respectively), and the masking patterns are randomly sampled β all sources of potential variance in final model quality.
The consequence. A practitioner cannot assess whether the reported differences between Fast-dLLM v2 and baselines are reliable or might disappear under replication. Several of the claimed advantages are very small: Fast-dLLM v2 (7B) at 60.3 average vs. Qwen2.5-7B-Nemo-FT at 59.6 is a 0.7-point difference across a 7-benchmark average. On individual benchmarks, the gaps are often within 1β2 percentage points (e.g., GSM8K: 83.7 vs. 84.1; MMLU: 66.6 vs. 68.6). Without variance estimates, it is impossible to know whether these differences are meaningful or within the noise of training randomness and evaluation sampling. The paper's central "lossless adaptation" claim β that the block diffusion fine-tuning preserves AR model quality β rests on the 0.7-point average advantage, but if the standard deviation of the average across training runs is, say, 1β2 points, then the claim is not statistically supported.
The small test sets for some benchmarks compound this problem. HumanEval has 164 problems; a 1β2 percentage point difference corresponds to 1β3 additional correct solutions, which could easily arise from sampling variance in the masking patterns during a single training run. GPQA and IFEval similarly have modest test set sizes (not specified in the paper, but typically on the order of a few hundred examples). Without multiple evaluation runs or confidence intervals, the point estimates in Table 1 cannot be interpreted as reliable measurements of model capability differences.
The throughput measurements are similarly unaccompanied by variance estimates. GPU throughput depends on many factors (other processes on the machine, thermal throttling, measurement timing protocol), and reporting a single tokens/sec number without specifying whether it is the mean of multiple runs, the median, the best of N, or a single measurement makes the numbers non-reproducible and their precision unknowable.
What evidence exists in the paper. None. The paper does not mention variance, standard deviation, confidence intervals, statistical tests, or multiple training runs anywhere in the main text or appendices. The evaluation protocol (Appendix A.4) specifies greedy decoding and zero-shot prompting but says nothing about the number of evaluation runs or any statistical methodology. This absence is systematic: every number in every table and figure is a single point estimate from an unspecified number of measurements.
Mitigation status. Not addressed. The paper shows no awareness of this as a limitation. Minimum standards for addressing it would include: reporting standard deviations across at least 3 training runs with different random seeds for the main results in Table 1, reporting confidence intervals for the throughput measurements in Figures 4 and 5, and performing a statistical test (e.g., paired bootstrap test) for the claim that Fast-dLLM v2 matches or exceeds the AR baselines. Given the small training budgets, multiple runs would also provide evidence about the stability of the fine-tuning process β whether the ~1B token adaptation reliably converges to similar performance or is sensitive to data ordering and mask sampling.
6.6 Hard Problems at the Limits of the Base Model's Capability Cannot Be Solved by Block Diffusion
The assumption or constraint. The block diffusion approach amplifies the base model's existing capabilities β it does not create new ones. The model can only predict tokens that it has learned to predict during pretraining and fine-tuning. If a problem requires reasoning steps or knowledge that the base model cannot produce even with perfect autoregressive decoding, then no amount of intra-block parallelism or iterative refinement will produce a correct answer. This is an inherent property of the adaptation approach: the fine-tuning process teaches the model a new decoding strategy (predicting masked tokens with bidirectional context) but does not substantially expand its knowledge or reasoning capability beyond what the original AR model possessed.
This limitation is analogous to the "hard problems" finding in the compute-optimal test-time scaling literature, where the hardest difficulty bin shows near-zero improvement regardless of inference compute budget. Fast-dLLM v2 operates in a similar regime: the block diffusion mechanism can make the model faster at producing answers it already could produce, but cannot make it produce correct answers to problems it fundamentally cannot solve.
The consequence. The throughput gains from block diffusion are only valuable for problems that the base model can already solve correctly (at least some of the time). For problems entirely outside the model's capability range, the parallel decoding and caching mechanisms still operate β they still generate tokens faster β but the output is no more likely to be correct than the AR baseline's output. The paper does not characterize what fraction of real-world queries fall into this "unsolvable" category for the 7B model, nor does it analyze whether the failure modes of block diffusion (incorrect parallel predictions that get locked in by the confidence threshold) differ from the failure modes of AR decoding in ways that could make block diffusion worse on hard problems.
This connects to a broader concern about the interaction between the confidence threshold and problem difficulty. On hard problems, the model's token-level confidence may be systematically lower (because it is uncertain about the correct answer), causing the parallel decoding with Ο = 0.9 to leave many tokens masked for more refinement steps β reducing the throughput advantage precisely when the model needs more computation. Conversely, on easy problems where confidence is high, tokens are finalized quickly, and the speedup is largest. This creates a situation where the method is fastest on easy problems (which the AR baseline also handles quickly) and slowest on hard problems (where speed matters most for user experience). The paper does not analyze this difficulty-speed interaction.
What evidence exists in the paper. The paper does not directly measure this limitation. The benchmark results in Table 1 show that Fast-dLLM v2 underperforms the AR Nemo-FT baseline on MATH (61.6 vs. 72.0, a 10.4-point gap) and GPQA (31.9 vs. 34.2) β two of the harder benchmarks in the suite. This could indicate that block diffusion is less effective on difficult reasoning tasks, but the paper does not investigate this pattern or break down results by question difficulty within benchmarks. The parallel decoding threshold sweep (Figure 4) shows that accuracy degrades as the threshold is lowered (Ο = 0.5 achieves only ~62% on GSM8K vs. ~83% at Ο = 1.0), but this is an aggregate effect across all GSM8K questions without difficulty stratification.
The paper does not provide any analysis of performance conditioned on problem difficulty β no difficulty bins, no comparison of speedup at different accuracy levels, and no characterization of how the token-level confidence distribution varies with problem hardness. Without this, it is impossible to know whether the 2.5Γ speedup reported on GSM8K applies uniformly across easy and hard questions, or whether it is concentrated on easy questions where the model is already confident.
Mitigation status. Not addressed. The paper does not discuss the relationship between problem difficulty and block diffusion effectiveness, does not analyze failure modes on hard problems, and does not stratify results by difficulty. The closest the paper comes to acknowledging capability bounds is in the context of model scale: the 7B model outperforms the 1.5B model across all benchmarks (as expected from scaling), but the relative benefit of block diffusion over AR baselines at each scale is similar (~0.7 points average improvement for both), suggesting the adaptation itself does not become more or less effective with scale. To properly address this limitation, the paper would need difficulty-stratified results (similar to the difficulty bin analysis in test-time compute scaling work), an analysis of how the confidence threshold's speed-accuracy tradeoff varies with problem difficulty, and a characterization of whether block diffusion introduces systematic errors on hard problems that differ from AR decoding errors.
7. Implications and Future Directions
How This Work Changes the Landscape
This paper introduces a compatibility-first framing for adapting autoregressive models to diffusion-based decoding, shifting the field's implicit question from "how do we train a diffusion model efficiently?" to "how close can we stay to the AR attention structure while still getting parallelism?" This is a genuine reframing rather than an incremental improvement. Prior work on scaling diffusion language models (Dream, LLaDA) treated the move from causal to bidirectional attention as an unavoidable cost of parallel generation β you either train a diffusion model from scratch or accept massive retraining budgets (Dream's ~580B tokens). Fast-dLLM v2 demonstrates that this cost is not fixed; it varies dramatically with the architectural distance between the source (AR) and target (diffusion) attention patterns. By restricting diffusion to within-block bidirectional attention while preserving block-level causality, the attention mask remains close enough to the pretrained structure that fine-tuning with ~1B tokens suffices β a 500Γ reduction.
The practical consequence is that diffusion-based decoding becomes accessible as a post-training step rather than requiring a separate training pipeline. This lowers the barrier to entry substantially. A team with a pretrained 7B model and modest computational resources (~64 A100s for 12 hours) can now convert it to a block diffusion decoder and evaluate whether the throughput gains justify the effort. Dream's 580B-token requirement made such experimentation prohibitively expensive for most research groups; Fast-dLLM v2's ~1B-token requirement makes it a feasible weekend experiment.
The paper also clarifies a previously muddled conversation about KV-cache compatibility in diffusion models. Prior works like Fast-dLLM (Wu et al., 2025) introduced approximate caching mechanisms (DualCache) that provided partial solutions but, as the paper notes, "do not fundamentally resolve incompatibility of dLLMs with KV-cache, since such approximate caches are not equivalent to the original computation." Fast-dLLM v2 shows that exact KV caching is achievable in a diffusion model if and only if the attention structure guarantees that past blocks are causally isolated from future refinement. The block-wise autoregressive loop provides exactly this guarantee: once block is finalized, its tokens never change, and future blocks attend to them causally, making the cached KV entries mathematically identical to what a recomputation would produce. This is a crisp architectural insight: causal isolation is the necessary and sufficient condition for exact KV caching in iterative refinement models. This principle likely generalizes beyond block diffusion to any multi-step generation process that interleaves parallel and sequential computation.
The work also reconciles a tension between two lines of evidence. On one side, full-attention dLLMs like Dream showed that diffusion could achieve competitive accuracy (~81% on GSM8K, 57.6 average score) but at the cost of slower inference than AR models (48.2 vs. 39.5 tokens/sec at batch size 1, Figure 1b) β defeating the purpose of parallel generation. On the other side, block diffusion models like BD3-LMs showed conceptual promise for KV caching but were never validated at scale. The pessimistic interpretation was that diffusion's parallelism and AR's caching efficiency were fundamentally incompatible. Fast-dLLM v2 resolves this: they are compatible, but only within a specific architectural regime β block-level autoregressive structure with intra-block diffusion β and the training cost to reach this regime is low enough to be practical.
One research direction that becomes less attractive is training large diffusion language models from scratch. Table 1 shows LLaDA-8B (trained from scratch on MDM loss) achieving only 43.3 average score β far below Fast-dLLM v2's 60.3 and even below the 1.5B Fast-dLLM v2's 45.0. While LLaDA-1.5 with preference optimization partially closes this gap, the incomplete benchmark reporting makes the comparison tentative. Unless there is a compelling reason to avoid any AR pretraining whatsoever (e.g., specific inductive biases that only emerge from diffusion training), the evidence strongly favors adapting pretrained AR models rather than building diffusion models from the ground up.
What becomes more attractive is the systematic study of attention mask design as a first-class architectural parameter. The paper's core finding β that a block-diagonal + block-causal composite mask enables data-efficient adaptation β is one point in a larger design space. Future work can explore non-uniform block sizes (larger blocks for semantically coherent segments, smaller blocks for token-level precision), hierarchical block structures (blocks within blocks, enabling multi-scale diffusion), or learned block boundaries that adapt to content rather than being fixed at regular intervals. The attention mask is no longer a binary choice between causal and bidirectional; it becomes a tunable structure whose shape determines the training-efficiency-vs-parallelism tradeoff.
Follow-Up Research This Work Enables
Characterizing the per-task speed-accuracy Pareto frontier for confidence threshold and sub-block size. The paper selects Ο = 0.9 and sub-block size 8 based on GSM8K alone (Figure 4, Table 3). Table 3 already shows that the optimal sub-block size varies by task (GSM8K peaks at size 2 with 62.8, HumanEval at size 8 with 43.9). A natural follow-up would sweep both Ο (from 0.5 to 1.0) and sub-block size (2, 4, 8, 16, 32) jointly across the full benchmark suite, producing per-task Pareto curves. The key question is whether a single (Ο, sub-block size) pair suffices across all tasks with acceptable degradation, or whether task-adaptive decoding is necessary. The paper's throughput measurement inconsistencies (Figure 4's 39.1 tokens/sec at Ο = 1.0 vs. Figure 1b's 113.2 tokens/sec at batch size 1) would also need resolution β a controlled experiment measuring throughput under identical hardware, prompt distribution, and measurement protocol for all configurations.
Training-time block size sweeps to characterize the attention-distance-versus-training-cost frontier. The paper fixes block size for all experiments without systematic justification. Since the block size determines both the degree of intra-block parallelism (larger blocks = more tokens decoded in parallel) and the architectural distance from AR attention (larger blocks = more bidirectional context per block), it is the central hyperparameter governing the training-efficiency-vs-speedup tradeoff. A systematic sweep training Fast-dLLM v2 variants at on the same base model and dataset (controlling for total training tokens) would answer: (1) How does the required fine-tuning budget scale with block size? (2) Is there a "compute-optimal" block size that maximizes downstream accuracy for a fixed training budget? (3) Does the optimal block size depend on model scale? The finding from Table 4 that inference-time block size mismatch causes catastrophic degradation means this sweep must happen at training time, making it expensive but essential for understanding the method's scaling properties.
Replication on alternative model families to test generalizability. All experiments use Qwen2.5-Instruct. The paper's core claim β that block-wise attention compatibility enables data-efficient adaptation β implicitly assumes that Qwen2.5's pretrained representations are not uniquely amenable to this adaptation. A direct test would replicate the 7B training recipe on LLaMA-3-8B-Instruct and Mistral-7B-Instruct, using the same LLaMA-Nemotron dataset and the same ~1B token budget, then evaluate on the same benchmark suite. If both alternative models achieve comparable accuracy preservation and throughput gains, the method is robust. If one model family degrades significantly, it would reveal that the compatibility claim is contingent on specific architectural properties of Qwen2.5 β perhaps its training data distribution, its attention head structure, or its positional encoding scheme. This is a stress test the paper needs but does not provide.
Difficulty-stratified analysis of block diffusion performance. The prior analysis in this paper's Section 6.6 noted that Fast-dLLM v2 underperforms the AR baseline on MATH (61.6 vs. 72.0) and GPQA (31.9 vs. 34.2) β two of the harder benchmarks β while outperforming on code tasks. This pattern, combined with the confidence threshold's speed-accuracy tradeoff (Figure 4 shows accuracy degrading from ~83% to ~62% as Ο drops from 1.0 to 0.5 on GSM8K), suggests that block diffusion may be systematically worse on difficult problems. A follow-up study would stratify GSM8K and MATH questions by the base AR model's pass@1 rate (similar to the difficulty bin methodology in compute-optimal test-time scaling work) and measure Fast-dLLM v2's accuracy and throughput within each bin, both with and without parallel decoding. The key diagnostic: does the throughput advantage of block diffusion concentrate on easy problems (where the model is confident and parallel decoding finalizes tokens quickly) while providing no benefit β or even hurting β on hard problems? If so, the method's practical value depends on the difficulty distribution of the deployment's query stream.
FLOPs-normalized comparison between block diffusion and speculative decoding. The paper frames diffusion as an alternative to AR decoding, but the dominant practical approach for reducing AR latency is speculative decoding (draft-then-verify using a smaller model) or Medusa-style multi-token prediction heads. These methods also achieve ~2Γ speedups on standard benchmarks without any fine-tuning of the target model. A direct comparison measuring total FLOPs per generated token (not just wall-clock throughput) would reveal whether block diffusion's speedup represents genuine computational efficiency or just better hardware utilization of a more FLOPs-intensive process. The experiment would compare: (a) Qwen2.5-7B with speculative decoding (using Qwen2.5-0.5B as drafter), (b) Qwen2.5-7B with Medusa heads, and (c) Fast-dLLM v2 with Ο = 0.9, all generating the same GSM8K test set. Metrics: tokens/sec, total GPU FLOPs, and accuracy. If block diffusion uses more total FLOPs per correct token than speculative decoding, its throughput advantage reflects hardware utilization rather than algorithmic efficiency β a distinction with significant implications for cost-sensitive deployments where FLOPs, not wall-clock time, determine cloud billing.
Training a difficulty-aware adaptive confidence threshold. The paper's confidence threshold Ο is fixed at 0.9 for all tokens in all blocks across all problems. But token-level confidence likely varies systematically with problem difficulty and position within a block (early tokens in a block have less bidirectional context than later tokens). A learned policy that dynamically adjusts Ο based on features of the current decoding state β the model's entropy at each position, the number of refinement steps already taken, the average confidence in the block so far β could finalize high-certainty tokens aggressively while being conservative on uncertain ones, potentially extracting more speedup without the accuracy degradation that uniform Ο incurs. The training signal would be the downstream task accuracy, and the policy could be a lightweight classifier trained on the base model's confidence patterns during evaluation. This would directly address the concern that fixed Ο leaves throughput on the table for easy problems while degrading hard problems.
Practical Applications and Downstream Use Cases
Latency-sensitive interactive assistants with high query volumes. For a deployment like a customer support chatbot or coding assistant serving thousands of concurrent users, the 2.12Γ throughput improvement at batch size 4 on A100 (217.5 vs. 102.5 tokens/sec, Figure 1b) directly reduces the number of GPUs needed to serve a given query load. If the baseline AR deployment requires 100 A100s to maintain a target latency SLA, switching to Fast-dLLM v2 with Ο = 0.9 could reduce that to approximately 47 GPUs for the same throughput β a substantial cost saving. The key deployment consideration not addressed by the paper is whether the small accuracy degradation at Ο = 0.9 (approximately 0.5 percentage points on GSM8K, Figure 4) is acceptable for the specific application. For a coding assistant where a minor accuracy drop on HumanEval translates to a few additional incorrect completions that the user can quickly identify and discard, the tradeoff may be favorable. For a high-stakes medical or legal assistant, even small degradations may be unacceptable, and the deployment would need to use Ο = 1.0 (no parallel decoding), sacrificing the speedup. The paper's lack of per-benchmark accuracy-at-Ο=0.9 measurements means the deployer cannot make this decision from the paper alone.
Batch inference pipelines for evaluation and data generation. Organizations running large-scale batch inference β evaluating hundreds of thousands of test cases across benchmarks, or generating synthetic training data for self-improvement pipelines β care primarily about total throughput, not per-query latency. Figure 5 shows that at batch size 64 on H100, Fast-dLLM v2 achieves ~1.8Γ the throughput of AR decoding (~900 vs. ~500 tokens/sec). For a pipeline processing 1 billion tokens, this reduces processing time from approximately 23 days to approximately 13 days on a single H100 β or equivalently, processes the same workload with 44% fewer GPU-hours. This is a direct cost reduction for any organization running large-scale LLM evaluation or data generation. The training cost of ~12 GPU-hours (64 A100s Γ 12 hours / 64 β 12 GPU-days equivalent) is negligible compared to the inference savings for any pipeline processing more than a few hundred million tokens. The breakeven point is roughly 50 million inference tokens at batch size 64 on H100 (saving ~5.5 GPU-days of inference time against a ~0.5 GPU-day training cost), which is well within the scale of typical benchmark evaluation or data generation workloads.
On-device or edge deployment where memory bandwidth is the bottleneck. The paper's throughput measurements are on datacenter GPUs (A100, H100) with high memory bandwidth. However, the block diffusion architecture's ability to trade off between sequential steps and parallel computation per step is particularly relevant for edge devices (phones, laptops, embedded systems) where memory bandwidth is severely constrained. In a memory-bandwidth-bound regime, the cost of sequential forward passes (each requiring a full read of model weights from memory) dominates, and reducing the number of sequential steps β even at the expense of more total FLOPs β can substantially improve throughput. The sub-block cache (Figure 6b) specifically targets this regime, providing "substantial speedup in the compute-bound regime, such as when the batch size is 32." On an edge device with limited memory bandwidth, the effective "batch size" in terms of compute-vs-memory tradeoff may be much larger, making the cache even more valuable. However, the paper provides no edge-hardware measurements, so this application is speculative based on architectural properties rather than demonstrated results.
When to Prefer This Method
The paper frames Fast-dLLM v2 primarily against two alternatives: standard autoregressive decoding (Qwen2.5-7B-Instruct) and full-attention diffusion models (Dream, LLaDA). The tradeoffs it articulates are:
-
Prefer Fast-dLLM v2 over full-attention diffusion models (Dream, LLaDA) when you have a pretrained AR model you want to accelerate and cannot afford the ~580B-token retraining budget of full-attention diffusion. Fast-dLLM v2 achieves higher accuracy (60.3 vs. 57.6 average) and higher throughput (217.5 vs. 54.4 tokens/sec at batch size 4) while requiring 500Γ less fine-tuning data. This is not a nuanced tradeoff β the paper's evidence suggests block diffusion dominates full-attention diffusion on both accuracy and efficiency for the Qwen2.5-7B base model.
-
Prefer standard AR decoding over Fast-dLLM v2 when you need maximal accuracy on difficult reasoning tasks (MATH, GPQA) where the paper shows Fast-dLLM v2 underperforms the AR Nemo-FT baseline (61.6 vs. 72.0 on MATH, 31.9 vs. 34.2 on GPQA). The paper does not explicitly characterize this difficulty-accuracy tradeoff, so a deployer would need to evaluate on their specific task distribution. Also prefer AR when you cannot tolerate any accuracy degradation from parallel decoding (Ο < 1.0), noting that the paper's throughput advantage depends on Ο = 0.9 β the accuracy-preserving Ο = 1.0 configuration has unknown speedup relative to AR on most benchmarks.
-
Prefer Fast-dLLM v2 over speculative decoding or Medusa when you want multi-token parallelism without training an auxiliary drafter model or prediction heads. Fast-dLLM v2's parallelism comes from the same model's architecture (block-wise bidirectional attention), not from a separate component. However, the paper provides no direct comparison to speculative decoding, so this preference is based on architectural simplicity rather than demonstrated superiority. A deployer choosing between these methods would need to run their own comparison.
The paper does not articulate tradeoffs against other AR acceleration methods (speculative decoding, Medusa, lookahead decoding) or against simply using a larger AR model with greedy decoding β the FLOPs-matched comparison that would make the "when to prefer" decision quantitative is absent.