Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

IGC-Net for conditional average potential outcome estimation over time

About

Estimating potential outcomes for treatments over time based on observational data is important for personalized decision-making in medicine. However, many existing methods for this task fail to properly adjust for time-varying confounding and thus yield biased estimates. There are only a few neural methods with proper adjustments, but these have inherent limitations (e.g., division by propensity scores that are often close to zero), which result in poor performance. As a remedy, we introduce the iterative G-computation network (IGC-Net). Our IGC-Net is a novel, neural end-to-end model which adjusts for time-varying confounding in order to estimate conditional average potential outcomes (CAPOs) over time. Specifically, our IGC-Net is the first neural model to perform fully regression-based iterative G-computation for CAPOs in the time-varying setting. We evaluate the effectiveness of our IGC-Net across various experiments. In sum, this work represents a significant step towards personalized decision-making from electronic health records.

Konstantin Hess, Dennis Frauen, Valentyn Melnychuk, Stefan Feuerriegel• 2024

Related benchmarks

TaskDatasetResultRank
Factual outcome predictionMIMIC-III extract
RMSE9.14
105
Counterfactual Outcome EstimationTumor Growth tau=2 synthetic (test)
RMSE3.13
77
Tumor Volume ForecastingSynthetic Cancer Dataset (test)
RMSE0.11
72
Counterfactual outcome predictionMIMIC-III semi-synthetic (N=1000) (test)
RMSE0.3
35
Counterfactual outcome predictionMIMIC-III semi-synthetic (N=2000) (test)
RMSE0.27
35
Counterfactual outcome predictionMIMIC-III semi-synthetic (N=3000) (test)
RMSE0.24
35
Causal outcome forecastingMIMIC-IV sepsis SOFA-SCORE
RMSE0.09
24
Causal outcome forecastingMIMIC-IV sepsis (CREATININE)
RMSE0.09
24
Causal outcome forecastingMIMIC sepsis BILIRUBIN-TOTAL IV
RMSE0.07
24
Causal outcome forecastingMIMIC-IV sepsis (ALT)
RMSE0.06
24
Showing 10 of 11 rows

Other info

Follow for update