Skip to content

Latest commit

 

History

History
75 lines (48 loc) · 8.04 KB

README.md

File metadata and controls

75 lines (48 loc) · 8.04 KB

Density estimation using Diffusion models (Pytorch + Jax/Haiku/Optax)

I will be demonstrating critical concepts of the diffusion model using a toy 2D distribution first, followed by using the same concepts on the EMNIST datasets.

Completed the following

Unexplored ideas

Notebook Github Link Colab
Basic: Predicting Original Distribution Vanilla Implementation Colab (Large)
Predicting Error and Score Function Error / Score Prediction Colab (Large)
Classifier free Guidance and other improvements Advanced concepts Colab (Large)
EMINST De-noising and Conditional generation Colab EMNIST Colab (Large)Colab (Small)

Generating names using EMNIST

Conditional denoising using the trained UNet model

alt text alt text

alt text

alt text

alt text

Generation toy distributions using diffusion models

Parabola

alt text

Circles

alt text

Half Moon

alt text

Circles + half-moon

alt text

Circles + moon using Clipping

alt text

Generating class-conditioned distributions

alt text

Generating class-conditioned distributions (few shots only using 2k samples)

alt text