Extracted main text — title through conclusion, appendix excluded. This is what our citation measures are computed over, published so the extraction can be checked by eye.
72,990 characters · 19 sections · 27 citation commands
Machine-Learning Tests for Effects on Multiple Outcomes
\begingroup \footnote{ Jens Ludwig, Harris School of Public Policy, University of Chicago and NBER, [email removed]. Sendhil Mullainathan, Booth School of Business, University of Chicago and NBER, [email removed]. Jann Spiess, Microsoft Research New England, [email removed]. An earlier versions of this manuscript was titled “Testing Effects on Groups of Outcomes” (November 2016). We thank seminar participants at Harvard University, the University of Chicago, and the University of Pennsylvania for helpful comments. } \addtocounter{footnote}{-1} \endgroup
In this paper we present tools for applied researchers that re-purpose off-the-shelf methods from the computer-science field of machine learning to create a “discovery engine” for data from randomized controlled trials (RCTs). The applied problem we seek to solve is that economists invest vast resources into carrying out RCTs, including the collection of a rich set of candidate outcome measures. But given concerns about inference in the presence of multiple testing, economists usually wind up exploring just a small subset of the hypotheses that the available data could be used to test. This prevents us from extracting as much information as possible from each RCT, which in turn impairs our ability to develop new theories or strengthen the design of policy interventions.
As a concrete example, consider an RCT of some health-related intervention like subsidized health insurance. This type of RCT would typically include original in-person data collection to supplement administrative data like electronic health records (EHR). Since these types of health interventions could plausibly affect a wide range of health problems (or precursors to health problems) as well as health-related behaviors, we typically cast a wide net during the data-collection stage and assemble as wide a range of plausibly relevant measures as we can (recognizing this is just a subset of what could be potentially measured).
Then we reach the analysis stage. Because under most multiple-testing procedures the penalty to a given outcome’s $p$-value increases as the number of outcomes considered increases, all else equal we usually try to discipline ourselves and limit the number of measures we turn into outcomes to analyze. Then there is the question of how to turn measures into outcomes. Our theory says “health” should be affected, but does that mean self-reported global health or specific health problems or particular physical limitations or body mass index or cholesterol or glycated hemoglobin? Should we focus just on how the intervention affects the mean values of these outcomes, or should we explore other parts of the distributions for some outcomes? Or should we examine the joint distributions of multiple outcomes, for example if we thought “self-reported health very good and sees doctor for regular check-ups” might be particularly affected and also reveal something about underlying mechanisms of action?
In the end, applied researchers do the best they can using some combination of theory, previous findings and intuition to specify which hypotheses to test. But as our example makes abundantly clear, this is usually just a small subset of the hypotheses that could be tested. Of course, applied researchers know this better than anyone. They also know that exploration often leads to the biggest discoveries. For example, when Congress passed the Housing and Community Development Act of 1992 that set aside \$102 million for the Moving to Opportunity (MTO) demonstration, it required HUD to submit a report to Congress “describing the long-term housing, employment, and educational achievements of the families assisted under the demonstration program,” as well as “any other information the Secretary considers appropriate in evaluating the demonstration.” But qualitative interviews with the MTO participants raised the possibility that the biggest changes occured not in the areas of housing, employment, or education, but rather the reduced trauma and anxiety associated with substantial gains in safety Kling:2007cl. Future quantitative evaluations confirmed that some of the most important impacts fell under what at the launch of MTO were thought to be either so unimportant or unlikely that they were relegated to the category of “any other information” Kling:2007cl,sanbonm2011,ludwig2011,ludwig2012,kessler2014.
The machine-learning approach we propose here is a “discovery engine” intended to complement existing joint- and multiple-testing methods that are based on ex-ante curation by the researcher. It is based on two simple observations from the statistics and computer-science literature. First, the `false-positive' problem that a group of outcomes poses can be thought of finding something in a given experiment that is idiosyncratic to that sample, rather than a true feature of the underlying data-generating process. Put this way, we can see that the general structure of this concern is similar to the concern within machine learning of `over-fitting,' which has been the focus of a large literature in statistics and computer science. We can thus leverage sample-splitting methods from the standard machine-learning playbook, which are designed to control over-fitting to ensure that statistical models capture true structure in the world rather than idiosyncracies of any particular dataset.
The second simple observation, going back to at least friedman2004multivariate, is that the question of whether treatment $T$ (say binary) affects a group of variables $Y= (Y_1,...,Y_k)$ in an experiment is equivalent to the question whether $T$ is predictable using $Y$ (better than some trivial benchmark). In the parlance of Kleinberg:2015eo and Mullain:2017, this simple observation turns testing effects on many outcome variables into a prediction task (“$\hat{y}$ problem”, which here really is a “$\hat{T}$ problem”). Predictability takes the place of a treatment effect on the full distribution of outcomes. This formulation allows us to leverage data-driven predictors from the machine-learning literature to flexibly mine for effects, rather than rely on more rigid approaches like multiple-testing corrections and pre-analysis plans.
We show that, for any samples size, this test produces a $p$-value that is exactly sized under the null hypothesis of jointly no effect on the group of outcomes. We also discuss how we can use our procedure to learn something about {\em where} any effect happens for purposes like testing theories or carrying out benefit--cost analyses of specific interventions. In ongoing work, we also extend the test to deal with typical features of real-world experiments, namely missing data, conditional or clustered randomization, and stratification. And since this method is based on re-purposing existing off-the-shelf methods from machine learning, it has (from the perspective of applied research) the great virtue of being quite straightforward to implement.
In framing the many-outcomes problem as testing whether two distributions are the same, we relate our work to the general two-sample problem, in particular non-parametric approaches based on matching rosenbaum2005exact or on kernels gretton2007kernel. In terms of using a prediction approach to the two-sample problem, we are building on a classic literature in economics and econometrics that studies discrimination using reverse regressions Goldberger:1984bp, as well as a literature in statistics that connects the two-sample problem to classification friedman2004multivariate. Relatedly, Gagnon-Bartsch2019-ay develop a classification test for differences between two distributions, focusing on testing covariate imbalance in experiments. Like them, we use a permutation test to obtain a valid $p$-value, paralleling recent uses of omnibus permutation tests and randomization inference in the evaluation of experiments Potter:2006kt,Ding:2015hza,Chetty:2016ev,Young:2016ve.
In terms of typical applications, our research is related to multiple testing procedures based on individual mean comparison tests. Multiple testing procedures control the overall probability of at least one false positive (family-wise error rate) or the proportions of false positives among all rejections benjamini1995controlling. The most prominent such procedure, the Bonferroni correction bonferroni1936teoria,dunn1961multiple, and its holm1979simple step-wise improvement, ignore the correlation between individual test statistics. Other, more recent procedures take the dependence structure of test statistics into account, for example the step-wise procedure by Romano:2005bw; most closely related to our framework, List:2016dt propose a bootstrap-based adaption to experiments.
We set up the many-outcomes problem in Section (ref). In Section (ref), we review some standard approaches, and discuss their applicability in selected economic examples. Section (ref) presents our prediction procedure and analyzes its properties. In Section (ref), we present some ideas on interpreting the prediction output. Section (ref) proposes a framework and concrete tools for directly providing simple representations of the causal effect on the distribution of outcomes. Section (ref) illustrates the testing approach on simulated data. In Section (ref), we conclude by discussing challenges from typical features of real-world experiments.
We consider the problem of testing whether a group of outcomes variables is affected in an experiment. Assume we have a sample of $n$ iid observations $(T_i,Y_i)$, with binary treatment $T_i$ assigned randomly. Randomization may be by clusters of observation and within some strata.
We have a group of $k$ scalar outcomes:
Our primary goal is to test whether treatment has an effect on this group of outcomes, that is, whether the distributions of $Y|T=1$ is the same as the distribution of $Y|T=0$. If, for example, the two distributions differ only by a ($k$-dimensional) mean shift by $\tau = (\tau_1,\ldots,\tau_k)'$, our null hypothesis of interest is $\tau_j = 0$ for all $j$ simultaneously.
A related but different question is whether there are effects on the individual outcome variables $Y_{ij}$. This question can be useful for testing a specific theory, or if researchers have a clear sense of which individual outcome variable is of central interest. But often this question is complementary to the testing the {\em joint} hypothesis of no effect on any outcome. Having run a complex and expensive trial, a minimal first-pass question is commonly to ask the question “Did the experiment have any effect at all?” Tests of this joint hypothesis can be useful even when researchers are interested in a single specific hypothesis, for example when we do not know how to convert a specific hypothesis into individual variables (such as “health” being measured via numerous outcomes).
In this section we consider the standard approaches to testing whether treatment has an effect on multiple outcomes. Standard approaches fall into two broad categories: {\em Tests based on mean comparisons of the outcomes}, such as a Wald test in a seemingly unrelated regression (SUR) of the outcomes on the treatment dummy; and {\em tests based on a pre-defined index}, where we aggregate all outcomes (usually linearly) into a single index and run one test on the treatment-control mean difference for this index Kling:2007cl. We begin by considering the implicit assumptions behind these standard approaches. We then consider how these assumptions fare when confronted with the range of applications that experimental economists may encounter in practice.
We first consider the assumptions behind tests of effects on a group of outcomes that are based on aggregating the results of individual mean-comparison tests. Before we consider the aggregation of mean values, it is useful to first focus on the means tests themselves.
To simplify the exposition we consider a simple class of data-generating processes. For outcome $j$, write
where the control baseline $C_j$ is constant, while the treatment effect $\tau_j$ and the mean-zero error term $\epsilon_j$ are random and possibly correlated both within and between outcomes, but independent of treatment assignment $T_j$. For simplicity, we assume here that the $\tau_j$ and $\epsilon_j$ are jointly Normally distributed. Our goal is to test whether treatment and control distributions are the same -- that is, whether $\tau_j=0$ for all $j$. What would be an appropriate test for the null hypothesis?
Many standard approaches are based on individual mean comparisons, that is, they estimate the individual mean differences and then aggregate those to test that all mean differences are simultaneously zero. If the variances were different between treatment and control groups, testing for $\tau_j=0$ by only testing means would potentially leave information about different spreads on the table, no matter how we aggregate between individual mean estimates. If we base our inference solely on mean comparisons, we thus implicitly assume that mainly the means are affected by the intervention.
Individual means tests alone do not directly provide a rejection criterion for the overall null hypothesis of no effect on the full group of outcomes; to obtain the latter, we need to aggregate individual estimates while ensuring that our test is properly sized (that is, that the probability of rejecting the null hypothesis does not exceed the desired level when there is indeed no effect on the outcomes). A particular simple form of aggregation are multiple comparison tests that aggregate the $p$-values $p_1,\ldots,p_k$ that come from the individual tests (in the case of treatment--control differences, typically individual $t$-tests). A standard aggregation procedure that is sometimes used to test the joint null hypothesis of no effect on a group of outcomes because of the simplicity of its implementation is the Bonferroni correction bonferroni1936teoria,dunn1961multiple: Only reject the null hypothesis of no effect at all if one of the $p$-values is smaller than $\alpha/k$, where $\alpha$ is the size of the test (typically $\alpha=5\%$) and $k$ is the number of outcomes. \footnote{There is a related class of multiple testing approaches that controls the false discovery rate benjamini1995controlling. Since our focus is on the group hypothesis of no overall effect, we do not further elaborate on this approach.} Improvements over these classical methods perform step-wise corrections that take into account the overall distribution of $p$-values holm1979simple and/or the correlation between test statistics Romano:2005bw.
When would a procedure like the Bonferroni-corrected multiple-comparison test be appropriate? To understand how multiple-testing correction procedures perform in our application to the many-outcomes problem, it is important to understand that they are designed as an answer to a different question: when they reject the overall null hypothesis of no effect, they also provide rejections of individual hypotheses that are informative about which of the outcomes is affected by treatment. This additional information, however, comes at a cost; in particular with many outcomes, multiple-testing corrections tend to be conservative, and a direct test of the joint hypothesis of no overall effect is a more efficient answer to the question we are asking in this paper.
Beyond being limited to pre-specified means tests and paying a cost for providing individual rejections, Bonferroni-type corrections are inefficient in another way: Since they map individual (valid) $p$-values $p_1,\ldots,p_k$ from mean comparisons to an overall $p$-value without taking into account the correlations between test statistics, they are typically not both valid (that is, have size bounded by the nominal size) and optimal (that is, not be dominated by some other test across alternative hypotheses) without restrictions on the between-outcome covariance structure of the $\tau_j$ and $\epsilon_j$. In particular:
Here, we assume that the equal-variance assumption from above holds, and that $p$-values are obtained from two-sided $t$-tests.
Since such aggregation procedures do not use the information contained in the correlation of test statistics, they cannot generally be adequate, motivating variants that take the joint dependence structure of test statistics into account Romano:2005bw. The two specific tests have largest size (although Bonferroni is still inefficient) for independent error terms and treatment effects between outcomes. This assumption means that outcomes are independent conditional on treatment assignment (only covary through the treatment), as expressed in the graphical model in Figure (ref).
In applications in which all outcomes belong to a group, we typically assume them to be (conditionally) correlated, for example because they represent different aspects of one or more abstract (latent) outcomes that are affected by treatment. In this case, we may want to aggregate individual outcomes to one (or a few) aggregate outcomes on which we then perform means tests, rather than aggregating many tests with a multiple-testing correction. As an extreme case, assume we know how the outcomes are related, and that this relationship is through a single latent variable:
If all correlation stems from one latent linear factor, and the treatment only operates through this factor, the outcomes are independent conditionally on this factor (Figure (ref)). In this case, an index (which estimates this latent factor) yields an adequate test by weighting each outcome by its signal–to–noise ratio.
The above linear index aggregates outcomes efficiently into a single test statistic provided that:
So how can we aggregate across outcomes efficiently if we do not know the correlation structure? Joint tests like a Wald test in a SUR, which run all means tests simultaneously and produce a single $p$-value, can correct for unknown linear correlation in test statistics by estimating this correlation from the data; as long as the impact on the distribution is fully captured by the means, a linear SUR is thus adequate:
The intuition behind this result is straightforward: If the test comes from a regression that is correctly specified, it is adequate. If the two distributions, on the other hand, differ in more than their means, this specific regression is misspecified, and a test that also tests the effect on variances or correlations between outcomes may have higher power.
Note that we do not require that individual outcomes are conditionally independent -- indeed, the SUR-based test can correct for correlation between the test statistics. However, the structural assumptions we put on the test to show efficiency imply that the outcomes are only affected by treatment through constant treatment effects on at most $k$ latent variables that are independent conditional on treatment (Figure (ref)). We can thus think of the correction that SUR makes in constructing a test statistic as running the test not on the observed, correlated variables, but instead on the underlying (conditionally) uncorrelated latent factors $L_j$, each of which is affected by the treatment through a mean shift, and then aggregating these independent test statistics efficiently.
\footnotetext{Leaving out/setting to zero all $K_j,\eta_j$ is without loss of generality.}
So how do the structures assumed by standard approaches to testing effects on a group of outcomes map into economic applications of the sort encountered by experimental economists in actual data applications? We discuss several hypothetical models that are intended to be quite simple in their structure and yet sufficient to highlight the difficulty with which standard methods will be able to capture even this simple structure.
In these examples, standard approaches leave information on the table, as they are misspecified for the economics of the situation. Each asks us to anticipate the effects we expect to see, but hardly any of the assumptions that justify the above tests maps well into the structure we have presented.
There are two ways to go from here: First, we could try to build the right test for the right situation, using case-specific knowledge and theory to guide which structural assumptions we feel comfortable with each time and how we go about exploiting this structure by creating an appropriate set of individual hypotheses; or second, we could be agnostic about the specific structure and instead leverage tools –- such as those provided by machine learning –- that are able to adapt flexibly to any given dataset, and thereby explode the number of individual hypotheses they search over. We take this second path.
We turn the basic null hypothesis – “the distribution of outcomes is the same in treatment as in control” -- into a prediction statement: “treatment status is not predictable from the outcomes”. We thus aim to predict $T$ from $Y$ using a flexible prediction function; if we find a function that predicts treatment status significantly better than chance, we have evidence of a difference between the two distributions.
The workhorse of our testing procedure will thus be a prediction function: It takes as input a sample of treatment assignment and outcomes, and constructs from this sample a prediction function that maps values of the outcomes to the probability of being in the treatment group. The better this prediction is -- that is, the lower the discrepancy between the predicted probabilities and realized treatment assignment -- the better the evidence that treatment and control groups are different.
One advantage of re-framing the group outcomes problem into one of prediction is that it enables us to take advantage of machine-learning approaches that are capable of considering very complicated functional forms. Moreover we can use the machinery of machine learning that guards against over-fitting in standard prediction applications to help find effects in a given experiment that reflect real underlying structure, rather than simply being artifacts of a particular sample.
To lay out our testing procedure, we make minimal assumptions on the data-generating process and the prediction function:
We will first introduce the testing procedure within this minimal framework, and then provide formal statements of its econometric properties. The validity of the test will not require any additional assumptions; for efficiency, we will assume more structure both about the data-generating process and the prediction technology.
Note that we intentionally do not restrict the class of algorithms for now and do not even formally require that they minimize out-of-sample loss $L(f)$ over some class of functions. This is to lay bare the relationship between prediction (of any quality) and the underlying hypothesis testing problem. Practically, of course, the power of our testing procedure comes from the existence of machine-learning algorithms $A$ that produce low out-of-sample prediction loss $L(\ensuremath{\hat{f}})$. As a result, when we prove formal statements about the power of our procedure, we will make further assumptions.
The basis of our test is simple: compute out-of-sample loss $\hat{L}$ in predicting treatment $T$ from $Y$; compute the distribution of out-of-sample loss under the null of no treatment effect (where $T$ cannot be predicted from $Y$ at all); and form a test based on where $\hat{L}$ falls in this distribution. Since in-sample loss is a biased measure of out-of-sample performance (the function may look good in-sample through overfitting, even if there is no signal), and since we do not want to restrict prediction algorithms to those few for which we know how to estimate out-of-sample loss from in-sample loss, we rely on sample-splitting methods in evaluating loss: we never fit a function on the same data that we evaluate it on.
The first variant applies this idea in the most straightforward way:
Note that this hold-out strategy ensures that all predictions are out-of-sample; however, it has an obvious inefficiency, as only a fraction of datapoints is used to fit the function -- and only a fraction is used to test its fit. This is corrected by our second variant, which uses every datapoint once for evaluating loss, and multiple times for fitting:
This procedure thus predicts $K$ times from $m = n - n/K$ datapoints; for $K=n$ (leave-one-out), the sample size of each training sample is maximal at $m=n-1$.
\paragraph{Obtaining a $p$-value.} Given a loss estimate, how do we obtain a $p$-value for the null hypothesis that the distribution of outcomes is not affected by the treatment?
We rely on sample-splitting methods and restrict ourselves to evaluating out-of-sample loss because we want to be agnostic about the algorithm used for prediction -- after all, we want to allow for complex machine-learning algorithms that may not fit our usual analytical estimation frameworks. For the same reason, we want to obtain a valid test under minimal assumptions. We therefore propose a permutation test:
The logic behind the permutation test is straightforward: If the null hypothesis is true, the data with permuted treatment assignment looks just like the original sample; if, to the contrary, we can predict better from the actual sample, this is evidence for a treatment effect. Crucially, this logic does not depend on any specific feature of the data-generating process or the algorithm, but it reflects the strength of the null hypothesis.
Note that the hold-out test does not require the re-estimation of prediction functions in every run; the cross-validation scheme, on the other hand, refits $K$ prediction function for every permutation draw.\footnote{In the Appendix, we will discuss a hybrid approach that permutes both training and hold-out set in the hold-out design as a compromise between computational cost and statistical power.} We see the main disadvantage of the cross-validation procedure over the holdout procedure as coming from the fact that it is much more computationally costly -- both because of repeated predictions for one sample and re-estimation for every permutation -- although this cost can be mitigated to some degree by the fact that all predictions can be computed in parallel.
We start with an analysis of our testing procedure under the minimal assumptions from Section (ref). By construction, the hold-out and $K$-fold permutation test have exact size (provided appropriate interpolation to form the $p$-value):
Note that we did not make any assumptions about the prediction algorithm; in particular, the tests have exact size even for complex machine-learning prediction algorithms.
When deciding whether to employ the above testing procedure, we care not just about size, but also power. In particular, we may be worried that we lose efficiency relative to a standard joint test based on mean differences in cases in which the effect is exclusively on the means of the outcomes, or relatively to a prediction-based test that uses the full sample and corrects for overfitting analytically. Towards efficiency, we first show asymptotic equivalence if we restrict ourselves to linear predictors, where the data comes from a linear model with Normal, homoscedastic errors:
We formulate these specific conditions because they represent a world in which a linear SUR is correctly specified and the corresponding Wald test is a natural test for the joint hypothesis of no effect. Indeed, these assumptions ensure that the full treatment effect is captured by mean shifts in the individual outcome dimensions. The normalization (ref) provides that asymptotic power does not take off to one asymptotically. We claim that the following equivalence results holds under these conditions:
A precise statement of this and all following results will be found in the Appendix.
Hence, as long as we use the linear-least squares estimator as a predictor, we do not asymptotically lose any power from putting the outcomes on the right-hand side of the regression, or from running a cross-validation permutation test. In particular, there is no (asymptotic) loss from using the prediction test with a linear predictor relative to a standard test based on pairwise comparisons.
While these results are reassuring, we motivated our prediction procedure through more flexible functional forms. So what happens if OLS is not correctly specified, and a more flexible algorithm will find additional structure?
This result is merely a formalization of the intuition we started our inquiry with: If our predictor picks up structure that is not linear, and that corresponds to a definite loss improvement in the large-sample limit, then it will ultimately reject the null hypothesis of no treatment effect, while a linear-based test may achieve strictly smaller power.
The above test gives us a $p$-value for the null hypothesis that there is no difference in outcome distribution between treatment and control groups, based on a prediction task. However, we may also be interested in what the effect is on -- that is, where in the outcome space the prediction algorithm found signal about treatment. This is relevant for using experiments to test economic theories, which often generate testable implications about which outcomes should and should not be affected in an experiment, and for carrying out benefit--cost analyses of specific interventions. However, with machine-learning techniques this is more complicated than just testing for an overall effect on a group of outcomes -- that is, testing whether treatment assignment is predictable. The reason is that machine-learning tools are designed to generate good out-of-sample predictions by extracting as much signal as possible from the explanatory variables, rather than to isolate the individual relationships of each predictor with the left-hand side variable. In this section we discuss how we navigate this constraint.
A prediction function of treatment assignment from the outcomes is itself a function $\ensuremath{\hat{f}}(Y)$ of the outcomes $Y$, and thus an index. Indeed, we can interpret the prediction exercise with respect to squared-error loss as the estimation of an optimal index:
When we choose a prediction function, we thus also estimate an optimal index -- where “optimal” refers to it being maximally different between treatment and control. This mean difference also offers a quantification of the difference between the distributions.
If we are not just interested in whether there is an effect at all (expressed by said $p$-value), but also where it is, we can use the predicted treatment assignment values $\hat{T}_i$ obtained within the hold-out or $K$-fold testing procedure as a guide to where the two distributions differ; in particular, for standard loss functions, $\hat{T}$ can be seen as estimating $\mathop{}\!\textnormal{P}(T=1|Y)$, as $\mathop{}\!\textnormal{P}(T=1|Y)$ is the prediction minimizing squared-error loss and maximizing the likelihood.
To obtain summary statistics of the difference in distributions expressed by these predicted values, we can calculate the implied treatment effect on any function of the outcome vector (including a given index), such as the outcomes vector itself:
While primarily an expression of the treatment effect expressed by the prediction function, it becomes an estimate of the average treatment effect vector $\tau =\mathop{}\!\textnormal{E}(Y|T=1) - \mathop{}\!\textnormal{E}(Y|T=0)$ provided that the prediction function is loss-consistent (where loss is assumed to be out-of-sample squared-error loss of the algorithm throughout):
Note that this result can be extended to the effect on any fixed function $g(Y)$ of the outcomes. Similar results can be achieved for expectations of functions $h(Y,\mathop{}\!\textnormal{P}(T=1|Y))$ that are estimated by the average of $h(Y,\hat{T})$.
In this section, we propose one approach that directly optimizes for a simple representation of the causal effect on the outcome distributions. Specifically, we consider a discretization of treatment and control distributions stemming from a partition of the outcomes space. In maximizing the expressiveness of this simple representation, we link its construction back to the reverse regression problem our tests are based on.
We assume for simplicity that the outcome vector $Y$ is continuously distributed (with overall density $f(y)$) both in treatment (density $f_1(y)$) and control (density $f_0(y)$), and that $Y$ takes values in $\mathcal{Y}$. Our goal is to find a partition
such that the discretized distributions
preserve as much information about the difference between treatment and control as possible. For example, we consider as a criterion for the normalized differences $\Delta(\ell) = \frac{\bar{f}_1(\ell) - \bar{f}_0(\ell)}{\bar{f}(\ell)}$ the variance
which we want to maximize to obtain a discretization that is as expressive as possible about the causal effect on the distribution of outcomes. \footnote{We present one approach, based on variance and leading to regression. Alternatively, expressing informatin about the distribution in terms of entropy/divergence could yield maximum-likelihood classification.}
The difference measure $\Delta$ connects the exercise of providing a representation of the difference between two distributions to the reverse regression of predicting $T$ from $Y$ since
for $p = \mathop{}\!\textnormal{P}(T=1), p(y) = \mathop{}\!\textnormal{P}(T=1|Y=y) = \mathop{}\!\textnormal{E}(T|Y=y)$. In particular, if we do not further restrict the partitions (other than in terms of number of parts ans possibly their size), optimal sets are obtained as levels sets of $p(y)$:
This claim characterizes an optimal partition in terms of level sets of the oracle predictor $p(y) = \mathop{}\!\textnormal{E}(T|Y=y)$. A natural empirical implementation leverages predictions $\hat{T} = \hat{f}(Y)$, where $\hat{f}$ is formed on a training dataset $S_T$ by predicting treatment $T$ from outcomes $Y$, yielding a partition
Here, the $\hat{c}_{\ell}$ could be appropriate quantiles of $\hat{T}$ obtained from the training dataset (e.g. by cross-validation). Specifically, we obtain a partition
of the units in the holdout, which allows for honest estimation Athey2016-zi of the difference between these groups. Specifically, we can test whether treatment and control counts across subsets provide significant evidence of a difference in distributions as represented by the partition, and obtain unbiased estimates of the mean differences of individual outcome variables between groups.
While level sets based on predicted treatment provide a solution to the problem of partitioning the outcome space to represent the causal effect on the distribution, these level sets may have complex shapes that may at best be represented graphically for relatively low-dimensional outcome vectors, but are in general hard to describe. Following the work of Athey2016-zi on heterogeneous treatment effects and Gagnon-Bartsch2019-ay on analyzing baseline imbalance, we also consider further restricting the partitions to recursively defined axis-aligned rectangles (i.e., decision trees), assuming that $\mathcal{Y} \subseteq \mathbb{R}^k$. Restricting to partitions that can be represented as the leaves of a decision tree that splits on outcome variables, the problem of finding an optimal partition is infeasible even for the in-sample analog. However, maximizing the variance in $\Delta$ is equivalent to minimizing the prediction error of predicting $T$ from its average within subsets:
As a consequence, we can apply standard regression trees that minimize mean-squared error in the prediction of $T$ from $Y$ to obtain a partition of $\mathcal{Y}$ into axis-aligned rectangles defined by the resulting decision tree. Running this prediction exercise in the training sample, we again obtain a partition of units in the hold-out (given by the leaves of the tree) that allows for honest estimation of conditional treatment probabilities as well as outcome mean differences between groups.
As an illustration for the basic prediction test and to calibrate power, we provide a simulation example with $k=2$ outcomes based on the data-generating process
with $\epsilon$ two-dimensional standard Normal, where $m$ (“move”) and $s$ (“stretch”) are both scalar parameters: $m$ moves the treated distribution towards the north-east (relative to the control distribution, which is centered at the origin) and $s$ stretches the variance of the treated distribution equally in all directions (relative to the control distribution, which has rotation-invariant unit variance). We assume that $\mathop{}\!\textnormal{P}(T=1) = .5$ throughout. In this setting, the null hypothesis of no effect (same distributions) corresponds to $m= 0 = s$. For different values of the parameters, we study the power of a test of this null hypothesis from samples of size $n=100$.
We compare two test for no overall effect of treatment:
As an illustration, (ref) presents the data and fitted ensemble predictions for one draw from this data-generating process with moderate move ($m=0.2$) and considerable stretch ($s=0.5$). For this draw, the predicted values $\hat{T}_i$ correctly locate the lower density of treatment in the lower-left quadrant and the higher density in the periphery ((ref)); the permutation test produces a $p$-value of $2\%$ ((ref)) compared to a $p$-value of $10\%$ for the Wald test in a SUR on the same data.
(ref) reports the rejection frequency for the Wald and prediction test at a nominal level $\alpha = 5\%$ based on 439 Monte Carlo draws. Under the null hypothesis ($m = 0 = s$), the rejection frequencies (size) are at or below the nominal level for both tests. As move increases, the power of the Wald test and the prediction test increase, with the Wald test attaining higher power if there is no stretch. The picture is quite different for varying values of the stretch parameter: In the framework of linear regression of $Y_1,Y_2$ on $T$, higher stretch only adds variance and thus decreases power; for the prediction test, on the other hand, differential variance between control and treatment provides evidence of an effect and thus increases the rejection rate.
The results in (ref) are restricted to a nominal level of $\alpha = 5\%$; (ref) extends the analysis to the full empirical distribution of $p$-values (where the results in (ref) can be obtained by evaluating each at $5\%$). Under the null hypothesis (zero stretch, zero move), the $p$-values are approximately uniform within the unit interval, representing performance close to nominal. The effect of move and stretch are as below, with move improving power of both tests, while stretch induces an increasing gap between the power of prediction and linear Wald tests represented by the wedge between the empirical cumulative distribution curves.
We have presented a specific methodology that allows for leveraging powerful machine-learning predictors to test for the effect of a randomized intervention on a group of outcome variables. We note that our methodology can be adopted for outcome tests that include control variables (by including them in the prediction exercise), cluster randomization (by splitting and permuting data by clusters), stratified/conditional (within-site) randomization (by splitting and permuting within strata/sites), and missing data (e.g. by testing whether the outcomes predict treatment assignment better than missingness information alone).