Causal machine learning for predicting treatment outcomes

S Feuerriegel, D Frauen, V Melnychuk, J Schweisthal… - Nature Medicine, 2024 - nature.com
Causal machine learning (ML) offers flexible, data-driven methods for predicting treatment
outcomes including efficacy and toxicity, thereby supporting the assessment and safety of …

Uncertainty quantification over graph with conformalized graph neural networks

K Huang, Y Jin, E Candes… - Advances in Neural …, 2024 - proceedings.neurips.cc
Abstract Graph Neural Networks (GNNs) are powerful machine learning prediction models
on graph-structured data. However, GNNs lack rigorous uncertainty estimates, limiting their …

Using machine learning to individualize treatment effect estimation: Challenges and opportunities

A Curth, RW Peck, E McKinney… - Clinical …, 2024 - Wiley Online Library
The use of data from randomized clinical trials to justify treatment decisions for real‐world
patients is the current state of the art. It relies on the assumption that average treatment …

Conformal meta-learners for predictive inference of individual treatment effects

AM Alaa, Z Ahmad… - Advances in Neural …, 2024 - proceedings.neurips.cc
We investigate the problem of machine learning-based (ML) predictive inference on
individual treatment effects (ITEs). Previous work has focused primarily on developing ML …

Selection by prediction with conformal p-values

Y Jin, EJ Candès - Journal of Machine Learning Research, 2023 - jmlr.org
Decision making or scientific discovery pipelines such as job hiring and drug discovery often
involve multiple stages: before any resource-intensive step, there is often an initial screening …

Sharp bounds for generalized causal sensitivity analysis

D Frauen, V Melnychuk… - Advances in Neural …, 2024 - proceedings.neurips.cc
Causal inference from observational data is crucial for many disciplines such as medicine
and economics. However, sharp bounds for causal effects under relaxations of the …

Conformalized matrix completion

Y Gui, R Barber, C Ma - Advances in Neural Information …, 2023 - proceedings.neurips.cc
Matrix completion aims to estimate missing entries in a data matrix, using the assumption of
a low-complexity structure (eg, low-rankness) so that imputation is possible. While many …

Conformal sensitivity analysis for individual treatment effects

M Yin, C Shi, Y Wang, DM Blei - Journal of the American Statistical …, 2024 - Taylor & Francis
Estimating an individual treatment effect (ITE) is essential to personalized decision making.
However, existing methods for estimating the ITE often rely on unconfoundedness, an …

Doubly-valid/doubly-sharp sensitivity analysis for causal inference with unmeasured confounding

J Dorn, K Guo, N Kallus - Journal of the American Statistical …, 2024 - Taylor & Francis
We consider the problem of constructing bounds on the average treatment effect (ATE) when
unmeasured confounders exist but have bounded influence. Specifically, we assume that …

Policy learning" without''overlap: Pessimism and generalized empirical Bernstein's inequality

Y Jin, Z Ren, Z Yang, Z Wang - arXiv preprint arXiv:2212.09900, 2022 - arxiv.org
This paper studies offline policy learning, which aims at utilizing observations collected a
priori (from either fixed or adaptively evolving behavior policies) to learn the optimal …