← Back to Topics

VAE Loss Calculation

VAE Loss Formula

VAE Loss = Reconstruction Loss + KL Divergence

Part 1: (x - x̂)²
Part 2: ½ × (μ² + σ² - log(σ²) - 1)

Shortcut: When μ=0 and σ²=1 → KL = 0
View in LibraryTry in Sandbox