Conceptual
Login

Layer Normalization Gradient Through Mean and Variance in Backpropagation

For \( \hat{x} = (x-\mu)/\sigma \), the gradient is \( \frac{1}{\sigma}\big(g - \bar{g} - \hat{x}\,\overline{g\hat{x}}\big) \): the upstream gradient with its mean and its projection on \( \hat{x} \) removed.

This Concept is waiting for its first lesson!

Are you a teacher? Sign in to start contributing.

Sign In