collaborators

7 papers

stat.ML2026

Stochastic Gradient Variational Inference with Price's Gradient Estimator from Bures-Wasserstein to Parameter Space

Kyurae Kim, Qiang Fu, Yi-An Ma +2

For approximating a target distribution given only its unnormalized log-density, stochastic gradient-based variational inference (VI) algorithms are a popular approach. For example…

cs.LG2026

Knowledge Gradient for Preference Learning

Kaiwen Wu, Jacob R. Gardner

The knowledge gradient is a popular acquisition function in Bayesian optimization (BO) for optimizing black-box objectives with noisy function evaluations. Many practical settings,…

stat.ML2025

Tuning Sequential Monte Carlo Samplers via Greedy Incremental Divergence Minimization

Kyurae Kim, Zuheng Xu, Jacob R. Gardner +1

The performance of sequential Monte Carlo (SMC) samplers heavily depends on the tuning of the Markov kernels used in the path proposal. For SMC samplers with unadjusted Markov kern…

stat.ML2025

Provably Scalable Black-Box Variational Inference with Structured Variational Families

Joohwan Ko, Kyurae Kim, Woo Chang Kim +1

Variational families with full-rank covariance approximations are known not to work well in black-box variational inference (BBVI), both empirically and theoretically. In fact, rec…

stat.ML2025

Nearly Dimension-Independent Convergence of Mean-Field Black-Box Variational Inference

Kyurae Kim, Yi-An Ma, Trevor Campbell +1

We prove that, given a mean-field location-scale variational family, black-box variational inference (BBVI) with the reparametrization gradient converges at a rate that is nearly i…

cs.LG2025

Computation-Aware Gaussian Processes: Model Selection And Linear-Time Inference

Jonathan Wenger, Kaiwen Wu, Philipp Hennig +3

Model selection in Gaussian processes scales prohibitively with the size of the training dataset, both in time and memory. While many approximations exist, all incur inevitable app…