Search Results for author: Stefan Wager

Found 47 papers, 28 papers with code

Off-Policy Evaluation in Markov Decision Processes under Weak Distributional Overlap

no code implementations13 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.

Off-policy evaluation

Policy Learning under Biased Sample Selection

no code implementations23 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.

Experimenting under Stochastic Congestion

no code implementations22 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.

Experimental Design

Learning from a Biased Sample

1 code implementation5 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.

Decision Making Length-of-Stay prediction

What Makes Forest-Based Heterogeneous Treatment Effect Estimators Work?

2 code implementations21 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.

Policy Learning with Competing Agents

1 code implementation4 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.

Partial Likelihood Thompson Sampling

no code implementations2 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.

Thompson Sampling

Thompson Sampling with Unrestricted Delays

no code implementations24 Feb 2022 Han Wu, Stefan Wager

We investigate properties of Thompson Sampling in the stochastic multi-armed bandit problem with delayed feedback.

Thompson Sampling

Covariate Balancing Sensitivity Analysis for Extrapolating Randomized Trials across Locations

no code implementations9 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.

Evaluating Treatment Prioritization Rules via Rank-Weighted Average Treatment Effects

1 code implementation15 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.

Marketing

Off-Policy Evaluation in Partially Observed Markov Decision Processes under Sequential Ignorability

no code implementations24 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).

Off-policy evaluation

Treatment Effects in Market Equilibrium

1 code implementation23 Sep 2021 Evan Munro, Stefan Wager, Kuang Xu

When randomized trials are run in a marketplace equilibriated by prices, interference arises.

Experimental Design

Weak Signal Asymptotics for Sequentially Randomized Experiments

no code implementations25 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.

Thompson Sampling

Estimating heterogeneous treatment effects with right-censored data via causal survival forests

2 code implementations27 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.

Confidence Intervals for Policy Evaluation in Adaptive Experiments

1 code implementation7 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.

Experimental Design Multi-Armed Bandits

Doubly robust treatment effect estimation with missing attributes

2 code implementations23 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

Smoothness-Adaptive Contextual Bandits

1 code implementation22 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.

Decision Making Multi-Armed Bandits

Sufficient Representations for Categorical Variables

3 code implementations26 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.

Covariate-Powered Empirical Bayes Estimation

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.

Nonparametric Heterogeneous Treatment Effect Estimation in Repeated Cross Sectional Designs

2 code implementations28 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

Learning When-to-Treat Policies

1 code implementation23 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.

Decision Making

Sparsity Double Robust Inference of Average Treatment Effects

no code implementations2 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

Estimating Treatment Effects with Causal Forests: An Application

2 code implementations20 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

Confidence Intervals for Nonparametric Empirical Bayes Analysis

1 code implementation7 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

Synthetic Difference in Differences

4 code implementations24 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

Offline Multi-Action Policy Learning: Generalization and Optimization

1 code implementation10 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.

Local Linear Forests

3 code implementations30 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.

Causal Inference regression +1

Quasi-Oracle Estimation of Heterogeneous Treatment Effects

2 code implementations13 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.

regression

Augmented Minimax Linear Estimation

2 code implementations30 Nov 2017 David A. Hirshberg, Stefan Wager

Many statistical estimands can expressed as continuous linear functionals of a conditional expectation function.

Methodology 62F12

Optimized Regression Discontinuity Designs

1 code implementation4 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

Policy Learning with Observational Data

1 code implementation9 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.

Fairness

Generalized Random Forests

5 code implementations5 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.

valid

High-dimensional regression adjustments in randomized experiments

no code implementations22 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.

regression valid +1

Approximate Residual Balancing: De-Biased Inference of Average Treatment Effects in High Dimensions

1 code implementation25 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

Data Augmentation via Levy Processes

1 code implementation21 Mar 2016 Stefan Wager, William Fithian, Percy Liang

The framework imagines data as being drawn from a slice of a Levy process.

Image Augmentation

Estimation and Inference of Heterogeneous Treatment Effects using Random Forests

6 code implementations14 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.

Marketing valid

High-Dimensional Asymptotics of Prediction: Ridge Regression and Classification

1 code implementation10 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.

Classification General Classification +2

The Statistics of Streaming Sparse Regression

no code implementations13 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.

regression

Bootstrap-Based Regularization for Low-Rank Matrix Estimation

no code implementations30 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.

Altitude Training: Strong Bounds for Single-Layer Dropout

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.

Asymptotic Theory for Random Forests

no code implementations2 May 2014 Stefan Wager

Random forests have proven to be reliable predictive algorithms in many application areas.

Confidence Intervals for Random Forests: The Jackknife and the Infinitesimal Jackknife

no code implementations18 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).

Feedback Detection for Live Predictors

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.

Causal Inference

Weakly supervised clustering: Learning fine-grained signals from coarse labels

no code implementations4 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.

Classification Clustering +1

Dropout Training as Adaptive Regularization

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.

Document Classification

Cannot find the paper you are looking for? You can Submit a new open access paper.