Unbiased Implicit Variational Inference
About
We develop unbiased implicit variational inference (UIVI), a method that expands the applicability of variational inference by defining an expressive variational family. UIVI considers an implicit variational distribution obtained in a hierarchical manner using a simple reparameterizable distribution whose variational parameters are defined by arbitrarily flexible deep neural networks. Unlike previous works, UIVI directly optimizes the evidence lower bound (ELBO) rather than an approximation to the ELBO. We demonstrate UIVI on several models, including Bayesian multinomial logistic regression and variational autoencoders, and show that UIVI achieves both tighter ELBO and better predictive performance than existing approaches at a similar computational cost.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Regression | UCI Yacht | NLL1.808 | 20 | |
| Regression | UCI Power | NLL2.794 | 15 | |
| Regression | UCI Wine | NLL0.907 | 15 | |
| Bayesian Neural Networks | UCI CONCRETE (test) | RMSE0.5 | 8 | |
| Regression | UCI Boston | NLL2.49 | 6 | |
| Regression | UCI Concrete | NLL3.331 | 6 | |
| Regression | UCI Protein | NLL2.973 | 6 | |
| Bayesian Neural Network Regression | Protein (test) | RMS Error0.92 | 4 | |
| Density Estimation | Banana | Rejection Rate7 | 4 | |
| Bayesian Neural Network Regression | Yacht (test) | RMS0.18 | 4 |