← Search

Kaiwen Wu

12 accepted papers

2024

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

NeurIPS 2024poster

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 approximation error. Recent work accounts for this error in the form of computational uncertainty, which enables---at the c…

Cited by 3SourcePDFScholar
2024

Large-Scale Gaussian Processes via Alternating Projection

AISTATS 2024poster

Training and inference in Gaussian processes (GPs) require solving linear systems with $n\times n$ kernel matrices. To address the prohibitive $\mathcal{O}(n^3)$ time complexity, recent work has employed fast iterative methods, like conjugate gradients (CG). However, as datasets increase in magnitud…

2023

Discovering Many Diverse Solutions with Bayesian Optimization

AISTATS 2023poster

Bayesian optimization (BO) is a popular approach for sample-efficient optimization of black-box objective functions. While BO has been successfully applied to a wide range of scientific applications, traditional approaches to single-objective BO only seek to find a single best solution. This can be…

2023

On the Convergence of Black-Box Variational Inference

NeurIPS 2023poster

We provide the first convergence guarantee for black-box variational inference (BBVI) with the reparameterization gradient. While preliminary investigations worked on simplified versions of BBVI (e.g., bounded domain, bounded support, only optimizing for the scale, and such), our setup does not ne…

Cited by 26SourcePDFScholar
2023

Practical and Matching Gradient Variance Bounds for Black-Box Variational Bayesian Inference

ICML 2023oral

Understanding the gradient variance of black-box variational inference (BBVI) is a crucial step for establishing its convergence and developing algorithmic improvements. However, existing studies have yet to show that the gradient variance of BBVI satisfies the conditions used to study the convergen…

Cited by 6SourcePDFScholar
2023

The Behavior and Convergence of Local Bayesian Optimization

NeurIPS 2023spotlight

A recent development in Bayesian optimization is the use of local optimization strategies, which can deliver strong empirical performance on high-dimensional problems compared to traditional global strategies. The "folk wisdom" in the literature is that the focus on local optimization sidesteps the…

2023

Variational Gaussian Processes with Decoupled Conditionals

NeurIPS 2023poster

Variational Gaussian processes (GPs) approximate exact GP inference by using a small set of inducing points to form a sparse approximation of the true posterior, with the fidelity of the model increasing with additional inducing points. Although the approximation error in principle can be reduced th…

Cited by 2SourcePDFScholar
2022

Local Bayesian optimization via maximizing probability of descent

NeurIPS 2022accept

Local optimization presents a promising approach to expensive, high-dimensional black-box optimization by sidestepping the need to globally explore the search space. For objective functions whose gradient cannot be evaluated directly, Bayesian optimization offers one solution -- we construct a proba…

2020

On Minimax Optimality of GANs for Robust Mean Estimation

AISTATS 2020poster

Generative adversarial networks (GANs) have become one of the most popular generative modeling techniques in machine learning. In this work, we study the statistical and robust properties of GANs for Gaussian mean estimation under Huber’s contamination model, where an epsilon proportion of training…

2019

Distributional Reinforcement Learning for Efficient Exploration

ICML 2019oral

In distributional reinforcement learning (RL), the estimated distribution of value functions model both the parametric and intrinsic uncertainties. We propose a novel and efficient exploration method for deep RL that has two components. The first is a decaying schedule to suppress the intrinsic unce…

Cited by 112SourcePDFScholar