University of Maryland, College Park
*Equal contribution.
Scale Space Diffusion is a pixel-space diffusion model that integrates scale-space theory by combining Gaussian noise with linear degradations (downsampling). It introduces Flexi-UNet for resolution-aware denoising.
Scale Space Diffusion teaser
Scale Space Diffusion sampling animation
This model release accompanies the official implementation of Scale Space Diffusion.
Model Details
The released checkpoints include SSD models trained on CelebA and ImageNet:
CelebA-64: 2L and 4L Flexi-UNet checkpoints
CelebA-128: 3L and 5L Flexi-UNet checkpoints
CelebA-256: 3L and 6L Flexi-UNet checkpoints
ImageNet-64: 2L Flexi-UNet checkpoint
Citation
@article{mukhopadhyay2026scale,
title={Scale Space Diffusion},
author={Mukhopadhyay, Soumik and Udhayanan, Prateksha and Shrivastava, Abhinav},
journal={arXiv preprint arXiv:2603.08709},
year={2026}
}
