Treatment Allocation under Uncertain Costs

Hao Sun,Evan Munro,Georgy Kalashnov,Shuyang Du,Stefan Wager
2021-03-20
Abstract:We consider the problem of learning how to optimally allocate treatments whose cost is uncertain and can vary with pre-treatment covariates. This setting may arise in medicine if we need to prioritize access to a scarce resource that different patients would use for different amounts of time, or in marketing if we want to target discounts whose cost to the company depends on how much the discounts are used. Here, we show that the optimal treatment allocation rule under budget constraints is a thresholding rule based on priority scores, and we propose a number of practical methods for learning these priority scores using data from a randomized trial. Our formal results leverage a statistical connection between our problem and that of learning heterogeneous treatment effects under endogeneity using an instrumental variable. We find our method to perform well in a number of empirical evaluations.
Methodology
What problem does this paper attempt to address?
This paper attempts to solve the problem of how to optimally allocate treatment resources under budget constraints, especially when the treatment cost is uncertain and may vary with pretreatment covariates. Specifically, the paper focuses on how to optimize the allocation of treatment or intervention measures within a limited budget according to individual characteristics (such as health status, consumption habits, etc.) to maximize the overall benefit. For example, in the medical field, this may involve how to prioritize the allocation of scarce intensive care beds; in marketing, it may involve how to provide the most effective discount offers for different customer groups. The paper proposes a method based on priority scores to solve this problem. This method first calculates a cost - aware priority score according to an individual's pretreatment covariates. This score is the ratio of the conditional expected incremental revenue to the conditional expected incremental cost. Then, individuals are ranked according to this score, and those with higher scores are treated within the budget limit. The paper also explores how to estimate these priority scores from randomized trial data and proposes several practical learning methods, including a non - parametric estimation method based on generalized random forests. In addition, the paper proves the effectiveness of the proposed method through multiple empirical evaluations. In summary, the main contribution of this paper is to provide a framework for optimizing the allocation of treatment resources based on individual characteristics in the presence of budget constraints and cost uncertainty.