cs.CVSep 4, 2026

Importance-Aware Low-Rank Distillation of Diffusion Transformers

Authors: Denis Zavadski, Sebastian Heid, Damjan Kalšan, Stefan Roth, Carsten Rother

Organizations: Heidelberg University · Zuse School ELIZA · TU Darmstadt

Abstract

Diffusion Transformers (DiTs) have emerged as a dominant architecture for high-quality text-to-image generation, yet their scale poses challenges for efficient deployment. While truncated singular value decomposition (SVD) is a principled tool for parameter reduction, evidence from large language models (LLMs) suggests that naive low-rank approximation can cause catastrophic failure. In contrast, we find that truncated SVD in DiTs produces smooth degradation even under substantial global compression, with redundancy distributed across projection matrices throughout the whole network rather than concentrated in a few transformer blocks. Building on these insights, we introduce SVDtrunc, a two-step block-level compression scheme, first allocating ranks across blocks and compressing the least important ones via truncated SVD under a global parameter budget, and then fine-tuning all blocks with modular knowledge distillation and a rectified-flow objective. We apply SVDtrunc to FLUX.dev across compression levels ranging from 40-90% of the original parameter count. Across three benchmarks, GenEval, HPSv2, and DPG, we outperform all competing approaches. Notably, and in contrast to prior work, we retain near-full performance at 68% and remain competitive even at 57% of the original parameter budget. Furthermore, we show that SVDtrunc complements step distillation and achieves strong results even without fine-tuning, positioning it as a practical continuation of efficiency improvements beyond diffusion step reduction for large-scale generative models. Project page: https://vislearn.github.io/SVDtrunc/

Figures & tables

Appendix figures & tables10 assets

Supplementary material from the paper’s appendix.

Appendix

Explore similar work

CardsList
  1. Amber-Image: Efficient Compression of Large-Scale Diffusion Transformers

    Feb 19, 2026Chaojie Yang, Tian Li, Yue Zhang +1Diffusion Model CompressionT2I Diffusion Models

  2. GA-SVD: Generation-aware Singular Value Decomposition for LLM Compression

    Oct 11, 2026Shengkun Tang, Cong Zeng, Zongfang Liu +2LLM CompressionLow-Rank Compression