Explain VAEs, ELBO, KL, and Reparameterization

Quick Overview

Connect VAE components to the ELBO, reconstruction and KL terms, posterior approximation, reparameterized gradients, and collapse risk.

Explain VAEs, ELBO, KL, and Reparameterization

Company: Google

Role: Machine Learning Engineer

Category: Machine Learning

Difficulty: hard

Interview Round: Technical Screen

# Explain VAEs, ELBO, KL, and Reparameterization Explain a variational autoencoder, derive the evidence lower bound, and show why the objective contains a reconstruction term and a KL term. Explain the reparameterization trick and why it enables gradient-based training. ### Constraints & Assumptions - Use a latent-variable model p(x, z), encoder q(z | x), and decoder p(x | z). - State the prior over z rather than assuming it silently. - Distinguish maximizing the ELBO from exactly maximizing log evidence. ### Clarifying Questions to Ask - Which likelihood model defines reconstruction quality? - What posterior family does the encoder produce? - How is the KL term weighted in the stated variant? ```hint Add and subtract the posterior Relate log p(x) to the ELBO plus a nonnegative divergence between the approximate and true posterior. ``` ### What a Strong Answer Covers - Encoder, decoder, prior, and generative story. - ELBO derivation and the roles of reconstruction and regularization. - Closed-form or estimated KL behavior and posterior collapse risk. - Reparameterization as deterministic noise transformation and gradient path. ### Follow-up Questions 1. What changes in a beta-VAE? 2. Why can a powerful decoder ignore the latent variable?

Quick Answer: Connect VAE components to the ELBO, reconstruction and KL terms, posterior approximation, reparameterized gradients, and collapse risk.

|Home/Machine Learning/Google
Google logo
Google
Aug 14, 2026
hardMachine Learning EngineerTechnical ScreenMachine Learning
6
0

Explain VAEs, ELBO, KL, and Reparameterization

Explain a variational autoencoder, derive the evidence lower bound, and show why the objective contains a reconstruction term and a KL term. Explain the reparameterization trick and why it enables gradient-based training.

Constraints & Assumptions

  • Use a latent-variable model p(x, z), encoder q(z | x), and decoder p(x | z).
  • State the prior over z rather than assuming it silently.
  • Distinguish maximizing the ELBO from exactly maximizing log evidence.

Clarifying Questions to Ask Guidance

  • Which likelihood model defines reconstruction quality?
  • What posterior family does the encoder produce?
  • How is the KL term weighted in the stated variant?

What a Strong Answer Covers Guidance

  • Encoder, decoder, prior, and generative story.
  • ELBO derivation and the roles of reconstruction and regularization.
  • Closed-form or estimated KL behavior and posterior collapse risk.
  • Reparameterization as deterministic noise transformation and gradient path.

Follow-up Questions Guidance

  1. What changes in a beta-VAE?
  2. Why can a powerful decoder ignore the latent variable?
Loading comments...