mlsanityData to 7 Oct 2026

SAFE-DiT: Semantics-Aware Fast-path Execution for High-Resolution Diffusion Transformers

arXiv 2606.29360 ↗First posted 28 Jun 2026cs.CVcited by 1

On the map: Diffusion models → Diffusion foundations and sampling

Abstract (arXiv)

High-resolution Diffusion Transformer (DiT) inference contains substantial spatial redundancy, but many spatially adaptive implementations encode regional computation as attention masks, which can inadvertently move scaled dot-product attention (SDPA) away from FlashAttention fast paths. We identify this avoidable systems bottleneck as Mask-Induced Dispatch Tax (MIDT) and show that it grows with latent sequence length. We introduce SAFE-DiT, a training-free Semantics-Aware Fast-path Execution framework that separates exact mask elision from approximation-based spatial scheduling. SAFE-DiT removes only provenance-certified image self-attention masks that induce a row-wise constant shift in attention logits, preserves semantics-bearing masks such as text-padding masks, and realizes spatial adaptation through prompt-conditioned token partitioning, selective state updates with global context, and periodic context refresh. We call this acceleration-only configuration SAFE-Core and report sensitivity-weighted classifier-free guidance separately as SAFE-DiT+SW. On the evaluated PyTorch SDPA stack, redundant masks make long-sequence attention 4.1× to 5.8× slower than the mask-free path. On Lumina-Next, SAFE-DiT achieves 2.69× end-to-end acceleration at 10242 resolution and 5.09× at 25602, reduces peak memory at 25602 from 94.1 to 27.9 GB, and enables 30722 generation when dense inference runs out of memory. Paired metrics, component ablations, and a blinded human study support visual non-inferiority of SAFE-Core to the dense fast-path baseline, while SAFE-DiT+SW provides a separate prompt-alignment operating point without reintroducing spatial self-attention masks. Code is available at https://github.com/xuanhuayin/SAFE-DiT.

Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Placed in Diffusion foundations and sampling, one of 14 directions in Diffusion models. Builds on 53 of its 61 references that are in the map. First posted on arXiv 28 Jun 2026.

Builds on

53 of 61 references.

Closest ancestors · 28

Same direction, newest first

  1. 2026Denoising as Path Planning: Training-Free Acceleration of Diffusion Models with DPCache26 Feb
  2. 2026SDiT: Semantic Region-Adaptive for Diffusion Transformers18 Jan
  3. 2025Compute Only 16 Tokens in One Timestep: Accelerating Diffusion Transformers with Cluster-Driven Feature Caching12 Sep
  4. 2025FastCache: Fast Caching for Diffusion Transformer Through Learnable Linear Approximation26 May
  5. 2025From Reusing to Forecasting: Accelerating Diffusion Models with TaylorSeers10 Mar
  6. 2025Region-Adaptive Sampling for Diffusion Transformers14 Feb
  7. 2024Accelerating Diffusion Transformers with Token-wise Feature Caching5 Oct
  8. 2024FORA: Fast-Forward Caching in Diffusion Transformer Acceleration1 Jul
  9. 2024Delta-DiT: A Training-Free Acceleration Method Tailored for Diffusion Transformers3 Jun
  10. 2024Rethinking the Spatial Inconsistency in Classifier-Free Diffusion Guidance8 Apr
  11. 2024Fast High-Resolution Image Synthesis with Latent Adversarial Diffusion Distillation18 Mar
  12. 2024PixArt-Sigma: Weak-to-Strong Training of Diffusion Transformer for 4K Text-to-Image Generation7 Mar
16 more
  1. 2024Scaling Rectified Flow Transformers for High-Resolution Image Synthesis5 Mar
  2. 2023Cache Me if You Can: Accelerating Diffusion Models through Block Caching6 Dec
  3. 2023DeepCache: Accelerating Diffusion Models for Free1 Dec
  4. 2023Latent Consistency Models: Synthesizing High-Resolution Images with Few-Step Inference6 Oct
  5. 2023PixArt-α: Fast Training of Diffusion Transformer for Photorealistic Text-to-Image Synthesis30 Sep
  6. 2023SDXL: Improving Latent Diffusion Models for High-Resolution Image Synthesis4 Jul
  7. 2023Consistency Models2 Mar
  8. 2022Scalable Diffusion Models with Transformers19 Dec
  9. 2022DPM-Solver++: Fast Solver for Guided Sampling of Diffusion Probabilistic Models2 Nov
  10. 2022All are Worth Words: A ViT Backbone for Diffusion Models25 Sep
  11. 2022Classifier-Free Diffusion Guidance26 Jul
  12. 2022DPM-Solver: A Fast ODE Solver for Diffusion Probabilistic Model Sampling in Around 10 Steps2 Jun
  13. 2021High-Resolution Image Synthesis with Latent Diffusion Models20 Dec
  14. 2021Diffusion Models Beat GANs on Image Synthesis11 May
  15. 2020Denoising Diffusion Implicit Models6 Oct
  16. 2020Denoising Diffusion Probabilistic Models19 Jun

Other references · 25

  1. 2017Attention Is All You Needcited by 59,932
  2. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
  3. 2014Microsoft COCO: Common Objects in Contextcited by 17,498
  4. 2018The Unreasonable Effectiveness of Deep Features as a Perceptual Metriccited by 11,029
  5. 2022Hierarchical Text-Conditional Image Generation with CLIP Latentscited by 6,539
  6. 2022Photorealistic Text-to-Image Diffusion Models with Deep Language Understandingcited by 6,322
  7. 2023Adding Conditional Control to Text-to-Image Diffusion Modelscited by 5,466
  8. 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,611
  9. 2021GLIDE: Towards Photorealistic Image Generation and Editing with Text-Guided Diffusion Modelscited by 3,595
  10. 2023FlashAttention-2: Faster Attention with Better Parallelism and Work Partitioningcited by 2,551
  11. 2022Prompt-to-Prompt Image Editing with Cross Attention Controlcited by 2,409
  12. 2021SDEdit: Guided Image Synthesis and Editing with Stochastic Differential Equationscited by 1,903
13 more
  1. 2022Scaling Autoregressive Models for Content-Rich Text-to-Image Generationcited by 1,287
  2. 2021Blended Diffusion for Text-driven Editing of Natural Imagescited by 924
  3. 2022eDiff-I: Text-to-Image Diffusion Models with an Ensemble of Expert Denoiserscited by 921
  4. 2023Human Preference Score v2: A Solid Benchmark for Evaluating Human Preferences of Text-to-Image Synthesiscited by 847
  5. 2023GLIGEN: Open-Set Grounded Text-to-Image Generationcited by 774
  6. 2023Attend-and-Excite: Attention-Based Semantic Guidance for Text-to-Image Diffusion Modelscited by 696
  7. 2023MultiDiffusion: Fusing Diffusion Paths for Controlled Image Generationcited by 529
  8. 2024FlashAttention-3: Fast and Accurate Attention with Asynchrony and Low-precisioncited by 507
  9. 2024Timestep Embedding Tells: It's Time to Cache for Video Diffusion Modelcited by 245
  10. 2022What the DAAM: Interpreting Stable Diffusion Using Cross Attentioncited by 233
  11. 2023SnapFusion: Text-to-Image Diffusion Model on Mobile Devices within Two Secondscited by 227
  12. 2024FlashMask: Efficient and Rich Mask Extension of FlashAttentioncited by 18
  13. 2026AccelAes: Accelerating Diffusion Transformers for Training-Free Aesthetic-Enhanced Image Generationcited by 4

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 1.

1 paper.

Similar, not cited

Close in wording, no citation either way

  1. 2024DiTFastAttn: Attention Compression for Diffusion Transformer Models
  2. 2026DiTango: Cost-Effective Parallel Diffusion Generation with Selective Attention State Reuse
  3. 2024Layer- and Timestep-Adaptive Differentiable Token Compression Ratios for Efficient Diffusion Transformers
  4. 2025Chipmunk: Training-Free Acceleration of Diffusion Transformers with Dynamic Column-Sparse Deltas
  5. 2026CoReDiT: Spatial Coherence-Guided Token Pruning and Reconstruction for Efficient Diffusion Transformers
  6. 2026One Model, Many Budgets: Elastic Latent Interfaces for Diffusion Transformers
  7. 2025CorGi: Contribution-Guided Block-Wise Interval Caching for Training-Free Acceleration of Diffusion Transformers
  8. 2026Edit Where You Mean: Region-Aware Adapter Injection for Mask-Free Local Image Editing