Noise Prediction Loss Computation For Diffusion Training

via “unet-based iterative noise prediction and denoising”

text-to-image model by undefined. 5,45,314 downloads.

Unique: Combines UNet architecture with cross-attention conditioning (injecting CLIP embeddings at 4 resolution scales) and sinusoidal timestep embeddings. Uses a fixed linear noise schedule (beta_start=0.0001, beta_end=0.02) with 1000 timesteps, enabling stable training and inference.

vs others: More parameter-efficient than transformer-based alternatives (e.g., DiT) while maintaining strong semantic conditioning; comparable to proprietary models' architectures but fully open and reproducible.