Externally Valid Selection of Experimental Sites via the k-Median Problem111We would like to thank Isaiah Andrews, Tim Armstrong, Michèle Belot, Michael Gechter, Jesse Goodman, Kei Hirano, Yiqi Liu, Francesca Molinari, Charles Manski, Guillaume Pouliot, David Shmoys, and Davide Viviano for helpful feedback, comments, and suggestions. We would also like to thank Eddie Ramirez Saquic for excellent research assistance. We gratefully acknowledge financial support from the NSF under grant SES-2315600.
Abstract
We present a decision-theoretic justification for viewing the question of how to best choose where to experiment in order to optimize external validity as a -median (clustering) problem, a popular problem in computer science and operations research. We present conditions under which minimizing the worst-case, welfare-based regret among all nonrandom schemes that select sites to experiment is approximately equal—and sometimes exactly equal—to finding the most central vectors of baseline site-level covariates. The -median problem can be formulated as a linear integer program. Two empirical applications illustrate the theoretical and computational benefits of the suggested procedure.
1 Introduction
A common concern in randomized evaluations of new policies is their external validity; that is, whether the estimated effects of a policy intervention carry over to new samples or populations (Vivalt, 2020; Gechter, 2024; Duflo et al., 2007, Chapter 8). There is a large body of literature arguing that the external validity of randomized evaluations can be improved by explicitly incorporating this goal into the experimental design by, for example, carefully deciding where to experiment; see Degtiar and Rose (2023) for an overview and references, and also the recent work of Chassang and Kapon (2022). For instance, if a researcher has access to multiple sites to experimentally evaluate a new policy, it is possible to use the information available before the evaluation (such as site-level characteristics) to nonrandomly select one site—or more—that are “representative” of the populations of interest. Recently, both Egami and Lee (2024) and Gechter et al. (2024) have shown that a research design that nonrandomly selects experimental sites—referred to broadly in the literature as purposive sampling (Cook et al., 2002, p. 511)—is useful to improve the external validity of randomized evaluations.
This paper presents a theoretical justification for viewing the question of how to best design a purposive sampling scheme—or, equivalently, how to best nonrandomly select sites to experiment—with the goal of optimizing external validity as a -median problem. This is a classical problem in computer science and operations research (Williamson and Shmoys, 2011; Cohen-Addad et al., 2022). Broadly speaking, in a -median problem, there is a set of facilities and a set of clients; the goal is to open at most facilities and connect each client to at least one facility at minimal total connection cost. We next provide more specific details on how these problems relate.
Statistical Decision Theory: Under conditions that will be explained clearly in Sections 2 and 3, Theorem 1 in this paper shows that the worst-case (welfare-based) regret of any purposive sampling scheme that selects at most sites to experiment is approximately equal—and sometimes exactly equal—to the objective function of a -median problem with the following features. First, the sites available for experimentation are treated as facilities. Second, the sites where the policy maker would like to implement the new policy are treated as clients. Third, the connection cost between clients and facilities is proportional to the Euclidean distance between the available site-level covariates. Importantly, we show that the solution of the -median problem is exactly “minimax-regret” optimal (among all purposive sampling schemes) when i) the candidate sites for experimentation and the policy-relevant sites are disjoint (Remark 4) and ii) the treatment effect heterogeneity across sites accommodated in the parameter space is substantial (in a sense Theorem 1 makes precise). When ii) holds but i) does not, we show that the optimized value of the -median problem only approximates the minimax-regret value, and that the approximation error improves as either the number of policy-relevant sites increases or the experiments conducted become more precise (Remark 5). The link with the -median problem established in this paper thus shows that selecting the sites that have the most central vector of covariates tends to optimize external validity (in a minimax-regret sense).
In order to formalize this connection, we leverage recent developments in the literature of treatment choice problems with partial identification; for example Yata (2021); Ishihara and Kitagawa (2021); Montiel Olea et al. (2023). Note first that in the -median problem, each client is typically connected to only one facility (otherwise, the connection cost would not be minimized). In the context of the site selection problem, a connection between a site (where an experiment was conducted) and a site (where no experimentation occurred) means that the estimated effects obtained in site are used to inform policy decisions in site . This means that even when experimental outcomes in sites are available, the solution to the -median problem would typically prescribe the policy maker to only use the information from the site with the smallest connection cost: the nearest neighbor. But when is it decision-theoretically optimal for a policy maker to behave in this way? After sites have been selected for experimentation, the policy maker faces the “evidence aggregation” problem introduced in Ishihara and Kitagawa (2021) and recently discussed in Yata (2021), Christensen et al. (2022) and Montiel Olea et al. (2023). That is, the policy maker must decide how to use the estimated treatment effects in different—and potentially heterogeneous—sites to make a treatment choice (implement the new policy or preserve the status quo) in the site of interest. It is known that there are conditions under which it is minimax-regret optimal for the policy maker to base decisions on the nearest neighbor (the site with the most similar site-level characteristics), provided the true treatment effects are allowed to vary substantially as a function of site-level covariates; see, for example, Proposition 1 in Montiel Olea et al. (2023). Moreover, in this case, the optimized worst-case regret is proportional to the distance between the baseline covariates of the site of interest and those of its nearest neighbor. Thus, these results on treatment choice problems with partial identification are fundamental to provide a decision-theoretic justification for the use of the -median (clustering) problem to optimize external validity.
Algorithms: The connection with the -median problem clarifies the problem’s difficulty but also suggests efficient algorithms. To see the need for those, recall that any purposive sampling scheme optimizes over “ choose ” potential site combinations, where is the total number of sites available for experimentation. Optimization over purposive sampling schemes also requires the evaluation of some measure of performance that depends on the dimension of the site-level covariates, . Thus, optimally choosing a purposive sampling scheme by simply evaluating the performance of each combination is costly when “ choose ” or is large.
Conceptually, the connection with the -median problem allows us to understand the computational complexity of finding a minimax-regret optimal purposive sampling scheme under the conditions of Theorem 1. Since the -median problem is known to be NP-hard (Kariv and Hakimi, 1979; Megiddo and Supowit, 1984; Cohen-Addad et al., 2018), there is no algorithm whose computational time scales polynomially in the problem’s inputs .
However, from a practical perspective, it is also known that the -median problem admits a linear integer program formulation (Williamson and Shmoys, 2011, Chapter 7.7, p. 185), and is routinely solved with off-the-shelf algorithms such as different versions of the branch-and-bound method (Bertsimas and Weismantel, 2005, Chapter 11.1). In addition, different branch-and-bound algorithms either find a solution with provable optimality or, if stopped early, generate a report on the suboptimality of the solution found (Bertsimas et al., 2016). As we explain later, in our application with , , any problem for can be solved to provable optimality in just a few seconds using a personal laptop (see Figure 5 in Section 5).
Related Literature: Our results build on two recent papers that presented novel purposive sampling strategies to select experimental sites so as to optimize external validity. Gechter et al. (2024) present an elegant decision-theoretic approach that frames external validity as a policy problem and—under the assumption that the policy maker has a priori information about the effects of the new policy across sites—recommend a Bayesian approach for choosing where to experiment. Egami and Lee (2024) use the principle behind synthetic control (Abadie et al., 2010) to recommend what they call the synthetic purposive sampling of sites; specifically, they select good “donors,” defined as sites whose weighted average of observed baseline characteristics are close to the characteristics of the sites of interest. It is important to note that our decision-theoretic framework leans heavily on Gechter et al. (2024), but also that the sites selected through the -median problem can be interpreted as a degenerate synthetic purposive sampling strategy, whereby each unit’s associated synthetic unit is just its nearest neighbor.
This paper also contributes to the literature arguing that a “modern, decision-theoretic framework can help clarify important practical questions of experimental design” (Banerjee et al., 2017). Although decision-theoretic approaches to external validity are recent, a large body of work used statistical decision theory to analyze other aspects of experimental design, such as sample size determination (Raiffa and Schlaifer, 1961; Manski and Tetenov, 2016, 2019; Azevedo et al., 2020, 2023; Hu et al., 2024). Finally, our notion of external validity is conceptually related to several areas of research in econometrics, machine learning, and statistics such as domain adaptation (Mansour et al., 2009; Ben-David et al., 2010), distributional shifts (Duchi and Namkoong, 2021; Sugiyama et al., 2007; Adjaho and Christensen, 2022), learning under biased sampling (Sahoo et al., 2022), and cross-domain transfer estimation and performance (Andrews et al., 2022; Menzel, 2023). To the best of our knowledge, none of these papers contain decision-theoretic analyses of where to experiment.
Outline: This paper is organized as follows. Section 2 introduces the formal framework. Section 3 presents the main result of this paper (Theorem 1) and explains how the -median problem arises naturally when trying to incorporate external validity concerns into the decision problem of where to experiment. Section 4 presents the linear integer program formulation of the -median problem. Section 5 presents two illustrative empirical applications. Section 6 considers extensions of the baseline model. Section 7 concludes. Proofs of the main results can be found in Appendix A. Additional results are collected in the Supplementary Appendix.
2 Setting up the Decision Problem
2.1 Notation
A policy maker considers a set of candidate sites to evaluate and, eventually, implement a new policy of interest. The sites are indexed by . In order to accommodate situations in which the policy maker is not necessarily able to experiment in all the candidate sites, we assume there is a nonempty subset of what we term experimental sites.222As discussed in Allcott (2015), there are often systematic reasons determining the eligibility of certain sites for experimentation. For example, in microfinance RCT studies, experiments often require large sample sizes and well-managed microfinance institutions (MFIs), characteristics more commonly found in older and larger institutions. To qualify for clinical trials involving a new surgical procedure, hospitals and surgeons need to have both experience in the procedure and a history of low mortality rates. Throughout the paper, and to avoid a trivial instance of the site selection problem, we assume that there are at least two experimental sites (i.e., ). It is also possible that institutional restrictions preclude the eventual implementation of the policy of interest in all of the candidate sites. Thus, it will be convenient to denote by the nonempty set of policy or policy-relevant sites.
In principle—and in order to accommodate different scenarios that could arise in empirical work—we allow for the case in which there is overlap between experimental and policy sites (i.e., ). We note, however, that the case in which these two sets are disjoint (i.e., ) will be of particular interest. This is because when there is no overlap between experimental and policy sites, the extrapolation problem faced by the policy maker becomes evident: a policy decision needs to be made in sites where there has never been a previous experimental evaluation of the policy of interest.
For each site, the policy maker observes a vector of site characteristics that may affect the treatment effect. Thus, we allow for treatment effect heterogeneity across sites, but we restrict this heterogeneity by assuming that it depends on observable characteristics. Specifically, let the function define conditional (on ) average treatment effects. We posit that any pair of sites with “similar” observed characteristics also have “similar” treatment effects, formally by assuming that is a Lipschitz function (with respect to the Euclidean norm) with known constant . We formally state this assumption in Section 2.4. It is not innocuous, but we will argue that it can be replaced by other continuity-like conditions (such as Hölder continuity) while maintaining the main message of the paper.333In Section 6.3, we discuss how our results extend to other functional classes that could be used to restrict treatment effect heterogeneity, such as collections of functions that belong to a convex and centrosymmetric space. Such restrictions were recently used in the econometrics literature to analyze estimation, inference, and other decision problems that arise in a nonparametric regression setup (Yata, 2021; Armstrong and Kolesár, 2018). We let denote the space of all Lipschitz functions from to with constant .
2.2 Statistical Model for the Site Selection Problem
As in Gechter et al. (2024), the policy maker must choose a strict subset of experimental sites .444We require to be a strict subset of because if we allow the policy maker to experiment in all sites, and there is no cost of experimentation that varies at the site level, then there is no site selection problem. We consider the case in which is allowed to equal in Section 6.1. In this paper, we focus on the case in which there is a restriction on the total number of experimental sites that the policy maker can select. That is, there is an integer , , such that must belong to the set
(1) |
What we have in mind with this restriction is that the policy maker may not have enough administrative resources to run more than experiments to evaluate the policy of interest. Our notation also allows for the possibility that the policy maker does not want to experiment at all.
If the policy maker decides to experiment in a nonempty set of cardinality , then she will observe treatment effect estimates. We collect these estimates in a vector of dimension . In a slight abuse of notation, let denote the indices of the experimental sites in . Letting denote the estimated treatment effect in site , we can define the vector
(2) |
Analogously, we can denote the vector of true treatment effects for the experimental sites in as
(3) |
We assume that the treatment effect estimators obtained in each site are normally (and independently) distributed around the vector of true effects:
(4) |
Following Gechter et al. (2024), treat as known.
The normality assumption in (4) is unlikely to hold exactly; however, it is common to assume that treatment effect estimates from randomized controlled trials are asymptotically normal with asymptotic variances that can be estimated consistently. Treating the limiting normal model as an exact finite-sample statistical model then eases exposition and allows us to focus on the core features of the site selection problem. Indeed, working directly with such a limiting model is common in applications of statistical decision theory to econometrics; see Müller (2011) and the references therein for theoretical support and applications in the context of testing problems and Ishihara and Kitagawa (2021), Stoye (2012), or Tetenov (2012) for precedents in closely related work. Gechter et al. (2024) use the same statistical model, but our parameter space has a more specific form as treatment effects are controlled by the Lipschitz function .
After observing , the policy maker chooses an action at each policy-relevant site . We interpret this action as the proportion of a population in the site that will be randomly assigned to the new policy. Thus, means that everyone in site is exposed to the new policy, and means that the status quo at the site is preserved. Under this interpretation, means that 50% of the population at site will be exposed at random to the new policy; however, the formal development equally applies to either individual or population-level randomization. Our interpretation abstracts from integer issues arising with small populations.
Thus, we can define a treatment rule as a (measurable) function that maps experimental outcomes to (possibly) randomized policy actions in each of the policy-relevant sites. It will be sometimes convenient to use to denote the specific treatment rule for site implied by and to denote the set of all treatment rules. Note that we index the treatment rules by the selected experimental sites, , to be explicit about the fact that the data used to inform policy will vary depending on the choice of . We call nonrandomized if for every we have for (Lebesgue) almost every . Otherwise, we say that the rule is randomized. We will focus on decision rules that belong to the set
These are treatment rules in which the ex-ante probability of implementing the policy is whenever the true treatment effects at the sites experimented on are zero. For the moment, and for the sake of exposition, we assume that there is no cost of experimentation. While this assumption is clearly unrealistic, we later show that the main conclusions of our analysis are robust to adding fixed costs to the objective function of the -median problem.
2.3 Welfare and Regret
Let . We assume that the welfare of a decision rule , given that sites are selected for experimentation, corresponds to the average expected welfare across policy-relevant sites:
(5) |
where means that the expectation is taken assuming , and is the true treatment effect at site .
The regret of policy can then be shown to equal
(6) |
Our focus will be on finding the purposive sampling scheme that minimizes worst-case regret.
Definition 1 (MMR optimal purposive sampling scheme and treatment rule).
The pair is minimax-regret (MMR) optimal among all purposive sampling schemes and treatment rules if
(7) |
Remark 1.
In the standard definition of minimax-regret optimality, it is common to allow the decision maker to select randomized decision rules. Definition 1 implies an asymmetric treatment of randomization: While we allow the policy maker to randomize policy implementation choices, we are restricting her to pick the experimental sites in a deterministic fashion. The nonrandom selection of experimental sites—referred to broadly in the literature as purposive sampling (Cook et al., 2002, p. 511)—is common in practice; see the discussion about purposive sampling in Egami and Lee (2024). In Section 6, we discuss the challenges we encountered in trying to allow for the random selection of experimental sites. ∎
Remark 2.
It will sometimes be convenient to rewrite the right-hand side of (7) as
This suggests that, conceptually, the solutions of the minimax-regret problem can be done in two steps. First, one can analyze the problem of policy implementation given the experimental outcomes at sites . Then, one can optimize over the sites where to experiment.
This distinction is helpful because the first step is related to the “evidence aggregation” problem analyzed in Ishihara and Kitagawa (2021). As mentioned in the Introduction, we can leverage recent results in Yata (2021) and Montiel Olea et al. (2023) to better understand the minimax-regret solutions to this problem. A key difference from Ishihara and Kitagawa (2021) is that we typically have more than one policy-relevant site. ∎
2.4 Main Assumptions
We make the following assumptions. To begin, treatment effects vary continuously with the site-level covariates:
Assumption 1.
is a Lipschitz function (with respect to the Euclidean norm) with known constant . That is, for any , where denotes the Euclidean norm.
We also impose a regularity condition on site-level covariates. That is, we assume that all observed covariates are different:
Assumption 2.
.
Even if this were not the case in raw data, one would presumably want to induce it by adding site fixed effects.555In section 6.3, we discuss how Assumption 1 can be modified to accommodate other general distance measures, as well as some forms of unobserved treatment heterogeneity (for which Assumption 2 may be dropped).
Finally, for each policy site and every , denote by its nearest neighbor in (or the nearest neighbor with the smallest index in case of multiplicity). That is, for every :
3 Main Result
This section presents our main result: The problem of finding the purposive sampling scheme that minimizes the worst-case regret is approximately equal—and sometimes equal—to solving a -median problem.
In order to derive this, first recall the definition of the -median problem.
Definition 2 (-median problem).
We say that a purposive sampling scheme, , solves the -median problem if it solves
(8) |
In Definition 2, the policy-relevant sites (with indexes in ) are clients and the experimental sites (with indexes in ) are facilities. The connection cost between a facility and a client is the Euclidean distance . Since the goal of the -median problem is to choose the facilities that minimize connection cost, each client gets connected to the facility that is closest among those in . Hence, the term appears in Definition 2. Finally, Equation (8) can be also written as
(9) |
where the equality follows from the fact that for each facility that is also a client , connection cost becomes zero.
There is no scope for interesting analysis, or indeed for justification of experiments, unless noise in the signal is bounded. Therefore, define . Let denote the largest standard deviation among the potential experimental sites. Our main theorem is the following:
Remark 3.
The left-hand expression in (10), i.e., , is the worst-case regret of the purposive sampling scheme assuming optimality of subsequent treatment choice . Theorem 1 thus shows that the (optimized) worst-case regret of any purposive sampling scheme can be uniformly approximated (after scaling by the factor )) by the objective function of the k-median problem in Definition 2—which is equivalent to the objective function of the problem in the right-hand side of Equation (9). ∎
Remark 4.
When the candidate sites for experimentation and the policy-relevant sites are disjoint, then for any :
(12) |
Consequently, any solution of the -median problem is a fortiori an exact minimax-regret solution for the site selection problem. This follows from the fact that if a purposive sampling scheme, , minimizes the right-hand side of Equation (12), then it will automatically minimize the left-hand side, which, by Remark 2, defines a minimax-regret optimal purposive sampling scheme. ∎
Remark 5.
Theorem 1 implies that even in the case where ,
(13) |
is at most equal to (11). Thus, the quality of this approximation improves as either the number of policy-relevant sites increases or the potential experiments conducted become more precise ( becomes smaller). The approximation deteriorates as increases. This happens because in approximating the worst-case regret of a purposive sampling scheme, we ignored the regret associated with sites selected for experimentation; we further discuss this below when we explain Lemma 1. This means that, as increases, more sites will be ignored in our approximation and the upper bound will become looser. ∎
Remark 6.
The tight connection to the -median problem relies on the Lipschitz constant to be large enough. In other words, the -median problem only has a decision-theoretic justification when the treatment effect heterogeneity across sites accommodated in the parameter space is substantial. ∎
The proof of Theorem 1 follows from the two Lemmas below, which bound the site selection problem’s minimax-regret value from above and below, respectively.
Lemma 1 (Upper bound).
Proof.
See Appendix A.1. ∎
Remark 7.
Lemma 1 implies that the minimax-regret value of the site selection problem in Definition 1 can be upper bounded by the solution of the uncapacitated -facility location problem.666Uncapacitated means that there is no capacity constraint on the number of clients that each facility can accommodate. See (Williamson and Shmoys, 2011, Chapter 4.5) and Zhang (2007). Just as in the -median problem, there is a set of facilities and a set of clients (or cities) . Now we assume that there is an opening cost associated with each facility . The connection cost between facilities and clients is as before. The goal is to open at most facilities and connect each client to one facility so that total cost is minimized. Thus, the problem
(15) |
can be viewed as a -facility location problem. Just as before, the set of facilities is and the set of clients is . The connection cost between sites and is . The opening cost for any facility is . Minimizing the upper bound in Lemma 1 is thus equivalent to solving the -facility location problem. As shown in Appendix A.1, the upper bound arises by bounding the worst-case sum of regrets across sites by the corresponding sum of worst-case regrets. For selected sites that are both facilities and clients, the worst-case regret is obtained by solving a point-identified treatment choice problem. For policy-relevant sites where no experiment was conducted, it is obtained by solving the partially-identified treatment choice problem in Ishihara and Kitagawa (2021). When and are disjoint, the first component of the upper bound vanishes, and the bound becomes proportional to the solution of the -median problem. ∎
Proof.
See Appendix A.2. ∎
Remark 8.
Lemma 2 implies that the minimax-regret value of the site selection problem in Definition 1 can be lower bounded by the solution of the k-median problem:
(17) |
Theorem 1 thus follows by noting that the upper and lower bound match up to the opening costs of the facilities. As noted before, the lower and upper bound match when . ∎
Remark 9.
While the computation of an exact minimax-regret sampling scheme (whether purposive or randomized) appears intractable, our ability to provide nontrivial upper bounds on worst-case regret does not hinge on being large enough. Indeed, one could always compute such a bound by (i) forcing the treatment decision to only depend on site and (ii) computing worst-case expected regret separately across sites, ignoring that the regret-maximizing parameter configurations may be mutually inconsistent across sites. Both manipulations increase regret and therefore define an upper bound. This bound is easy to compute for any given sampling scheme because the induced minimax-regret treatment choice problem was solved in Stoye (2012). For large enough, it will coincide with the bound in Theorem 1. The caveat is that this bound is nonlinear; hence, while it is easy to compute once, it does not generate a tight connection to the -median problem and will be hard to optimize for large instances of the site selection problem. ∎
4 Integer Programming and the -median Problem
It is well known (Williamson and Shmoys, 2011, Chapter 7.7, p. 185) that the -median problem in (17) can be formulated as the following linear integer program:
(18) | ||||
such that | (19) | |||
(20) | ||||
(21) | ||||
(22) |
In this formulation, the choice variables are the -valued variables and , for . The variable indicates whether site is chosen for experimentation, and indicates whether experimental site is the nearest neighbor of policy site . The total number of sites chosen for experimentation is restricted to be no greater than , and site can only be the nearest neighbor of policy site if site is chosen for experimentation. The connection cost between a facility and a client —denoted —is given by .777The connection cost in the objective function of the integer program differs from the connection cost in the -median problem in (17) by a constant factor , which does not affect the solution of the -median problem. Solving the linear integer program described above is equivalent to solving (17). It is worth highlighting that the program above is linear in the choice variables and .
A major advantage of the integer programming formulation of the -median problem is that most ecosystems for scientific computing offer different solvers for linear and nonlinear integer programs. For the applications in this paper, we use the MIP solver in Gurobi (Gurobi Optimization, LLC, 2023) through the Python package gurobipy. The Gurobi software is highly optimized, especially for large-scale problems, and it offers an academic license. Even though the scale of the applications presented in Section 5 is not large enough for the efficiency advantages of Gurobi to become salient over other solvers, we wanted to showcase its ease of use. It also integrates seamlessly with Google Colab, providing us a way to build self-contained and reproducible examples.888https://colab.research.google.com/
The MIP solver uses a version of the branch-and-bound algorithm; see Bertsimas and Weismantel (2005), Chapter 11.1, for a general description of this algorithm. Broadly speaking, this algorithm works by first finding the solution to the linear programming (LP) relaxation of the original integer problem. This is known as the relaxation step. Then, the problem is split into two sub-problems (branching), according to the integer constraints on the solution. This method then gives bounds on the integer solution, and the algorithm is applied recursively until the lower bound and the upper bound converge up to a tolerance parameter. The recursion creates nodes, and there are some strategies to determine which nodes should be explored further; for example, nodes that have integer solutions do not require any more branching.
Gurobi implements several additional steps that help the branch-and-bound algorithm be more efficient.999See https://www.gurobi.com/resources/mixed-integer-programming-mip-a-primer-on-the-basics/. The presolve step reduces the number of effective constraints of the problem by checking if the integer requirement can eliminate some of them. As the name suggests, it is performed before the start of the branch-and-bound algorithm. Cutting planes tightens the feasible region by adding linear inequalities to eliminate fractional solutions; it is performed during the branch-and-bound process, and for this reason the algorithm used by Gurobi can be referred to as a version of a “branch-and-cut” algorithm. Finally, the MIP solver implements several heuristics, for example by rounding the component of a solution that is closest to an integer, fixing it, and hoping the other components will converge to integers quicker.
In principle, one can always solve the -median problem in (17) by enumerating all possible size subsets of . Such an algorithm runs in time proportional to
time and therefore scales poorly when the term “ choose ” is large. However, we are able to evaluate the performance of our preferred solver by brute-force solving smaller but not trivial instances of the problem, notably the entire application in Section 5.2.
Figure 9 in Appendix B.1 shows an example of the output obtained after using the MIP solver in Gurobi to solve the linear integer program for the application in Section 5.2. In this example, the scale of the problem is given by , and . We defer the details of the application to Section 5.2.
While we do not use it in this paper, we finally note that there is a large literature studying efficient (polynomial-time) approximation algorithms for the -median problem, going back to seminal work of Charikar et al. (2002). A basic idea in these algorithms is to consider a linear programming relaxation of the integer program associated with the -median problem (Williamson and Shmoys, 2011, Chapter 7.7). Even though the scale of the problems analyzed in this paper does not require the implementation of any of these algorithms, there are several papers that present theoretical performance guarantees for them; see for example Cohen-Addad et al. (2022) and also the references in Cohen-Addad et al. (2018). Finally, it is worth mentioning that when is fixed (and not viewed as part of the problem’s input), there is an approximate algorithm that runs in time proportional to ; see Kumar et al. (2010). Such an algorithm could be useful when and are large and is small.
5 Applications
5.1 Mobile Financial Services in Bangladesh
Lee et al. (2021) conducted a randomized controlled trial in Bangladesh to estimate the effects of encouraging rural households to receive money transfers from migrant family members. They specifically conducted an encouragement design where poor rural households with family members who had migrated to a larger urban destination receive a 30–45 minute training about how to register and use the mobile banking service “bKash” to send instant remittances back home.
The experiment was conducted in the Gaibandha district, one of Bangladesh’s poorest regions. It focused on households that had migrant workers in the Dhaka district, the administrative unit in which the capital of Bangladesh is located. Lee et al. (2021) measure several outcomes of both receiving households and sender migrants; see their Figures 3 and 4. To give a concrete example of the measured outcomes, one question of interest is whether families that adopt the mobile banking technology are more (or less) likely to declare that the monga—the seasonal period of hunger in September through November—was not a problem for their household. Table 9, Column 7, p. 61 in Lee et al. (2021) presents results for this specific variable showing that households that used a bKash account in the treatment group are 9.2 percentage points more likely to declare that monga was not a problem.
We ask: Do the findings in Lee et al. (2021) generalize to other migration corridors, i.e., combinations of origin and destination districts, in Bangladesh? Is the corridor selected by Lee et al. (2021) a good choice for a researcher who is concerned about external validity? Following Gechter et al. (2024), we name the corridors using a destination-origin format; for example, the migration corridor studied in Lee et al. (2021) is “Dhaka-Gaibandha”. Figure 1 displays this corridor along with other common ones. The migration corridors analyzed in Gechter et al. (2024) are depicted as dotted lines connecting an origin and a destination.101010We thank Michael Gechter for gracefully sharing part of the dataset used in Gechter et al. (2024).
![Refer to caption](extracted/5797333/plots/bangladesh_migration_corridors_basic_with_lines_and_original_site.png)
Notes: The map of Bangladesh with the origins of the migration corridors marked as light blue dots and the destinations marked as dark blue stars. Following the terminology used in Gechter et al. (2024), origin refers to a worker’s home, and destination refers to where the worker migrates for work. The corridor where the experiment was originally implemented in Lee et al. (2021), Dhaka-Gaibandha, is highlighted in yellow.
In Lee et al.’s 2021 words, “[t]he particular nature of our sample potentially limits the external validity” of the analysis. In short, migration corridors differ in characteristics ranging from the distance between the origin and the destination to the cost of living and the average wages. Figure 2 presents a box plot of (standardized) characteristics that Gechter et al. (2024) collected for each of the migration corridors. We take these corridors to be our potential experimental and policy-relevant sites. That is, , and card() = card() = 41. Below we present results for .
Notes: For each covariate, the box represents the interquartile range (IQR), the vertical black line represents the median, and the horizontal line shows the “theoretical minimum” (defined as IQR) and “theoretical maximum” (defined as IQR). Black dots are outliers, defined as observations that fall beyond the theoretical minimum and maximum. Each panel depicts the sites selected by the different approaches when .
Selected Site when : When , the selected site based on the -median problem is Dhaka-Pabna. This is also the solution obtained by using the synthetic purposive sampling approach (henceforth, SPS) in Egami and Lee (2024).
Figure 2(a) presents a visual comparison of the covariates of Dhaka-Pabna (blue circles) and Dhaka-Gaibandha (pink diamonds), the original site selected by Lee et al. (2021). The covariate values for Dhaka-Gaibandha are slightly outside the interquartile range for at least three covariates: migrant density, mean remittance, and mean household size. In comparison, all but one covariate value for Dhaka-Pabna are within the interquartile range. The figure also shows that two key covariates of Dhaka-Gaibandha are right at the edges of the interquartile range: distance between sites in the corridor (3rd quartile) and mean household income (1st quartile). One might conjecture that the effects of adopting a mobile banking technology to transfer money particularly depend on distance and household incomes, suggesting that the Dhaka-Gaibandha corridor may not be the most representative.111111Gechter et al. (2024) suggest that these qualities could explain the large treatment effects found by Lee et al. (2021). Interestingly, Dhaka-Pabna has opposite features: the distance between destination and origin in this corridor is short (1st quartile) and households are relatively better off in terms of income (3rd quartile). The use of the minimax criterion might explain why a corridor with these characteristics may be a good choice for extrapolating experimental results.
Figure 2(b) also presents the covariates of Dhaka-Noakhali (yellow triangles), the migration corridor selected by the Bayesian approach of Gechter et al. (2024).121212By construction, the solution of Gechter et al. (2024) depends on the choice of prior. The results herein reported are based on their preferred prior specification; see Section 5.3 p.p.23 in Gechter et al. (2024). We note that for 10 out of the 13 variables that control treatment effect heterogeneity, Dhaka-Noakhali has covariates that are typically outside the interquartile range.
Selected Sites when : When , the -median solution is to pick Dhaka-Pabna and Dhaka-Pirojpur. The former also solved the -median problem for .
Figure 3(a) presents the covariates of Dhaka-Pabna (filled, blue circle) and Dhaka-Pirojpur (hollow, blue circle). While four covariate values of Dhaka-Pirojpur (standard deviation of male migrant wage, standard deviation of household income, price index, and migrant density) are outside the interquartile range, they still appear more central than Dhaka-Gaibandha and Dhaka-Noakhali. However, relative to Dhaka-Pabna, the solution for adds a considerably less central site; Figure 4 illustrates that this site is a good nearest neighbor for some sites that would not otherwise be well matched.
Figure 3(b) presents the solutions of Egami and Lee (2024) (squares) and Gechter et al. (2024) (triangles).131313Gechter et al. (2024) impose an additional constraint on purposive sampling schemes: they require the two migration corridors selected for experimentation to have origins in different divisions. We note that both the solutions of Egami and Lee (2024) and the -median problem satisfy this constraint as well. The solution of the -median problem and synthetic purposive sampling are no longer the same. We also note that the two sites selected by synthetic purposive sampling differ from the site selected by this procedure when .
Notes: Box plots for the distribution of covariates among migration corridors, constructed as explained in Figure 2. Each panel depicts the site selected by the different approaches when .
![Refer to caption](extracted/5797333/plots/distance_hh_kmedian.png)
![Refer to caption](extracted/5797333/plots/distance_hh_sps.png)
Notes: Each point represents a destination-origin migration corridor (site) on the two-dimensional covariate plane, using two covariates: distance between the two ends of each corridor and average household income in the home district (where the average is taken over households with a reported migrant). The solutions of each method are indicated by the shapes of the solid dots. In Panel 4(a), blue circles denote policy sites with Dhaka-Pabna as the nearest neighbor, and orange crosses represent those with Dhaka-Pirojpur as the nearest neighbor. Similarly, in Panel 4(b), blue circles and orange crosses follow the same coding, indicating that these policy sites rely on a single experimental site for constructing their synthetic control. Additionally, gray cross-circles represent policy sites that use both experimental sites for their synthetic control.
Figure 4 presents a simple visualization of optimal connections between sites under different optimality criteria. Note that the underlying scatter plots in both panels are the same; they visualize the location of corridors in (distance, household income)-space. However, Panel 4(a) indicates which sites were selected through solving the -median problem and which of these sites any other site was matched to, where blue circles represent matches with Dhaka-Pabna and orange crosses represent matches with Dhaka-Pirojpur. Figure 4(a) also presents the sites selected by Gechter et al. (2024), i.e. Dhaka-Magura and Dhaka-Noakhali, both marked with dark blue triangles. They appear close to the sites selected by the -median problem, but in terms of that problem’s criterion function, they only rank among candidate solutions and miss the problem’s optimal value by . Panel 4(b) similarly visualizes the sites selected by Egami and Lee’s (2024) synthetic purposive sampling approach, i.e. Dhaka-Gopalganj (orange square) and Dhaka-Barguna (blue square).141414We generated this figure by using Egami and Lee’s (2024) code on Gechter et al.’s (2024) data. We also visualize how policy sites are matched with experimental sites: Orange crosses correspond to migration corridors that assign all of their weight to Dhaka-Gopalganj; blue circles correspond to migration corridors that assign all of their weight to Dhaka-Barguna; all other sites (gray crossed circles) assign strictly positive weights to both donor sites. This illustrates that these sampling schemes meaningfully differ. This would even be true if the synthetic purposive sampling approach were implemented but forcing degenerate (single donor) matches, because Egami and Lee’s (2024) approach would then reduce to a -mean problem, i.e. using squared Euclidean distance as connection cost.
Computational Costs: The above examples are small enough so that the -median problem could easily be solved by brute-force enumeration of and candidate solutions, respectively. Needless to say, such an approach would not scale—for example, in this same application, induces candidate solutions.
Indeed, Figure 5(a) compares time to solve the -median problem for using a) the integer program formulation of the -median problem in Section 4 and b) brute-force enumeration. The MIP solver in Gurobi solves all instances of the problem to provable optimality in less than one second each. In contrast, brute-force enumeration takes approximately hours for .151515This was run in a Windows XPS with cores and GB of RAM using R. The code is parallelized, and to avoid memory issues when , it runs a C++ function in the background to evaluate one combination at a time. This ensures that we are giving brute-force enumeration the best chance of success.
One potential benefit of brute-force enumeration is that one can check for multiple solutions, which actually occurred at . In Gurobi, an ad hoc search could be conducted by modifying the random number seed or using the concurrent optimizer, but discovery would not be guaranteed.
Figure 5(b) reports the time needed to implement the synthetic purposive sampling approach of Egami and Lee (2024). This was done by using the spsR package with the option to use the Gurobi solver on the background. We consider it to be fast, taking less than seconds for . However, the synthetic purposive sampling problem can be formulated as a quadratic mixed integer program, while -median is linear. This is evident from the graph; the -median problem is solved almost instantly for every up to .
![Refer to caption](extracted/5797333/plots/gurobi_vs_brute_force_time.png)
![Refer to caption](extracted/5797333/plots/gurobi_vs_SPS_time.png)
Notes: Time comparison of different purposive sampling approaches. Time (vertical axis) is in minutes. The dark, blue line with circles represents the time it takes to solve the integer program in Section 4 using the MIP solver in Gurobi to provable optimality (Gurobi gives the solution in less than a second). The light, blue line with squares in Panel a) represents the time it takes to solve the -median problem using brute-force enumeration. The red line with triangles in Panel b) represents the time needed to implement the synthetic purposive sampling approach using the spsR package.
5.2 Multi-Country Survey Experiments in Europe
Our second application revisits a multi-country survey experiment originally conducted and analyzed in Naumann et al. (2018) and discussed in Egami and Lee (2024). The question of interest is whether native-born inhabitants of a particular country are more supportive of immigration depending on whether the potential migrants are high-skilled or low-skilled. Naumann et al. (2018) carried out a survey experiment in European countries: Austria, Belgium, Czech Republic, Denmark, Finland, France, Germany, Ireland, Netherlands, Norway, Slovenia, Spain, Sweden, Switzerland, and the United Kingdom. Respondents were native-born individuals and were randomly assigned to report their attitudes towards either high-skilled (“treatment”) or low-skilled (“control”) immigrants.
While the experiments have already been conducted and outcomes of each experiment are available for all countries, we consider the situation of a researcher that can only conduct experiments; here the choice of mirrors Egami and Lee (2024). We let all countries be both potential experimental and policy-relevant sites. That is, , and card() = card() = 15. 161616A potential situation we have in mind is one of a researcher who would like to give policy advice to the policy makers in these countries on whether to initiate an immigration reform that could favor either high-skilled or low-skilled immigrants. The researcher knows that policy makers are interested in voters’ attitudes towards these types of reforms. We assume that the researcher is only able to experiment in a subset of countries due to administrative or budget constraints, and that he/she needs to extrapolate voters’ attitudes of the other policy-relevant sites based on the experimental estimates. The researcher needs to decide whether to recommend the implementation of an immigration reform directed to either high-skilled or low-skilled immigrants.
The only data needed to solve the -median problem are site-level covariates of both experimental and policy-relevant sites. We use the same covariates as Egami and Lee (2024).171717These are: Gross Domestic Product (GDP), size of migrant population, unemployment rate, proportion of females, mean age, mean education, baseline level of support for immigration by the general public, and a categorical variable that indicates the subregions in Europe (i.e., South, North, East, and West). The covariate data can be accessed in the open-source software R package, spsR. They are in different scales: For example, GDP is measured in 2015 U.S. dollars, while the unemployment rate is reported in percentage points. As is commonly recommended in clustering problems and also done in Egami and Lee (2024), we standardize all of them.
When , the -median problem is solved by the Czech Republic, Denmark, France, Ireland, Spain, and Switzerland. Figure 6 visualizes the distribution of standardized covariates, along with the sites selected by both the -median and the synthetic purposive sampling approach. Four of the six selected sites are common to both approaches (Czech Republic, Denmark, Spain, Switzerland), but synthetic purposive sampling chooses Germany and the Netherlands instead of France and Ireland.
![Refer to caption](extracted/5797333/plots/boxplot_characteristics_k6_kmedian_multicountry.png)
![Refer to caption](extracted/5797333/plots/boxplot_characteristics_k6_SPS_multicountry.png)
Notes: Box plots showing the distribution of covariates among the fifteen countries. The box plots are constructed as explained in Figure 2. Each panel depicts the site selected by the different approaches when . Solid shapes indicate sites that are in both the solution of the -median problem and the synthetic purposive sampling approach. Hollow shapes indicate solutions that differ across methods.
Figure 7 visualizes the connection networks induced by the different solutions. In both panels, red circles represent countries that are selected for experimentation. The gray lines in Figure 7 indicate the connections between the experimental sites and the policy-relevant sites that were not selected for experimentation (blue circles). For example, we can see that each policy-relevant site is connected to exactly one experimental site, i.e. its nearest neighbor; for example, the United Kingdom uses only the information of France. The connection network in Figure 7(b) is considerably more dense, with all policy-relevant sites connected to more than one experimental site. For instance, the synthetic experiment for United Kingdom assigns positive weights to the Czech Republic, Netherlands, and Germany. To further aid the visual interpretation of the connection network, we color each connection differently to capture its strength. For example, for the United Kingdom, the strongest connection is to the Netherlands , whereas the weakest connection of the United Kingdom is to the Czech Republic .
Figure 7(a) also shows that three selected countries in the -median problem (Switzerland, Czech Republic, and Spain) are not connected to any of the other policy-relevant sites. This suggests that these countries are selected because no other country provides a close enough match for them. In contrast, in Figure 7(b), these countries (Switzerland, Czech Republic, and Spain) receive positive weights from at least five other countries.
![Refer to caption](extracted/5797333/plots/network_kmedian_reordered.png)
Notes: In Panel (7(a)), a connection between a blue dot and a red dot indicates that the corresponding experimental site is the nearest neighbor of the policy site. In Panel (7(b)), each blue dot may be connected to one or multiple red dots, indicating that the corresponding policy site uses the weighted average of one or more experimental sites to construct its synthetic control. The width and the transparency of the connection line indicate the weight that each policy site puts on the connecting experimental site.
Approximation Error in Theorem 1: Because experimental estimates and corresponding standard errors are available for all countries, we can compute the approximation error in Theorem 1. Define the relative approximation error of the -median approximation to minimax regret as
(23) |
Algebraic manipulations shows that this expression is bounded from below by
(24) |
and from above by
(25) |
where . The closer the lower and upper bounds are, the better the -median approximation is.
Figure 8 displays these bounds for values of . As increases, the approximation becomes worse; as mentioned before, this is driven by ignoring the experimental sites themselves in bounding regret. In this application, the -median solution still works relatively well when , with a relative error of about . As increases to , the relative error is about .181818For these computations, we picked to be the smallest Lipschitz constant needed to capture the heterogeneity of estimated treatment effects in the data. That is, we pick as (26) where denote the estimated treatment effects for site . It turns out that this is comfortably “large” in the sense of Theorem 1. We provide additional details in Appendix B.3.
![Refer to caption](extracted/5797333/plots/approx_error_minC.png)
6 Extensions
6.1 Fixed Costs of Experimentation
We next allow for the possibility that running an experiment in a site has a fixed cost . This means that the welfare of a decision rule , given that sites are selected for experimentation, corresponds to
(27) |
Based on this welfare function, the oracle action for the policy maker is to implement the policy in any policy-relevant site for which . Expected regret of becomes
(28) |
Under the assumptions of Theorem 1, it is possible to show (by extending Lemma 1 and 2 to account for the fixed costs of experimentation) that
can be approximated by times
(29) |
and that the approximation error is the same as the one given in Theorem 1.
The problem in (29) is the metric uncapacitated -facility location problem that was discussed in Remark 7. This is a common extension of the -median problem where there is a fixed cost of opening each facility. The problem can also be formulated as a linear integer program, namely
(30) | ||||
such that | (31) | |||
(32) | ||||
(33) | ||||
(34) |
Just as before, the choice variable indicates whether facility is open, and indicates whether client is assigned to facility . Constraints are imposed to ensure that each client is assigned to at least one facility, that there are no more than facilities in total, and that a client can only be assigned to an open facility. The connection cost between a facility and a client is still given by , but now there is a fixed cost of opening a facility given by . This means that—in contrast to the case in which there is no fixed cost—choosing the optimal sites to maximize external validity now requires knowledge of the Lipschitz constant (as this constant appears explicitly on the fixed cost ). This is simply because, as we have shown, the scale of regret absent costs of experimentation depends on . Consequently, every time a new site is considered for experimentation, there is a trade-off between its contribution to reduce regret and the fixed cost of experimentation. We note that when is large, the fixed costs in the objective function become negligible, and the solution of the uncapacitated facility location problem can be approximated by the solution of the -median problem.
6.2 Random Selection of Experimental Sites
So far, our analysis focused on minimizing the worst-case (welfare-based) regret among all purposive sampling schemes that select at most sites. That is, we excluded randomized (including nonuniformly randomized) sampling schemes. This is an important limitation—a widespread view among experimenters is that “the external validity of randomized evaluations for a given population (say, the population of a country) would be maximized by randomly selecting sites and, within these sites, by randomly selecting treatment and comparison groups” (Duflo et al., 2007, p. 3953).
We now extend our baseline framework (which excludes fixed costs of experimentation) to allow for randomized site selection. First, we note that whether randomization is potentially desired delicately depends on the decision-theoretic setup. For example, the optimal sampling scheme in the Bayesian setting of Gechter et al. (2024) will typically be purposive. Whether randomization improves minimax regret depends on how exactly the decision problem is formulated, which can be related to what we refer to as the timing assumptions in an implicit game that the decision maker plays against a malicious “nature”. We will first clarify this observation and then provide a brief illustration of randomized solutions.
To formalize the discussion, let denote the cardinality of . Let denote the set of all probability distributions over the elements of . We define a randomized site selection as a probability distribution . The econometrician will pick a subset from the experimental sites by drawing one realization of a distribution that she specified. Denote the randomly selected sites by .
In our setting, whether the decision maker will want to randomize crucially depends on timing assumptions; that is, the moment in the game in which nature can move to harm the decision maker. Consider first the case in which an adversarial nature may harm the policy maker by choosing only after seeing the realization of (and knowing ). Then, the risk of using treatment rule is
and the worst-case payoff becomes
The minimax problem faced by the econometrician after has been realized is:
With slight abuse of notation, let denote the probability of choosing at random under . The (ex ante) expected payoff of any randomized site selection is
and the optimal randomized site selection solves
But this problem is solved by any supported on , the set of purposive sampling schemes that solve the site selection problem. If that problem’s solution is unique, the policy maker will never randomize under this timing of the game. Furthermore, this “sequential” timing may feel natural in applications that we have in mind.
That said, the timing that seems more in line with Wald’s (1950) original application of the minimax principle is likely one in which the policy maker commits to both a randomized sampling scheme and a set of contingent (on sampling scheme) decision rules and nature adversarially picks before any randomization was realized. We next briefly discuss this possibility.
To see that randomization might strictly speaking be optimal, consider a stylized example where and the covariates of each site are equal to its index: , and . For simplicity, suppose furthermore that , i.e. there is no sampling uncertainty in the treatment effects. This example can be solved for those combinations of sampling scheme and treatment assignment rule that achieve exact MMR. As we formally show in Appendix B.2, the exact MMR attainable by purposive sampling equals , whereas the exact MMR with randomized sampling equals .191919The assumption of perfect signals is for simplicity. The example is rigged to resemble cases analyzed in Montiel Olea et al. (2023) and Stoye (2012) and we would accordingly be able to generalize it, but for our present purpose, solving for arbitrary sampling variances would only add tedium. Thus, in principle there can be a gain to randomized sampling.
To see that solving this problem can quickly become very hard, consider now the same example, except that the experimental sites coincide with the policy sites at . Then we can find the MMR optimal combination of purposive sampling scheme and treatment assignment rule, and we can also verify that randomized site selection will strictly reduce worst-case regret. However, we are unable to characterize the exact solution for this, still extremely structured, case. We leave further exploration of randomized site selection, including the potentially fruitful investigation of “good enough” (in terms of regret) such schemes, for future research.
6.3 Other Restrictions on Treatment Heterogeneity
With Assumption 1, the effect of the policy of interest is a Lipschitz function of observed covariates with respect to the Euclidean distance. This means that the effect of the policy of interest in two sites with the same covariates is assumed to be the same (in other words, there is no unobserved treatment effect heterogeneity). In this section, we show that our main result can also accommodate other general measures of the distance between covariates (thus allowing for other functional classes), as well as some forms of unobserved treatment effect heterogeneity (provided ).
6.3.1 Other Distance Measure Based on Observed Covariates
To see how our results can allow for other types of distances, let be a measure of distance (metric) between .202020That is, we assume that for any , and that for any we have . We also assume that the function is symmetric, in that . And finally, we assume that satisfies the triangle inequality for any . Also see Dudley (2002, p.20). Then, Assumption 1 may be modified as:
Assumption 3.
is a Lipschitz function (with respect to metric ) with known constant . That is, for any , .
For example, is the Euclidean distance in Assumption 1 and for some positive definite matrix is a -weighted Euclidean distance. Choosing for some and some distance measure also effectively allows us to model as a Hölder continuous function of order (Dudley, 2002, p. 56). With Assumptions 2 and 3, is understood to be a nearest neighbor in measured in terms of metric that appears in Assumption 3. Moreover, now stands for the space of all Lipschitz functions from to with constant , but in terms of metric . We note that we are slightly abusing notation, because we are not using to index the nearest-neighbor function or the Lipschitz functional class , although their respective definitions explicitly on the choice of metric . Under a general metric , the -median problem is simply modified as:
Definition 3.
We say that a purposive sampling scheme, , solves the -median problem with a metric cost function if it solves
(35) |
6.3.2 Unobserved Treatment Heterogeneity
Now, we show how our assumptions can be modified to accommodate some forms of unobserved treatment heterogeneity. Instead of viewing policy effects as a function of observed covariates only, we now model the policy effects at the site level. Specifically, for each , denote by the policy effect in site . Then, instead of viewing as a function, we will let be a vector of dimension that represents the policy effects of all sites in . Then, Assumptions 1 and 2 can be replaced with the following:
Assumption 4.
For any , , satisfies:
(36) |
where and are both known, and is a metric.
Assumption 4 allows sites with the same covariates to have different policy effects. The difference, however, is assumed to be at most . We motivate Assumption 4 using a simple linear regression model in Appendix B.4. Denote by the collection of all vectors of dimension satisfying (36). For this parameter class, the minimax-regret optimal purposive sampling scheme and treatment rule are redefined to solve
(37) |
It turns out that the purposive sampling scheme that solves (37) can still be approximated by the solution of the -median problem in Definition 3 with a different cost function.
To see this, note that the functional class is still convex and centrosymmetric. Thus, we can show that, for the evidence aggregation problem studied in Montiel Olea et al. (2023), the conclusion of their Proposition 1(iii) extends to as well. Then, under Assumption 4, Lemmas 1 and 2 continue to hold with minor modifications. In particular, when is large enough (with a threshold that can be exactly characterized), we have that
In addition, the following lower bound can also be verified to hold
The two bounds above imply that, even when treatment effect heterogeneity is characterized by Assumption 4, the optimized value of the minimax problem (37) can still be uniformly approximated by the following objective function
When , the solution to the problem above is the same as
This problem is equivalent to the -median problem in Definition 3, where the connection cost between sites and is given by (neither nor are required as inputs).
7 Conclusion
This paper presented a decision-theoretic justification for viewing the question of how to best choose where to experiment in order to optimize external validity as a -median (clustering) problem. More concretely, we presented conditions under which minimizing the worst-case, welfare-based regret among all purposive (nonrandomized) schemes that select sites is approximately equal, and can be exactly equal, to finding the most central vectors of baseline site-level covariates.
We believe there are many interesting directions for future work. For example, while we focused on purposive sampling schemes, it would be interesting to better understand the value of randomized sampling schemes and whether site-level covariates can be used to design such randomized selection with an eye on external validity. We also think that discussions around the relation between the -median problem and synthetic purposive sampling of Egami and Lee (2024) open interesting research directions to provide a decision-theoretic justification for the use the synthetic control of Abadie et al. (2010). Finally, it would be interesting to explore the use of (mixed) integer programming techniques to solve for the purposive sampling schemes suggested in Egami and Lee (2024) and Gechter et al. (2024).
Appendix A Proofs of Main Results
A.1 Proof of Lemma 1
Fix the selected sites , and denote the cardinality of as . Let denote the indices of the experimental sites in . For a given experimental site , let denote its corresponding treatment effect estimate. Let
denote the vector containing the estimates for each experimental site.
For each policy-relevant site, , recall that denotes its nearest neighbor among the sites (or the nearest neighbor with the smallest index in case of multiplicity). Partition the policy-relevant sites as
Consider the decision rule, , that recommends, for each policy-relevant site and given data , the following action:
-
i)
For ,
-
ii)
For , set
where
Note that the expression in ii) above is well-defined for every when is large enough. This decision rule is the minimax-regret optimal rule (provided is large enough) for the evidence aggregation framework discussed in Yata (2021) and Montiel Olea et al. (2023), both of which build upon Stoye (2012), and is indeed in .
Define to be any value of for which for every . Note that by definition of infimum,
where the second inequality follows from the fact that
It is a well-known result (and can be verified by algebra) that for any ,
(38) |
where .
Moreover, it follows from (Montiel Olea et al., 2023, Proposition 1 (iii) and its proof) that if , where denotes the standard deviation of the nearest neighbor estimate in for site , it holds that for any :
(39) |
(40) |
A.2 Proof of Lemma 2
Fix the selected sites , and denote the cardinality of as . Let denote the indices of the experimental sites in . For any , define to be the element in that is closest to in terms of (if there is more than one closest element, pick the associated to the smallest index). The proof has three parts.
Part I: Consider the function such that:
We start by showing that the function is Lipschitz with constant . To see this, consider three cases:
Case 1: Suppose first that . In this case, we trivially have
Case 2: Suppose now that , but . In this case,
where the last inequality follows by the definition of and the fact that .
Case 3: Finally, take . In this case,
Without loss of generality, assume that . Then,
where the first inequality uses the definition of , and the last display uses the triangle inequality. We conclude that is a Lipschitz function with constant , which means it is included in our parameter space.
References
- Abadie et al. (2010) Abadie, A., A. Diamond, and J. Hainmueller (2010): “Synthetic control methods for comparative case studies: Estimating the effect of California’s tobacco control program,” Journal of the American statistical Association, 105, 493–505.
- Adjaho and Christensen (2022) Adjaho, C. and T. Christensen (2022): “Externally Valid Treatment Choice,” arXiv preprint arXiv:2205.05561.
- Allcott (2015) Allcott, H. (2015): “Site selection bias in program evaluation,” The Quarterly Journal of Economics, 130, 1117–1165.
- Andrews et al. (2022) Andrews, I., D. Fudenberg, A. Liang, and C. Wu (2022): “The Transfer Performance of Economic Models,” arXiv preprint arXiv:2202.04796.
- Armstrong and Kolesár (2018) Armstrong, T. B. and M. Kolesár (2018): “Optimal inference in a class of regression models,” Econometrica, 86, 655–683.
- Azevedo et al. (2020) Azevedo, E. M., A. Deng, J. L. Montiel Olea, J. Rao, and E. G. Weyl (2020): “A/b testing with fat tails,” Journal of Political Economy, 128, 4614–000.
- Azevedo et al. (2023) Azevedo, E. M., D. Mao, J. L. Montiel Olea, and A. Velez (2023): “The A/B testing problem with Gaussian priors,” Journal of Economic Theory, 210, 105646.
- Banerjee et al. (2017) Banerjee, A. V., S. Chassang, and E. Snowberg (2017): “Decision theoretic approaches to experiment design and external validity,” in Handbook of Economic Field Experiments, Elsevier, vol. 1, 141–174.
- Ben-David et al. (2010) Ben-David, S., J. Blitzer, K. Crammer, A. Kulesza, F. Pereira, and J. W. Vaughan (2010): “A theory of learning from different domains,” Machine learning, 79, 151–175.
- Bertsimas et al. (2016) Bertsimas, D., A. King, and R. Mazumder (2016): “Best subset selection via a modern optimization lens,” The Annals of Statistics, 44, 813 – 852.
- Bertsimas and Weismantel (2005) Bertsimas, D. and R. Weismantel (2005): Optimization Over Integers, Dynamic Ideas.
- Charikar et al. (2002) Charikar, M., S. Guha, É. Tardos, and D. B. Shmoys (2002): “A Constant-Factor Approximation Algorithm for the k-Median Problem,” Journal of Computer and System Sciences, 65, 129–149.
- Chassang and Kapon (2022) Chassang, S. and S. Kapon (2022): “Designing randomized controlled trials with external validity in mind,” Tech. rep., National Bureau of Economic Research.
- Christensen et al. (2022) Christensen, T., H. R. Moon, and F. Schorfheide (2022): “Optimal decision rules when payoffs are partially identified,” arXiv preprint arXiv:2204.11748.
- Cohen-Addad et al. (2018) Cohen-Addad, V., A. De Mesmay, E. Rotenberg, and A. Roytman (2018): “The bane of low-dimensionality clustering,” in Proceedings of the Twenty-Ninth Annual ACM-SIAM Symposium on Discrete Algorithms, SIAM, 441–456.
- Cohen-Addad et al. (2022) Cohen-Addad, V., H. Esfandiari, V. Mirrokni, and S. Narayanan (2022): “Improved approximations for Euclidean k-means and k-median, via nested quasi-independent sets,” in Proceedings of the 54th Annual ACM SIGACT Symposium on Theory of Computing, 1621–1628.
- Cook et al. (2002) Cook, T. D., D. T. Campbell, and W. Shadish (2002): Experimental and quasi-experimental designs for generalized causal inference, vol. 1195, Houghton Mifflin Boston, MA.
- Degtiar and Rose (2023) Degtiar, I. and S. Rose (2023): “A review of generalizability and transportability,” Annual Review of Statistics and Its Application, 10, 501–524.
- Duchi and Namkoong (2021) Duchi, J. C. and H. Namkoong (2021): “Learning models with uniform performance via distributionally robust optimization,” The Annals of Statistics, 49, 1378–1406.
- Dudley (2002) Dudley, R. (2002): Real Analysis and Probability, vol. 74, Cambridge University Press.
- Duflo et al. (2007) Duflo, E., R. Glennerster, and M. Kremer (2007): “Using randomization in development economics research: A toolkit,” Handbook of development economics, 4, 3895–3962.
- Egami and Lee (2024) Egami, N. and D. D. I. Lee (2024): “Designing Multi-Context Studies for External Validity: Site Selection via Synthetic Purposive Sampling,” .
- Gechter (2024) Gechter, M. (2024): “Generalizing the Results from Social Experiments: Theory and Evidence from India,” Journal of Business & Economic Statistics, 42, 801–811.
- Gechter et al. (2024) Gechter, M., K. Hirano, J. Lee, M. Mahmud, O. Mondal, J. Morduch, S. Ravindran, and A. S. Shonchoy (2024): “Selecting Experimental Sites for External Validity,” .
- Gurobi Optimization, LLC (2023) Gurobi Optimization, LLC (2023): “Gurobi Optimizer Reference Manual,” .
- Hu et al. (2024) Hu, Y., H. Zhu, E. Brunskill, and S. Wager (2024): “Minimax-Regret Sample Selection in Randomized Experiments,” arXiv preprint arXiv:2403.01386.
- Ishihara and Kitagawa (2021) Ishihara, T. and T. Kitagawa (2021): “Evidence Aggregation for Treatment Choice,” ArXiv:2108.06473 [econ.EM], https://doi.org/10.48550/arXiv.2108.06473.
- Kariv and Hakimi (1979) Kariv, O. and S. L. Hakimi (1979): “An algorithmic approach to network location problems. I: The p-centers,” SIAM journal on applied mathematics, 37, 513–538.
- Kumar et al. (2010) Kumar, A., Y. Sabharwal, and S. Sen (2010): “Linear-time approximation schemes for clustering problems in any dimensions,” Journal of the ACM (JACM), 57, 1–32.
- Lee et al. (2021) Lee, J. N., J. Morduch, S. Ravindran, A. Shonchoy, and H. Zaman (2021): “Poverty and migration in the digital age: Experimental evidence on mobile banking in Bangladesh,” American Economic Journal: Applied Economics, 13, 38–71.
- Manski and Tetenov (2016) Manski, C. F. and A. Tetenov (2016): “Sufficient trial size to inform clinical practice,” Proceedings of the National Academy of Sciences, 113, 10518–10523.
- Manski and Tetenov (2019) ——— (2019): “Trial Size for Near-Optimal Choice Between Surveillance and Aggressive Treatment: Reconsidering MSLT-II,” The American Statistician, 73, 305–311.
- Mansour et al. (2009) Mansour, Y., M. Mohri, and A. Rostamizadeh (2009): “Domain Adaptation: Learning Bounds and Algorithms,” in Proceedings of The 22nd Annual Conference on Learning Theory (COLT 2009), Montréal, Canada.
- Megiddo and Supowit (1984) Megiddo, N. and K. J. Supowit (1984): “On the complexity of some common geometric location problems,” SIAM journal on computing, 13, 182–196.
- Menzel (2023) Menzel, K. (2023): “Transfer Estimates for Causal Effects across Heterogeneous Sites,” arXiv preprint arXiv:2305.01435.
- Montiel Olea et al. (2023) Montiel Olea, J. L., C. Qiu, and J. Stoye (2023): “Decision Theory for Treatment Choice Problems with Partial Identification,” arXiv preprint arXiv:2312.17623.
- Müller (2011) Müller, U. K. (2011): “Efficient tests under a weak convergence assumption,” Econometrica, 79, 395–435.
- Naumann et al. (2018) Naumann, E., L. F. Stöetzer, and G. Pietrantuono (2018): “Attitudes towards highly skilled and low-skilled immigration in Europe: A survey experiment in 15 European countries,” European Journal of Political Research, 57, 1009–1030.
- Raiffa and Schlaifer (1961) Raiffa, H. and R. Schlaifer (1961): “Applied statistical decision theory.” .
- Sahoo et al. (2022) Sahoo, R., L. Lei, and S. Wager (2022): “Learning from a biased sample,” arXiv preprint arXiv:2209.01754.
- Stoye (2012) Stoye, J. (2012): “Minimax regret treatment choice with covariates or with limited validity of experiments,” Journal of Econometrics, 166, 138–156.
- Sugiyama et al. (2007) Sugiyama, M., M. Krauledat, and K.-R. Müller (2007): “Covariate shift adaptation by importance weighted cross validation.” Journal of Machine Learning Research, 8.
- Tetenov (2012) Tetenov, A. (2012): “Statistical treatment choice based on asymmetric minimax regret criteria,” Journal of Econometrics, 166, 157–165.
- Vivalt (2020) Vivalt, E. (2020): “How much can we generalize from impact evaluations?” Journal of the European Economic Association, 18, 3045–3089.
- Wald (1950) Wald, A. (1950): Statistical Decision Functions, Wiley publications in statistics, New York; Chapman & Hall: London.
- Williamson and Shmoys (2011) Williamson, D. P. and D. B. Shmoys (2011): The design of approximation algorithms, Cambridge university press.
- Yata (2021) Yata, K. (2021): “Optimal Decision Rules Under Partial Identification,” ArXiv:2111.04926 [econ.EM], https://doi.org/10.48550/arXiv.2111.04926.
- Zhang (2007) Zhang, P. (2007): “A new approximation algorithm for the k-facility location problem,” Theoretical Computer Science, 384, 126–135.
Appendix B Supplemental Appendix
B.1 Gurobi’s Output
In this section, we discuss some parameters in the Gurobi optimizer that can be tuned to improve its performance.212121For more specifics, see https://www.gurobi.com/documentation/current/refman/mip_models.html. First, MIPFocus specifies, on a high level, whether the solution should prioritize speed or optimality. The default value of this parameter is , which achieves a balance between searching for new solutions and proving optimality of the current solution. We set this parameter to , prioritizing finding good-quality and optimal solutions. A second set of important parameters affect how the solver terminates. There are two termination choices for MIP models: 1) a restriction on the runtime of the algorithm, such as using TimeLimit to limit the wall-clock runtime, and 2) controlling the optimality gap, by setting a parameter MIPGap that stops the algorithm when the relative gap between the best-known solution and the best known bound on the solution objective is less than the specified value. In this application, we let the runtime to be the default value (infinity) and set the tolerance level to . In Figure 9, the Gurobi solver outputs not only the solution and the optimal value of the problem but also the output gap. In this example, we can see that the gap between bounds is , demonstrating that the -median problem has been solved to provable optimality. An interesting feature of using the integer programming formulation of the -median problem is that even when we are not able to solve the problem to provable optimality, the optimality gap provides a suboptimality guarantee for the obtained solution. Finally, the output reports the algorithm’s runtime, which was only seconds for this example. There are more parameters one can tune to control the root relaxation, the aggressiveness of the cutting plane strategies and the level of presolve, tolerance parameters for primal feasibility, the integer feasibility, and more. For the applications in this paper, we let the rest of the parameters be the default.
![Refer to caption](extracted/5797333/plots/multicntry_k6_gurobi.png)
Notes: An example output from the MIP solver in Gurobi using the multi-country experiment for . See details of the application in Section 5.2.
B.2 Exact Statement and Proof of Result Discussed in Section 6.2
Consider the following specialization of our setting: and therefore , , , and for all sites we just have . (Intuitively, we consider 4 sites lined up equidistantly on a straight line, where the middle two sites are the policy sites.) Suppose furthermore that sampling distributions of signals are degenerate; formally, for all .
Lemma 3.
Under this section’s additional assumptions (see paragraph immediately above):
-
1.
If sampling schemes may randomize, the lowest regret achievable by any combination of sampling scheme and treatment rule equals .
-
2.
If sampling has to be purposive (nonrandomized), the lowest regret achievable in combination with any treatment rule equals .
Proof.
To see claim 1, consider the distribution under which is uniformly distributed over . Then the experimental sites do not yield any information, and no sampling and decision rule can improve on tossing a coin, in which case expected regret equals . We conclude that the MMR value of this decision problem is at least .
Next, suppose the policy maker uniformly randomizes over for experimentation and then implements the new policy with probability ; here, the notation indicates clamping of to . We will show that the worst-case expected regret under this scheme is , which therefore is the problem’s MMR value and is attained by this rule.
Careful book-keeping reveals that the policy maker’s worst-case expected regret equals
We will solve this by considering subcases. Suppose first that and have different signs. Since both objective and constraints are invariant under multiplying by , suppose without further loss of generality that . The optimization problem now simplifies to
where the first inequality is justified in the display and the second one follows because and have different signs but differ by at most .
Now let and have the same sign, which we take without further loss of generality to be positive. Then we initially observe simplification to
and we can furthermore concentrate out , to get
Clamping of expressions at cannot bind because and are positive. If clamping at 0 binds for both fractions, then the objective equals . Suppose clamping at binds for one expression, say (without further loss of generality) because , then we have simplification to
Keeping in mind that and therefore also in this subcase, evaluation of derivatives shows that this expression decreases in ; hence, . Substituting this in, one can further verify the expression to be decreasing in ; therefore, the maximal value in this subcase is attained at a boundary point also covered by the next case (and, though not essential for the argument, this value can be verified to be ).
Finally, if no clamping binds, we can reduce the problem to
where the maximum is attained by setting .
Regarding claim 2, by the decision problem’s symmetry, it is without further loss of generality to assume that site is being sampled. MMR is then at least because this value is achieved if the true parameter values are equally likely to be or .
We next show that this value is attained by uniformly assigning treatment with probability . Indeed, worst case regret of this decision rule equals
If and have different signs, we can bound this value by just as before. Suppose they have the same sign, which we take to be positive without further loss of generality, then the problem simplifies to
Here, the first equality concentrates out and ; the second equality uses that clamping cannot bind (clamping at would set the expression to , clamping at would imply that ), after which the problem is straightforwardly solved by . ∎
B.3 Additional Analysis of the Survey Experiment
As mentioned in Section 5, we can obtain experimental estimates of all fifteen countries using the original experiments conducted in Naumann et al. (2018). Generally, experimental estimates of all policy sites are unknown and unattainable in most real-world applications; otherwise, there is no need to solve the site selection problem. However, as an illustrative example, we will leverage the information in these experiments to quantify the magnitude of the Lipschitz constant needed to explain the treatment heterogeneity in the data and the constant in the assumption of Lemma 1 that gives the result of this paper.
The outcome of interest from the survey is a categorical variable indicating survey respondents’ attitudes towards immigrants: for Allow None; for Allow A Few; for Allow Some; for Allow Many. For a more straightforward interpretation, we redefine the outcome variable to be binary: we let the outcome of the survey respondent be if she answers or , indicating “support”; otherwise, we let her outcome be , indicating “oppose.” The treatment is also a binary variable, which equals if the survey is about high-skilled (low-skilled) immigrants. We use a simple difference-in-means estimator to estimate the treatment effect of each country. The table below shows the point estimates and their standard errors. The point estimates speak to the difference between the percentage of people who support high-skilled immigrants and the percentage of people who support low-skilled immigrants. For example, 25.8% more survey respondents in Austria are more supportive of high-skilled immigrants, compared to low-skilled immigrants.
Country Estimate Standard Error Austria 0.258906 0.026490 Belgium 0.232145 0.024740 Switzerland 0.285371 0.027139 Czechia 0.222865 0.019601 Germany 0.339650 0.016396 Denmark 0.293745 0.025156 Spain 0.265763 0.022948 Finland 0.403363 0.020222 France 0.275320 0.022045 United Kingdom 0.407362 0.022651 Ireland 0.238961 0.020889 Netherlands 0.301243 0.022853 Norway 0.262747 0.025613 Sweden 0.149249 0.019271 Slovenia 0.301862 0.025517
Notes: This table presents the experimental estimates of the treatment effect of the policy in each country. The original outcome from the survey is a categorical variable indicating survey respondents’ attitudes towards immigrants: for Allow None; for Allow A Few; for Allow Some; for Allow Many. We redefine the outcome variable to be binary: we let the outcome of the survey respondent be if she answers or , indicating “support”; otherwise, we let her outcome be , indicating “oppose.” The treatment is also a binary variable, which equals if the survey is about high-skilled (low-skilled) immigrants. We use a simple difference-in-means estimator to estimate the treatment effect of each country and their standard error.
In Figure 10(a), each point represents a pair of two countries, and the slope from the origin to each point represents the smallest Lipschitz constant needed to explain the data observed for these two countries. Hence, the Lipschitz constant that is able to explain the treatment effect heterogeneity in the data for all countries is at least
(44) |
which equals 0.0892 and corresponds to the slope of the red dashed line in Figure 10. The pair of countries that give the maximum of equation 44 is Finland and Sweden. Additionally, Montiel Olea et al. (2023) show that, for each possible set of experimental sites , the that gives the solution in Lemma 1 is defined as
(45) |
Replacing with the corresponding estimated standard errors, we get equals , which corresponds to the slope of the blue dashed line in figure 10(a). Both the numbers and the plot indicate the smallest Lipschitz constant needed to explain the data is bigger than the largest lower bound of that gives the nearest neighbor result. Additionally, Figure 10(b) shows a histogram of for all possible , and the red dashed line is the smallest Lipschitz constant compatible with data. By visual inspection, we can conclude that, in this application, the assumption is likely to hold. It is worth pointing out that this assumption is in general not testable because the experimental estimates of all policy sites are unknown and we cannot compute .
![Refer to caption](extracted/5797333/plots/TE_X_diff.png)
![Refer to caption](extracted/5797333/plots/C_allExp.png)
Notes: The dots in Panel 10(a) represent all possible pairs among the countries. The slope of each point connecting to the origin represents the value of the Lipschitz constant needed to explain the estimated treatment effects and site-level covariates observed for that pair of countries. The red dash line is the smallest that is needed to explain all the data, computed using equation 44. The slope of the blue dashed line represents , computed using 45. Panel 10(b) presents a histogram of for all possible choices of experimental sites . The red dashed line is the smallest Lipschitz constant needed to explain the data, corresponding to the same red dashed line in panel 10(a).
B.4 A Simple Example that Motivates Assumption 4
In this section, we provide a simple linear regression example to motivate Assumption 4. Suppose the effect of the status-quo is known in all sites and normalized to zero. For each site , we have a random sample of units in the experiment. Let be the outcome under the policy of interest for unit . We assume that is generated as follows:
where is the observed unit-level covariate for individual in site , is the unobserved unit-level covariate, is an error term with and is also independent of , and are the same across different sites. For simplicity, suppose that and are jointly normal: , where and is a positive-definite covariance matrix.
Let be the sample average of the observed outcome at site , and be the observed and unobserved site-level aggregate covariates, respectively. Under the above assumptions and conditional on observed site-level aggregate covariate , we have
(46) |
where is the expectation of conditional , and . Then, Assumption 4 models a case for which the policy effect of interest is
Furthermore, (46) implies that, conditional on , we have an unbiased and normal estimator for . We may also calculate that for some that depends on each site . Then, for , , we have:
(47) |
implying that can be bounded as in Assumption 4 for some positive and , as long as we are willing to assume that are bounded, and that are bounded uniformly among all .