bot66/MNISTDiffusion
Implement a MNIST(also minimal) version of denoising diffusion probabilistic model from scratch.The model only has 4.55MB.
Built with depthwise convolutions, residual shortcuts, and simplified timestep embeddings, the model achieves full DDPM functionality at minimal footprint through architectural simplification rather than quantization. Trains end-to-end on MNIST using PyTorch with configurable hyperparameters, producing both inference checkpoints (9.1MB with EMA) suitable for educational purposes or resource-constrained environments. References the canonical DDPM paper and implements the core denoising diffusion framework as a learning resource for understanding generative modeling fundamentals.
143 stars. No commits in the last 6 months.
Stars
143
Forks
28
Language
Python
License
MIT
Category
Last pushed
Dec 09, 2022
Commits (30d)
0
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/diffusion/bot66/MNISTDiffusion"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Higher-rated alternatives
quantgirluk/aleatory
📦 Python library for Stochastic Processes Simulation and Visualisation
blei-lab/treeffuser
Treeffuser is an easy-to-use package for probabilistic prediction and probabilistic regression...
yuanchenyang/smalldiffusion
Simple and readable code for training and sampling from diffusion models
TuftsBCB/RegDiffusion
Diffusion model for gene regulatory network inference.
lvyufeng/denoising-diffusion-mindspore
Implementation of Denoising Diffusion Probabilistic Model in MindSpore