English

Regularization can make diffusion models more efficient

Machine Learning 2025-09-26 v2 Statistics Theory Machine Learning Statistics Theory

Abstract

Diffusion models are one of the key architectures of generative AI. Their main drawback, however, is the computational costs. This study indicates that the concept of sparsity, well known especially in statistics, can provide a pathway to more efficient diffusion pipelines. Our mathematical guarantees prove that sparsity can reduce the input dimension's influence on the computational complexity to that of a much smaller intrinsic dimension of the data. Our empirical findings confirm that inducing sparsity can indeed lead to better samples at a lower cost.

Keywords

Cite

@article{arxiv.2502.09151,
  title  = {Regularization can make diffusion models more efficient},
  author = {Mahsa Taheri and Johannes Lederer},
  journal= {arXiv preprint arXiv:2502.09151},
  year   = {2025}
}
R2 v1 2026-06-28T21:42:52.089Z