ControlVAE
ControlVAE
**ControlVAE** is a [variational autoencoder](https://paperswithcode.com/method/vae) (VAE) framework that combines the automatic control theory with the basic VAE to stabilize the KL-divergence of VAE models to a specified value. It leverages a non-linear PI controller, a variant of the proportional-integral-derivative (PID) control, to dynamically tune the weight of the KL-divergence term in the evidence lower bound (ELBO) using the output KL-divergence as feedback. This allows for control of the KL-divergence to a desired value (set point), which is effective in avoiding posterior collapse and learning disentangled representations.
ControlVAE: Tuning, Analytical Properties, and Performance Analysis
https://paperswithcode.com/paper/controlvae-tuning-analytical-properties-and
ControlVAE: Tuning, Analytical Properties, and Performance Analysis