Diffusion Without Tears
baincapitalventures.notion.site
baincapitalventures.notion.site
Also I had a hard time figuring out which part of the equation differentiate the equations for different data points, is that what the meaning of "theta" is in all the equations? To guide the initial noise towards one type of image instead of another is that what theta is responsible for? Is the innovation in GenAI images to use text embeddings to create the theta?
I definitley feel some tears coming on.
Theta represents all the model params — all the weights in the neural network. The convention is to write theta for the “learned” score function and omit theta for the “true” score function.
Means: maybe edit that title a bit :)
I liked the post though.
But to the quant crowd I’m guessing that couching diffusion in the language of stochastic calculus is helpful.
so in the end diffusion can’t explain that much about how diffusion models work. disappointingly, so many modern ML things are like this.
I personally find the SDEs the most intuitive, and the deterministic ODE / consistency models / rectified flow stuff as ideas that are easier to understand after the SDEs. But not everyone agrees!
I just find it a frustrating fact about modern machine learning that in fact, the nice SDE interpretation is somehow not the “truth”, it’s just a tool for understanding.
0. https://www.countbayesie.com/blog/2023/4/21/linear-diffusion
Thanks for reading. The 2D simulation section might be more interesting on a first read — it makes the math less mysterious, I hope!
Thanks, guys, super helpful.
Don’t try to catch all the mathematical Pikachus in the paper, just choose the insights that resonate with you. Thankfully, there isn’t a pop quiz lurking at the end.
In honor of the Bay Area roots of HN, “believe it if you need it, if you don't, just pass it on”. I liked the paper even when skipping the SDE material.
I too found it really surprising that the reverse-time equation has a simple closed form. Like, surely breaking a glass is easier than unbreaking it? That’s part of what got me interested in this stuff in the first place!
If you haven’t seen it yet, highly recommend the blogs of Sander Dieleman & Yang Song (who co-invented the SDE interpretation).