Training-free Diffusion Acceleration with Bottleneck Sampling

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Tian, Ye, Xia, Xin, Ren, Yuxi, Lin, Shanchuan, Wang, Xing, Xiao, Xuefeng, Tong, Yunhai, Yang, Ling, Cui, Bin
Format: Preprint
Veröffentlicht: 2025
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866916663808491520
author Tian, Ye
Xia, Xin
Ren, Yuxi
Lin, Shanchuan
Wang, Xing
Xiao, Xuefeng
Tong, Yunhai
Yang, Ling
Cui, Bin
author_facet Tian, Ye
Xia, Xin
Ren, Yuxi
Lin, Shanchuan
Wang, Xing
Xiao, Xuefeng
Tong, Yunhai
Yang, Ling
Cui, Bin
contents Diffusion models have demonstrated remarkable capabilities in visual content generation but remain challenging to deploy due to their high computational cost during inference. This computational burden primarily arises from the quadratic complexity of self-attention with respect to image or video resolution. While existing acceleration methods often compromise output quality or necessitate costly retraining, we observe that most diffusion models are pre-trained at lower resolutions, presenting an opportunity to exploit these low-resolution priors for more efficient inference without degrading performance. In this work, we introduce Bottleneck Sampling, a training-free framework that leverages low-resolution priors to reduce computational overhead while preserving output fidelity. Bottleneck Sampling follows a high-low-high denoising workflow: it performs high-resolution denoising in the initial and final stages while operating at lower resolutions in intermediate steps. To mitigate aliasing and blurring artifacts, we further refine the resolution transition points and adaptively shift the denoising timesteps at each stage. We evaluate Bottleneck Sampling on both image and video generation tasks, where extensive experiments demonstrate that it accelerates inference by up to 3$\times$ for image generation and 2.5$\times$ for video generation, all while maintaining output quality comparable to the standard full-resolution sampling process across multiple evaluation metrics.
format Preprint
id arxiv_https___arxiv_org_abs_2503_18940
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Training-free Diffusion Acceleration with Bottleneck Sampling
Tian, Ye
Xia, Xin
Ren, Yuxi
Lin, Shanchuan
Wang, Xing
Xiao, Xuefeng
Tong, Yunhai
Yang, Ling
Cui, Bin
Computer Vision and Pattern Recognition
Diffusion models have demonstrated remarkable capabilities in visual content generation but remain challenging to deploy due to their high computational cost during inference. This computational burden primarily arises from the quadratic complexity of self-attention with respect to image or video resolution. While existing acceleration methods often compromise output quality or necessitate costly retraining, we observe that most diffusion models are pre-trained at lower resolutions, presenting an opportunity to exploit these low-resolution priors for more efficient inference without degrading performance. In this work, we introduce Bottleneck Sampling, a training-free framework that leverages low-resolution priors to reduce computational overhead while preserving output fidelity. Bottleneck Sampling follows a high-low-high denoising workflow: it performs high-resolution denoising in the initial and final stages while operating at lower resolutions in intermediate steps. To mitigate aliasing and blurring artifacts, we further refine the resolution transition points and adaptively shift the denoising timesteps at each stage. We evaluate Bottleneck Sampling on both image and video generation tasks, where extensive experiments demonstrate that it accelerates inference by up to 3$\times$ for image generation and 2.5$\times$ for video generation, all while maintaining output quality comparable to the standard full-resolution sampling process across multiple evaluation metrics.
title Training-free Diffusion Acceleration with Bottleneck Sampling
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2503.18940