Gradient Estimation for Mixture Variational Inference
This work provides an elementary derivation of a single-sample post-stratified estimator—previously derived via transport equations—and proves a variance reduction relative to simple random sampling, expanding the applicability of implicit reparameterization and reducing its computational complexity.