In Rstan, I’m running ADVI on some models. The ADVI gives weird results and I would like to see the ELBO values and stepsize it took during iteration. Is there an easy way to extract these 2 things after ADVI converges in Rstan?
No. RStan has Pareto-k diagnostic to check whether the end result is a good approximation of the true posterior, but that is technically different from convergence. There is an issue to add this to C++ so that it could be made available in other interfaces. Stan’s advi implementation is not the best possible, but there is not much pressure to improve it as dynamic HMC is more accurate and likely to be also faster (see Langevin Dynamics as Nonparametric Variational Inference | OpenReview)