7 papers
BAG: Budget-Aware Gating for Diffusion Caching
Tong Zhao, Mingkun Lei, Yucheng Han +1
Diffusion caching is a lightweight strategy that accelerates Diffusion Transformers (DiTs) by reusing intermediate features across denoising steps, but existing paradigms face a fu…
Budget-Constrained Step-Level Diffusion Caching
Mingkun Lei, Tong Zhao, Liangyu Yuan +1
Step-level caching accelerates diffusion models by exploiting temporal redundancy across denoising steps. Existing methods make per-step cache decisions using threshold-based heuri…
Improving Diffusion Generalization with Weak-to-Strong Segmented Guidance
Liangyu Yuan, Yufei Huang, Mingkun Lei +5
Diffusion models generate synthetic images through an iterative refinement process. However, the misalignment between the simulation-free objective and the iterative process often…
Few-Step Diffusion Sampling Through Instance-Aware Discretizations
Liangyu Yuan, Ruoyu Wang, Tong Zhao +4
Diffusion and flow matching models generate high-fidelity data by simulating paths defined by Ordinary or Stochastic Differential Equations (ODEs/SDEs), starting from a tractable p…
DyWeight: Dynamic Gradient Weighting for Few-Step Diffusion Sampling
Tong Zhao, Mingkun Lei, Liangyu Yuan +5
Diffusion Models (DMs) have achieved state-of-the-art generative performance across multiple modalities, yet their sampling process remains prohibitively slow due to the need for h…
CleanStyle: Plug-and-Play Style Conditioning Purification for Text-to-Image Stylization
Xiaoman Feng, Mingkun Lei, Yang Wang +2
Style transfer in diffusion models enables controllable visual generation by injecting the style of a reference image. However, recent encoder-based methods, while efficient and tu…