Benchmarking Framework for Performance-Evaluation of Causal Inference Analysis

Abstract

Causal inference analysis is the estimation of the effects of actions on outcomes. In the context of healthcare data this means estimating the outcome of counter-factual treatments (i.e. including treatments that were not observed) on a patient's outcome. Compared to classic machine learning methods, evaluation and validation of causal inference analysis is more challenging because ground truth data of counter-factual outcome can never be obtained in any real-world scenario. Here, we present a comprehensive framework for benchmarking algorithms that estimate causal effect. The framework includes unlabeled data for prediction, labeled data for validation, and code for automatic evaluation of algorithm predictions using both established and novel metrics. The data is based on real-world covariates, and the treatment assignments and outcomes are based on simulations, which provides the basis for validation. In this framework we address two questions: one of scaling, and the other of data-censoring. The framework is available as open source code at https://github.com/IBM-HRL-MLHLS/IBM-Causal-Inference-Benchmarking-Framework

arXiv comment9 pages, 1 figure

Cited by in corpus (12)

• Adapting Neural Networks for the Estimation of Treatment Effects
• Identifying Causal-Effect Inference Failure with Uncertainty-Aware Models
• Generalization Bounds and Representation Learning for Estimation of Potential Outcomes and Causal Effects
• A Survey on Causal Inference
• An Evaluation Toolkit to Guide Model Selection and Cohort Definition in Causal Inference
• RealCause: Realistic Causal Inference Benchmarking
• Generating Synthetic Text Data to Evaluate Causal Inference Methods
• Bayesian Model Averaging for Causality Estimation and its Approximation based on Gaussian Scale Mixture Distributions
• AutoCP: Automated Pipelines for Accurate Prediction Intervals
• ADCB: An Alzheimer's disease benchmark for evaluating observational estimators of causal effects
• Identifying Causal Influences on Publication Trends and Behavior: A Case Study of the Computational Linguistics Community
• Causal Effect Estimation using Variational Information Bottleneck