prateksha-u/ssd

Scale Space Diffusion [CVPR 2026]

Soumik Mukhopadhyay^*,  Prateksha Udhayanan^*,  Abhinav Shrivastava 

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.

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}
}