no code implementations • 13 Feb 2024 • Mohammad Mehrabi, Stefan Wager
In this paper, we re-visit the task of off-policy evaluation in MDPs under a weaker notion of distributional overlap, and introduce a class of truncated doubly robust (TDR) estimators which we find to perform well in this setting.
no code implementations • 23 Apr 2023 • Lihua Lei, Roshni Sahoo, Stefan Wager
Practitioners often use data from a randomized controlled trial to learn a treatment assignment policy that can be deployed on a target population.
no code implementations • 22 Feb 2023 • Shuangning Li, Ramesh Johari, Xu Kuang, Stefan Wager
We study randomized experiments in a service system when stochastic congestion can arise from temporarily limited supply and/or demand.
1 code implementation • 5 Sep 2022 • Roshni Sahoo, Lihua Lei, Stefan Wager
Applying the distributionally robust optimization framework, we propose a method for learning a decision rule that minimizes the worst-case risk incurred under a family of test distributions that can generate the training distribution under $\Gamma$-biased sampling.
2 code implementations • 21 Jun 2022 • Susanne Dandl, Torsten Hothorn, Heidi Seibold, Erik Sverdrup, Stefan Wager, Achim Zeileis
A related approach, called "model-based forests", that is geared towards randomized trials and simultaneously captures effects of both prognostic and predictive variables, was introduced by Seibold, Zeileis and Hothorn (2018) along with a modular implementation in the R package model4you.
1 code implementation • 4 Apr 2022 • Roshni Sahoo, Stefan Wager
Decision makers often aim to learn a treatment assignment policy under a capacity constraint on the number of agents that they can treat.
no code implementations • 2 Mar 2022 • Han Wu, Stefan Wager
We consider the problem of deciding how best to target and prioritize existing vaccines that may offer protection against new variants of an infectious disease.
no code implementations • 24 Feb 2022 • Han Wu, Stefan Wager
We investigate properties of Thompson Sampling in the stochastic multi-armed bandit problem with delayed feedback.
no code implementations • 9 Dec 2021 • Xinkun Nie, Guido Imbens, Stefan Wager
The ability to generalize experimental results from randomized control trials (RCTs) across locations is crucial for informing policy decisions in targeted regions.
1 code implementation • 15 Nov 2021 • Steve Yadlowsky, Scott Fleming, Nigam Shah, Emma Brunskill, Stefan Wager
We propose rank-weighted average treatment effect (RATE) metrics as a simple and general family of metrics for comparing and testing the quality of treatment prioritization rules.
no code implementations • 24 Oct 2021 • Yuchen Hu, Stefan Wager
We consider off-policy evaluation of dynamic treatment rules under sequential ignorability, given an assumption that the underlying system can be modeled as a partially observed Markov decision process (POMDP).
1 code implementation • 23 Sep 2021 • Evan Munro, Stefan Wager, Kuang Xu
When randomized trials are run in a marketplace equilibriated by prices, interference arises.
no code implementations • 25 Jan 2021 • Xu Kuang, Stefan Wager
We use the lens of weak signal asymptotics to study a class of sequentially randomized experiments, including those that arise in solving multi-armed bandit problems.
2 code implementations • 27 Jan 2020 • Yifan Cui, Michael R. Kosorok, Erik Sverdrup, Stefan Wager, Ruoqing Zhu
Forest-based methods have recently gained in popularity for non-parametric treatment effect estimation.
1 code implementation • 7 Nov 2019 • Vitor Hadad, David A. Hirshberg, Ruohan Zhan, Stefan Wager, Susan Athey
In this context, typical estimators that use inverse propensity weighting to eliminate sampling bias can be problematic: their distributions become skewed and heavy-tailed as the propensity scores decay to zero.
2 code implementations • 23 Oct 2019 • Imke Mayer, Erik Sverdrup, Tobias Gauss, Jean-Denis Moyer, Stefan Wager, Julie Josse
We find, however, that doubly robust modifications of standard methods for average treatment effect estimation with missing data repeatedly perform better than their non-doubly robust baselines; for example, doubly robust generalized propensity score methods beat inverse-weighting with the generalized propensity score.
Methodology 93C41, 62G35, 62F35, 62P10
1 code implementation • 22 Oct 2019 • Yonatan Gur, Ahmadreza Momeni, Stefan Wager
In this work, we consider a framework where the smoothness of payoff functions is not known, and study when and how algorithms may adapt to unknown smoothness.
3 code implementations • 26 Aug 2019 • Jonathan Johannemann, Vitor Hadad, Susan Athey, Stefan Wager
Many learning algorithms require categorical data to be transformed into real vectors before it can be used as input.
4 code implementations • NeurIPS 2019 • Nikolaos Ignatiadis, Stefan Wager
We study methods for simultaneous analysis of many noisy experiments in the presence of rich covariate information.
2 code implementations • 28 May 2019 • Chen Lu, Xinkun Nie, Stefan Wager
Identifying heterogeneity in a population's response to a health or policy intervention is crucial for evaluating and informing policy decisions.
Methodology
1 code implementation • 23 May 2019 • Xinkun Nie, Emma Brunskill, Stefan Wager
Many applied decision-making problems have a dynamic component: The policymaker needs not only to choose whom to treat, but also when to start which treatment.
no code implementations • 2 May 2019 • Jelena Bradic, Stefan Wager, Yinchu Zhu
Many popular methods for building confidence intervals on causal effects under high-dimensional confounding require strong "ultra-sparsity" assumptions that may be difficult to validate in practice.
Statistics Theory Econometrics Methodology Statistics Theory
2 code implementations • 20 Feb 2019 • Susan Athey, Stefan Wager
We apply causal forests to a dataset derived from the National Study of Learning Mindsets, and consider resulting practical and conceptual challenges.
Methodology
1 code implementation • 7 Feb 2019 • Nikolaos Ignatiadis, Stefan Wager
In an empirical Bayes analysis, we use data from repeated sampling to imitate inferences made by an oracle Bayesian with extensive knowledge of the data-generating distribution.
Methodology
4 code implementations • 24 Dec 2018 • Dmitry Arkhangelsky, Susan Athey, David A. Hirshberg, Guido W. Imbens, Stefan Wager
We present a new estimator for causal effects with panel data that builds on insights behind the widely used difference in differences and synthetic control methods.
Methodology
1 code implementation • 10 Oct 2018 • Zhengyuan Zhou, Susan Athey, Stefan Wager
In many settings, a decision-maker wishes to learn a rule, or policy, that maps from observable characteristics of an individual to an action.
3 code implementations • 30 Jul 2018 • Rina Friedberg, Julie Tibshirani, Susan Athey, Stefan Wager
Random forests are a powerful method for non-parametric regression, but are limited in their ability to fit smooth signals, and can show poor predictive performance in the presence of strong, smooth effects.
2 code implementations • 13 Dec 2017 • Xinkun Nie, Stefan Wager
We first estimate marginal effects and treatment propensities in order to form an objective function that isolates the causal component of the signal.
2 code implementations • 30 Nov 2017 • David A. Hirshberg, Stefan Wager
Many statistical estimands can expressed as continuous linear functionals of a conditional expectation function.
Methodology 62F12
1 code implementation • 4 May 2017 • Guido Imbens, Stefan Wager
The increasing popularity of regression discontinuity methods for causal inference in observational studies has led to a proliferation of different estimating strategies, most of which involve first fitting non-parametric regression models on both sides of a treatment assignment boundary and then reporting plug-in estimates for the effect of interest.
Methodology
1 code implementation • 9 Feb 2017 • Susan Athey, Stefan Wager
In many areas, practitioners seek to use observational data to learn a treatment assignment policy that satisfies application-specific constraints, such as budget, fairness, simplicity, or other functional form constraints.
5 code implementations • 5 Oct 2016 • Susan Athey, Julie Tibshirani, Stefan Wager
We propose generalized random forests, a method for non-parametric statistical estimation based on random forests (Breiman, 2001) that can be used to fit any quantity of interest identified as the solution to a set of local moment equations.
no code implementations • 22 Jul 2016 • Stefan Wager, Wenfei Du, Jonathan Taylor, Robert Tibshirani
We study the problem of treatment effect estimation in randomized experiments with high-dimensional covariate information, and show that essentially any risk-consistent regression adjustment can be used to obtain efficient estimates of the average treatment effect.
1 code implementation • 25 Apr 2016 • Susan Athey, Guido W. Imbens, Stefan Wager
There are many settings where researchers are interested in estimating average treatment effects and are willing to rely on the unconfoundedness assumption, which requires that the treatment assignment be as good as random conditional on pre-treatment variables.
Methodology Econometrics Statistics Theory Statistics Theory
1 code implementation • 21 Mar 2016 • Stefan Wager, William Fithian, Percy Liang
The framework imagines data as being drawn from a slice of a Levy process.
6 code implementations • 14 Oct 2015 • Stefan Wager, Susan Athey
Many scientific and engineering challenges -- ranging from personalized medicine to customized marketing recommendations -- require an understanding of treatment effect heterogeneity.
1 code implementation • 10 Jul 2015 • Edgar Dobriban, Stefan Wager
We provide a unified analysis of the predictive risk of ridge regression and regularized discriminant analysis in a dense random effects model.
2 code implementations • 22 Mar 2015 • Stefan Wager, Guenther Walther
To support our analysis we introduce a notion of adaptive concentration for regression trees.
no code implementations • 13 Dec 2014 • Jacob Steinhardt, Stefan Wager, Percy Liang
We present a sparse analogue to stochastic gradient descent that is guaranteed to perform well under similar conditions to the lasso.
no code implementations • 30 Oct 2014 • Julie Josse, Stefan Wager
In the simplest case, with an isotropic noise model, our method is equivalent to a classical singular value shrinkage estimator.
no code implementations • NeurIPS 2014 • Stefan Wager, William Fithian, Sida Wang, Percy Liang
Dropout training, originally designed for deep neural networks, has been successful on high-dimensional single-layer natural language tasks.
no code implementations • 2 May 2014 • Stefan Wager
Random forests have proven to be reliable predictive algorithms in many application areas.
no code implementations • 18 Nov 2013 • Stefan Wager, Trevor Hastie, Bradley Efron
Our work builds on variance estimates for bagging proposed by Efron (1992, 2012) that are based on the jackknife and the infinitesimal jackknife (IJ).
no code implementations • NeurIPS 2014 • Stefan Wager, Nick Chamandy, Omkar Muralidharan, Amir Najmi
A predictor that is deployed in a live production system may perturb the features it uses to make predictions.
no code implementations • 4 Oct 2013 • Stefan Wager, Alexander Blocker, Niall Cardin
Consider a classification problem where we do not have access to labels for individual training examples, but only have average labels over subpopulations.
no code implementations • NeurIPS 2013 • Stefan Wager, Sida Wang, Percy Liang
Dropout and other feature noising schemes control overfitting by artificially corrupting the training data.