Optimal Transport Methods for Causal Inference, Multisample Testing, and Model Interpretation

The manuscript discusses three topics that utilize optimal transport and related methodologies to solve problems in statistics. Chapter 2 uses the Wasserstein distance to construct interpretable approximations to complicated models, Chapter 3 uses optimal transport distances to construct weighting estimators for causal inference, and Chapter 4 uses Hamiltonian paths and nearest neighbor graphs for multivariate testing. Each chapter is self-contained and the corresponding abstracts are given below. Chapter 2: Statistical models often include thousands of parameters. However, large models decrease the investigator’s ability to interpret and communicate the estimated parameters. Reducing the dimensionality of the parameter space in the estimation phase is a commonly used approach, but less work has focused on selecting subsets of the parameters for interpreting the estimated model — especially in settings such as Bayesian inference and model averaging. Importantly, many models do not have straightforward interpretations and create another layer of obfuscation. To solve this gap, we introduce a new method that uses the Wasserstein distance to identify a low-dimensional interpretable model projection. After the estimation of complex models, users can budget how many parameters they wish to interpret and the proposed generates a simplified model of the desired dimension minimizing the distance to the full model. We provide simulation results to illustrate the method and apply it to cancer datasets. Chapter 3: Weighting methods are a common tool to de-bias estimates of causal effects. And though there are an increasing number of seemingly disparate methods, many of them can be folded into one unifying regime: causal optimal transport. This new method directly targets distributional balance by minimizing optimal transport distances between treatment and control groups or, more generally, between a source and target population. Our approach is model-free but can also incorporate moments or any other important functions of covariates that the re- searcher desires to balance. We find that the causal optimal transport outperforms competitor methods when both the propensity score and outcome models are misspecified, indicating it is a robust alternative to common weighting methods. Finally, we demonstrate the utility of our method in an external control study examining the effect of misoprostol versus oxytocin for treatment of post-partum hemorrhage. Chapter 4: We prop