Per-Timestep Conditioned Diffusion Transformer
- The paper introduces per-timestep conditioning, where diffusion timesteps actively modulate Transformer computation, transforming static denoisers into adaptive models.
- It details several architectural patterns—such as dynamic width, spatial token routing, and spectral residual correction—that tailor model behavior at each diffusion timestep.
- Empirical evaluations report up to 51% FLOPs reduction and improved FID scores, demonstrating significant efficiency gains and enhanced output quality.
Searching arXiv for the cited papers and closely related work on timestep-conditioned diffusion transformers. Per-timestep conditioned diffusion transformers are diffusion-based Transformer denoisers in which the diffusion timestep does not merely index the reverse process, but actively modulates the model’s internal computation, conditioning pathway, or even the forward trajectory itself. In the standard formulation, a diffusion Transformer receives a noisy sample and a timestep representation, typically through Adaptive LayerNorm or FiLM-style modulation; recent work expands this design space so that timestep information can determine active attention heads and MLP channel groups, token selection, spectral residual correction, feature aggregation weights, condition-dependent trajectory shifts, or segment-wise noise levels in multimodal data (Zhao et al., 9 Apr 2025, Zhang et al., 2023, Kim et al., 2024).
1. Formal basis in diffusion Transformers
A conventional diffusion Transformer augments the noisy input with an explicit time embedding. One formulation computes
and injects it at each Transformer block through an Adaptive LayerNorm or FiLM-style modulation,
so that the denoiser can be written as
This is the baseline temporal-conditioning interface in DiT-like models (Chávez, 18 Jun 2026).
Within that baseline, the timestep embedding is already shared across the network. Dynamic Diffusion Transformer (DyDiT) makes this explicit by reusing the same timestep embedding employed by adaLN to drive additional per-block routing decisions. Each block inserts two lightweight routers that map to head-wise and channel-group-wise gating scores, thereby turning the timestep from a purely modulatory signal into a computation-allocation signal (Zhao et al., 2024).
A central conceptual distinction follows. In standard DiT, the timestep conditions feature normalization while the computational graph remains static. In per-timestep conditioned variants, the timestep may instead determine width, token participation, spectral emphasis, condition aggregation, or noise allocation. This suggests that “timestep conditioning” is not a single mechanism but a family of architectural interventions that share the property that denoising behavior changes systematically with diffusion time.
2. Architectural patterns of per-timestep conditioning
Several recurrent patterns have emerged for making diffusion Transformers timestep-adaptive.
| Mechanism | Timestep-conditioned quantity | Representative paper |
|---|---|---|
| Timestep-wise Dynamic Width (TDW) | Active MHSA heads and MLP channel groups | (Zhao et al., 9 Apr 2025) |
| Spatial-wise Dynamic Token (SDT) | Tokens routed through the expensive MLP | (Zhao et al., 9 Apr 2025) |
| Image-Timestep Adaptive Feature Aggregator (ITAFA) | Layer-wise weights over image-encoder features | (Hong et al., 26 Mar 2025) |
| Spectral residual correction | Low- and high-frequency residual amplification coefficients | (Tian, 17 Jun 2026) |
| Mixture of Noise Levels (MoNL) | Per-modality, per-time-segment diffusion indices | (Kim et al., 2024) |
| Shifted conditional trajectories | Per-timestep forward-process shift | (Zhang et al., 2023) |
In DyDiT and DyDiT++, TDW partitions each Transformer layer’s Multi-Head Self-Attention into heads of dimension and its MLP into parallel channel groups of size 0. Two router MLPs, 1 and 2, read the timestep embedding and output continuous scores
3
which are binarized into masks selecting active heads and channel groups. MHSA and MLP computation then sum only over the activated subsets. Because the masks depend only on 4, the active indices can be pre-computed for each timestep (Zhao et al., 9 Apr 2025).
DyDiT further couples TDW with Spatial-wise Dynamic Token. A lightweight token router 5 maps token features 6 to scores 7, which are binarized so that only selected tokens are processed by the expensive MLP, while the rest bypass it. The result is a joint timestep-and-space adaptive computation policy rather than a purely temporal one (Zhao et al., 9 Apr 2025).
ITA-MDT implements a different form of timestep conditioning. Its Image-Timestep Adaptive Feature Aggregator takes hidden-layer outputs from an image encoder,
8
projects the diffusion-timestep embedding and a three-component image-complexity descriptor into a shared space, combines them through a learned balance parameter 9, and produces softmax-normalized layer weights 0. The aggregated feature is
1
At each timestep, ITAFA recomputes 2 and hence 3, shifting emphasis among encoder layers as denoising progresses (Hong et al., 26 Mar 2025).
SpectralDiT localizes timestep conditioning even more narrowly. It leaves the standard flow-matching DiT structure intact and intervenes only in the MLP residual branch. The MLP residual is reshaped onto the patch-token grid, decomposed into low- and high-frequency components by a fixed depthwise low-pass filter, and corrected by timestep-conditioned coefficients 4 predicted by a zero-initialized two-layer MLP that consumes only the timestep embedding 5. The corrected residual is
6
This makes spectral emphasis an explicit function of diffusion time (Tian, 17 Jun 2026).
3. Conditioning the diffusion process rather than only the denoiser
A broader interpretation of per-timestep conditioning alters the diffusion trajectory itself. ShiftDDPMs introduce conditions into the forward process by allocating an exclusive diffusion trajectory for each condition 7. Given a shift predictor 8 and a schedule 9, the per-timestep shift is
0
and the conditional marginal becomes
1
The one-step kernel accordingly changes to
2
In the accompanying Transformer design, both the timestep embedding 3 and the shift 4 are injected into attention and MLP sublayers through FiLM (Zhang et al., 2023).
A different generalization appears in audiovisual latent diffusion with Mixture of Noise Levels. Instead of a single scalar timestep, the forward process uses a matrix
5
with one diffusion index for each modality 6 and time-segment 7. Each segment is noised at its own step, and the denoiser receives the entire noise-level pattern. Four sampling schemes are defined—Vanilla, Per-modality, Per-time, and Per-time-per-modality—and training draws from this mixture so that one model can cover joint generation, cross-modal generation, and multimodal interpolation tasks (Kim et al., 2024).
ITA-MDT occupies an intermediate position. Its forward diffusion remains standard,
8
but the conditioning feature 9 changes at every timestep because global garment features 0 and salient-region features 1 are recomputed through ITAFA and then combined with the time embedding before cross-attention. The timestep therefore controls not only denoising phase but also which conditional evidence is emphasized (Hong et al., 26 Mar 2025).
These formulations show that per-timestep conditioning need not be confined to a scalar embedding entering the reverse model. It can instead define the geometry of the forward trajectory, the granularity of the noise schedule, or the composition of the conditioning context itself.
4. Training objectives and parameter-efficient adaptation
The dominant training objective remains the standard noise-prediction mean-squared error. In DDPM form,
2
and several of the timestep-conditioned variants preserve this basic criterion even while changing what the network does with 3 (Chávez, 18 Jun 2026).
DyDiT++ augments the base diffusion loss with a FLOPs-aware penalty. For a batch of timesteps 4, it computes dynamic FLOPs relative to the static DiT budget and imposes
5
with overall loss
6
typically with 7. Optimization uses Gumbel-Sigmoid plus Straight-Through for discrete mask gradients, warm-up “complete” DiT supervision for first few steps, and a safeguard that at least one head and one group remain always active by magnitude-based ranking (Zhao et al., 9 Apr 2025).
ITA-MDT is trained with three objectives: denoising, mask reconstruction, and inpainting. The denoiser 8 follows MDT-v2, while a random binary mask in latent token space supports side-interpolator reconstruction and a garment-region-focused inpainting objective emphasizes the encoded garment mask. The reported total loss is
9
Within this setup, timestep-adaptive conditioning is embedded in ITAFA rather than in a specialized temporal auxiliary loss (Hong et al., 26 Mar 2025).
SpectralDiT is trained with a continuous-time linear flow-matching objective,
0
Its gate MLPs are zero-initialized, so early training matches the baseline DiT exactly. This design makes the timestep-conditioned spectral correction a residual refinement rather than a disruptive reparameterization (Tian, 17 Jun 2026).
DyDiT++ also introduces timestep-based dynamic LoRA (TD-LoRA) for parameter-efficient fine-tuning. Standard LoRA adapts a frozen weight 1 through low-rank matrices 2 and 3: 4 TD-LoRA replaces 5 by an MoE of experts 6 and computes timestep-dependent expert weights
7
so that the effective LoRA injection depends on the timestep. The fine-tuning schedule fully tunes the few router MLPs, LoRA-adapts AdaLN layers using standard LoRA, and applies TD-LoRA only on core Transformer weights 8 (Zhao et al., 9 Apr 2025).
5. Inference-time realization, complexity, and reported efficiency
Per-timestep conditioning becomes practically significant only when it produces real inference-time gains. DyDiT and DyDiT++ emphasize that head and channel masks depend only on 9, not on the sample, so the active indices can be pre-computed for all 0. At inference, the system simply looks up the masks rather than performing per-sample graph surgery; this is identified as critical for real-world batched speedups (Zhao et al., 9 Apr 2025, Zhao et al., 2024).
In DyDiT++, static DiT-XL per-step FLOPs for MHSA+MLP are reported as approximately 1. With 2, DyDiT-XL uses approximately 3, corresponding to an approximately 4 reduction and a 5 end-to-end speedup on V100. With 6, the budget is approximately 7, corresponding to a 8 FLOPs cut, a 9 wall-clock speedup, and FID 0 versus 1 baseline. The per-timestep savings vary, with early steps using as few as 2–3 of heads and channels, and SDT often skipping 4–5 of tokens on easy regions (Zhao et al., 9 Apr 2025).
The earlier DyDiT report gives closely aligned ImageNet 6 results for 7 DDPM steps and guidance 8: DiT-XL at 9, 0, FID 1; DyDiT-XL 2 at 3, 4, FID 5; and DyDiT-XL 6 at 7, 8, FID 9. On 0 ImageNet, DyDiT-XL 1 gives FID 2 versus 3 at 4 versus 5 (Zhao et al., 2024).
SpectralDiT illustrates a different efficiency regime. It adds only the gate MLPs, for a parameter overhead of 6, or approximately 7 relative to the DiT-S/2 base model, and approximately 8 more MACs in theory. The intent is not compute pruning but more effective timestep-conditioned residual correction with minimal overhead (Tian, 17 Jun 2026).
ITA-MDT frames efficiency differently again. It uses a lightweight, scalable transformer-based denoising diffusion model with a mask latent modeling scheme, and the Salient Region Extractor provides high-resolution local information as an additional condition while avoiding unnecessarily processing the entire garment image. Quantitatively, ITAFA cuts FID by approximately 9 points while keeping model size approximately 00 parameters and inference time approximately 01 per image (Hong et al., 26 Mar 2025).
6. Empirical scope, applications, and points of contention
The empirical scope of per-timestep conditioned diffusion Transformers now spans class-to-image generation, flow-matching, video generation, text-to-image generation, image-based virtual try-on, and audiovisual generation. DyDiT++ reports effectiveness on diverse visual generation models including DiT, SiT, Latte, and FLUX, and states that video generation and text-to-image generation show similar trade-offs and 02–03 speedups with minimal quality loss (Zhao et al., 9 Apr 2025).
On ImageNet 04, DyDiT++ reports a FLOPs–FID trade-off in which DyDiT-XL 05 at 06 reaches FID 07 and DyDiT-XL 08 at 09 reaches FID 10, compared with static DiT-XL at 11 and FID 12. For flow matching, DySiT-XL 13 is reported at 14 and FID 15 versus 16 static. In parameter-efficient fine-tuning, DyDiT-XL_PEFT uses only approximately 17 tunable parameters, approximately 18 total, yet reaches FID 19 versus 20 full fine-tune, while standard LoRA yields FID 21 (Zhao et al., 9 Apr 2025).
ITA-MDT situates timestep conditioning in a task-specific setting. On VITON-HD, the MDT-IVTON baseline without ITAFA reports FID(pair/unp.) approximately 22; adding ITAFA yields 23; adding ITAFA and SRE yields 24. The paper also reports that ITAFA reduces LPIPS from 25 to 26, improves SSIM marginally, and that Figure 1(a) shows early timesteps weighting low-level layers while late timesteps weight high-level layers (Hong et al., 26 Mar 2025).
In audiovisual generation, Mixture of Noise Levels enables a single transformer-based latent diffusion model to cover multiple conditional tasks. On the Monologues dataset, trained for 27 steps and sampled with 28 DDIM steps, MoNL reports joint generation FAD 29 versus task-specific 30 and FVD 31 versus 32; Audio33Video FVD 34 versus vanilla 35; Video36Audio FAD 37 versus 38; AV-inpaint FAD 39, FVD 40; and AV-continue FAD 41, FVD 42. Averaged across five tasks, MoNL reports FAD 43, FVD 44 versus conditional 45, per-modality 46, and vanilla 47 (Kim et al., 2024).
A major point of contention concerns the necessity of explicit timestep embeddings. “On the Redundancy of Timestep Embeddings in Diffusion Models” argues that, under stated assumptions, the global minimizer of the standard diffusion loss can be achieved without explicit timestep conditioning, because the network can implicitly infer noise scales from the corrupted input. Empirically, on CelebA 48, removing 49 hurts DiT slightly—FID 50 for DiT versus 51 for DiT*—but improves U-Net FID from 52 to 53. On CIFAR-10 54, class-conditional DiT* reports FID 55 versus 56 for DiT, with higher precision and recall, and U-Net* reports FID 57 versus 58 (Chávez, 18 Jun 2026).
That result does not invalidate per-timestep conditioned diffusion Transformers, but it narrows the claim that explicit temporal embeddings are universally required. A plausible implication is that there are two separable questions: whether the denoiser must be told the timestep explicitly, and whether a model benefits from timestep-dependent computation or conditioning policies. DyDiT, ITAFA, SpectralDiT, MoNL, and ShiftDDPMs address the second question in distinct ways, while the time-agnostic results primarily challenge the first (Chávez, 18 Jun 2026).