Papers › A Closer Look at Time Steps is Worthy of Triple Speed-Up for Diffusion Model Training

A Closer Look at Time Steps is Worthy of Triple Speed-Up for Diffusion Model Training

27 May 2024CVPR 2025 1arXiv:2405.17403archive 2025-07-28

Kai Wang, Mingjia Shi, Yukun Zhou, Zekai Li, Zhihang Yuan, Yuzhang Shang, Xiaojiang Peng, Hanwang Zhang, Yang You

Training diffusion models is always a computation-intensive task. In this paper, we introduce a novel speed-up method for diffusion model training, called, which is based on a closer look at time steps. Our key findings are: i) Time steps can be empirically divided into acceleration, deceleration, and convergence areas based on the process increment. ii) These time steps are imbalanced, with many concentrated in the convergence area. iii) The concentrated steps provide limited benefits for diffusion training. To address this, we design an asymmetric sampling strategy that reduces the frequency of steps from the convergence area while increasing the sampling probability for steps from other areas. Additionally, we propose a weighting strategy to emphasize the importance of time steps with rapid-change process increments. As a plug-and-play and architecture-agnostic approach, SpeeD consistently achieves 3-times acceleration across various diffusion architectures, datasets, and tasks. Notably, due to its simple design, our approach significantly reduces the cost of diffusion model training with minimal overhead. Our research enables more researchers to train diffusion models at a lower cost.

PaperPDFConference PDFCodeCode Syntology ran

In Syntology Open this paper in Syntology's Atlas, the map of the papers in Syntology's graph and their citations.

For agents, Syntology's MCP tool lists every function and class Syntology harvested from this paper and whether it ran (how to connect): get_harvested_code_for_paper(arxiv_id="2405.17403")

Code

Syntology Ran 6 of 11 code samples harvested from 2 repositories linked to this paper; 5 have no recorded run. Of those that ran: 2 ran · honoured contract; 2 ran · our draft was wrong; 1 ran · fixture could not drive it; 1 ran with no contract checked.

By repository: official repository: 11 samples from 2 repositories, 6 ran. The run record, sample by sample. “Ran” means executed on a synthesized input, not that the code is correct or reproduces the paper.

1zeryu/speed officialmentioned in papermentioned on GitHubpytorchApache-2.0 report
nus-hpc-ai-lab/speed officialmentioned in papermentioned on GitHubpytorchApache-2.0 report

Repository list and official/mentioned flags are the archive's, frozen 2025-07-28. Reachability, where shown, is from one Syntology probe window (2026-09-16 to 2026-09-18); repositories not probed show nothing. GitHub stars are not tracked.

Code Syntology ran Syntology

11 samples harvested; 6 ran; 2 honoured the contract we drafted; 5 have no recorded run. Read from Syntology's graph 2026-09-24; that is when this build read the record, not when the samples ran.

2ran · honoured contract
2ran · our draft was wrong
1ran · fixture could not drive it
1ran
5unverified

Licence: 0 of the 11 samples are pointer only, meaning Syntology does not serve that copy's text. This page shows no code text for any sample; each one links to its file in the repository.

Harvested from 2 repositories linked to this paper, official or community; each sample names its own and says which. “Ran” means the sample executed on a synthesized input. It does not mean the output is correct, and nothing here reproduces the paper's results. “Honoured” and “violated” refer to a contract Syntology drafted from the code itself; “our draft was wrong” and “fixture could not drive it” are failures of Syntology's instrument, not of the code.

Each sample ends with its code_sha256, Syntology's identity for that exact code. An agent fetches the stored sample with Syntology's MCP tool get_code(code_sha256="…") (how to connect); click an identity to copy that call.

Repository labels, per sample. official repository: The archive marks this repository official for the paper. named in the paper: The archive records that the paper mentions this repository; it is not marked official. community (archive-listed): In the archive's code links for this paper, not marked official and not recorded as mentioned in the paper. found in paper text by Syntology: Syntology found this repository in the paper's own text; whether it is the authors' implementation is not asserted. community: Not in the archive's code links for this paper; a community repository Syntology harvested. Samples from a repository marked official are listed first. Licence labels name the repository's licence as recorded at harvest. “Pointer only” means Syntology does not serve that copy's text, for one of four reasons: no licence file was found; the licence was not identified; the licence is recorded as permissive but that copy's record is not marked cleared; or the licence is outside the permissive list Syntology serves text under (MIT, Apache-2.0, BSD and similar). Some licences outside that list permit redistribution, such as WTFPL, and GPL-3.0 under its conditions; they are simply not on the list. Hover a licence label for the reason. File links open the file on GitHub at the default branch, which may have changed since the harvest.

center_crop_arr 1zeryu/speed/speed/dataset/transform.py official repository ran · our draft was wrong Apache-2.0 (permissive) · 1712a07966b542ee · report
get_1d_sincos_pos_embed_from_grid 1zeryu/speed/speed/networks/dit/mdt.py official repository ran · fixture could not drive it fingerprinted Apache-2.0 (permissive) · e5947aba1d10885f · report
get_2d_sincos_pos_embed 1zeryu/speed/speed/networks/dit/mdt.py official repository ran · honoured contract Apache-2.0 (permissive) · c92c27c924b517e8 · report
get_image_transform 1zeryu/speed/speed/dataset/transform.py official repository ran Apache-2.0 (permissive) · debdeb32aa1aa3e0 · report
modulate 1zeryu/speed/speed/networks/dit/dit.py official repository ran · our draft was wrong fingerprinted Apache-2.0 (permissive) · 03310bba324ae4fb · report
t2i_modulate nus-hpc-ai-lab/speed/speed/networks/pixart/PixArt_blocks.py official repository ran · honoured contract fingerprinted Apache-2.0 (permissive) · a5e3d9618aac5205 · report
get_1d_sincos_pos_embed_from_grid 1zeryu/speed/speed/networks/pixart/PixArt.py official repository unverified Apache-2.0 (permissive) · 7700f3428f58afbe · report
get_2d_sincos_pos_embed 1zeryu/speed/speed/networks/pixart/PixArt.py official repository unverified Apache-2.0 (permissive) · 5a10b94b2fe962d2 · report
get_2d_sincos_pos_embed_from_grid nus-hpc-ai-lab/speed/speed/networks/pixart/PixArt.py official repository unverified Apache-2.0 (permissive) · 60599713933af278 · report
get_2d_sincos_pos_embed_from_grid 1zeryu/speed/speed/networks/dit/mdt.py official repository unverified Apache-2.0 (permissive) · 665d8a4e8f673a4c · report
image_dataset 1zeryu/speed/speed/dataset/image.py official repository unverified Apache-2.0 (permissive) · 8849fb1444c6af62 · report

Results from the paper archive 2025-07-28

No leaderboard rows for this paper in the archive.

Methods

DiffusionSPEED

Report a problem or propose a change · a person checks every report against the paper or source before anything changes; decisions are listed on /corrections