Grokking of Diffusion Models: Case Study on Modular Addition

Joon Hyeok Kim, Yong-Hyun Park, Mattis Dalsætra Østby +1

Despite their empirical success, how diffusion models generalize remains poorly understood from a mechanistic perspective. We demonstrate that diffusion models trained with flow-ma…