Skip to content

Fix DDPM covariance notation and noise-scale prose in unit 1 - #120

Open
CharlesXmf wants to merge 1 commit into
huggingface:mainfrom
CharlesXmf:cursor/fix-unit1-02-formula-typo-0c47
Open

CharlesXmf wants to merge 1 commit into
huggingface:mainfrom
CharlesXmf:cursor/fix-unit1-02-formula-typo-0c47

Conversation

@CharlesXmf

Copy link
Copy Markdown

Fixes #63

The closed-form $q(x_t \mid x_0)$ in unit1/02_diffusion_models_from_scratch.ipynb used a standard-deviation scaling inside $\mathcal{N}(\mu, \Sigma)$, writing $\sqrt{(1-\bar\alpha_t)}\mathbf{I}$ instead of the covariance $(1-\bar\alpha_t)\mathbf{I}$. Notebook 01 already had the covariance form.

In the same cells, the stepwise process is $q(x_t \mid x_{t-1}) = \mathcal{N}(x_t; \sqrt{1-\beta_t}x_{t-1}, \beta_t I)$, so the equivalent sample adds noise scaled by $\sqrt{\beta_t}$, not $\beta_t$. That prose is updated in both notebook 01 and 02.

Plot labels such as sqrt_one_minus_alpha_prod are left unchanged, since those refer to the implementation's std-dev scale.

Match q(x_t | x_0) in notebook 02 to notebook 01 by using covariance
(1 - bar-alpha_t) I instead of a square-root on the covariance, and
describe the forward step as adding noise scaled by sqrt(beta_t).

Co-authored-by: Charles <CharlesXmf@users.noreply.github.com>
@review-notebook-app

Copy link
Copy Markdown

Check out this pull request on  ReviewNB

See visual diffs & provide feedback on Jupyter Notebooks.


Powered by ReviewNB

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

unit1/02_diffusion_models_from_scratch.ipynb typo in formulas

2 participants