← Search

Martin Wainwright

15 accepted papers

2023

Krylov–Bellman boosting: Super-linear policy evaluation in general state spaces

AISTATS 2023poster

We present and analyze the Krylov–Bellman Boosting algorithm for policy evaluation in general state spaces. It alternates between fitting the Bellman residual using non-parametric regression (as in boosting), and estimating the value function via the least-squares temporal difference (LSTD) procedur…

Cited by 3SourcePDFScholar
2022

A new similarity measure for covariate shift with applications to nonparametric regression

ICML 2022oral

We study covariate shift in the context of nonparametric regression. We introduce a new measure of distribution mismatch between the source and target distributions using the integrated ratio of probabilities of balls at a given radius. We use the scaling of this measure with respect to the radius t…

Cited by 48SourcePDFScholar
2022

Stabilizing Q-learning with Linear Architectures for Provable Efficient Learning

ICML 2022spotlight

The Q-learning algorithm is a simple, fundamental and practically very effective reinforcement learning algorithm. However, the basic protocol can exhibit an unstable behavior when implemented even with simple linear function approximation. While tools like target networks and experience replay are…

Cited by 6SourcePDFScholar
2021

Provable Benefits of Actor-Critic Methods for Offline Reinforcement Learning

NeurIPS 2021poster

Actor-critic methods are widely used in offline reinforcement learning practice, but are not so well-understood theoretically. We propose a new offline actor-critic algorithm that naturally incorporates the pessimism principle, leading to several key advantages compared to the state of the art. The…

Cited by 152SourcePDFScholar
2020

Sharp Analysis of Expectation-Maximization for Weakly Identifiable Models

AISTATS 2020poster

We study a class of weakly identifiable location-scale mixture models for which the maximum likelihood estimates based on $n$ i.i.d. samples are known to have lower accuracy than the classical $n^{- \frac{1}{2}}$ error. We investigate whether the Expectation-Maximization (EM) algorithm also converge…

Cited by 33SourcePDFScholar
2019

Derivative-Free Methods for Policy Optimization: Guarantees for Linear Quadratic Systems

AISTATS 2019poster

We study derivative-free methods for policy optimization over the class of linear policies. We focus on characterizing the convergence rate of a canonical stochastic, two-point, derivative-free method for linear-quadratic systems in which the initial state of the system is drawn at random. In partic…

Cited by 243SourcePDFScholar
2018

Approximate Ranking from Pairwise Comparisons

AISTATS 2018poster

A common problem in machine learning is to rank a set of n items based on pairwise comparison. Here, ranking refers to partitioning the items into sets of pre-specified sizes according to theirs scores, which includes identification of the top-k items as the most prominent special case. The score o…

Cited by 0SourcePDFScholar
2018

Convergence guarantees for a class of non-convex and non-smooth optimization problems

ICML 2018oral

Non-convex optimization problems arise frequently in machine learning, including feature selection, structured matrix learning, mixture modeling, and neural network training. We consider the problem of finding critical points of a broad class of non-convex problems with non-smooth components. We ana…

Cited by 59SourcePDFScholar
2018

Learning to Explain: An Information-Theoretic Perspective on Model Interpretation

ICML 2018oral

We introduce instancewise feature selection as a methodology for model interpretation. Our method is based on learning a function to extract a subset of features that are most informative for each given example. This feature selector is trained to maximize the mutual information between selected fea…

2018

SAFFRON: an Adaptive Algorithm for Online Control of the False Discovery Rate

ICML 2018oral

In the online false discovery rate (FDR) problem, one observes a possibly infinite sequence of $p$-values $P_1,P_2,…$, each testing a different null hypothesis, and an algorithm must pick a sequence of rejection thresholds $\alpha_1,\alpha_2,…$ in an online fashion, effectively rejecting the $k$-th…

2017

On the Learnability of Fully-Connected Neural Networks

AISTATS 2017poster

Despite the empirical success of deep neural networks, there is limited theoretical understanding on the learnability of these models using a polynomial-time algorithm. In this paper, we characterize the learnability of fully-connected neural networks via both positive and negative results. We focus…

Cited by 67SourcePDFScholar
2016

Stochastically Transitive Models for Pairwise Comparisons: Statistical and Computational Issues

ICML 2016poster

There are various parametric models for analyzing pairwise comparison data, including the Bradley-Terry-Luce (BTL) and Thurstone models, but their reliance on strong parametric assumptions is limiting. In this work, we study a flexible model for pairwise comparisons, under which the probabilities of…

Cited by 193SourcePDFScholar
2015

Distributed Estimation of Generalized Matrix Rank: Efficient Algorithms and Lower Bounds

ICML 2015poster

We study the following generalized matrix rank estimation problem: given an n-by-n matrix and a constant c > 0, estimate the number of eigenvalues that are greater than c. In the distributed setting, the matrix of interest is the sum of m matrices held by separate machines. We show that any determin…

Cited by 29SourcePDFScholar
2015

Estimation from Pairwise Comparisons: Sharp Minimax Bounds with Topology Dependence

AISTATS 2015poster

Consider the problem of identifying the underlying qualities of a set of items based on measuring noisy comparisons between pairs of items. The Bradley-Terry-Luce (BTL) and Thurstone models are the most widely used parametric models for such pairwise comparison data. Working within a standard minima…

Cited by 211SourcePDFScholar