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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Factual outcome prediction | MIMIC-III extract | RMSE9.14 | 105 | |
| Counterfactual Outcome Estimation | Tumor Growth tau=2 synthetic (test) | RMSE3.13 | 77 | |
| Tumor Volume Forecasting | Synthetic Cancer Dataset (test) | RMSE0.11 | 72 | |
| Counterfactual outcome prediction | MIMIC-III semi-synthetic (N=1000) (test) | RMSE0.3 | 35 | |
| Counterfactual outcome prediction | MIMIC-III semi-synthetic (N=2000) (test) | RMSE0.27 | 35 | |
| Counterfactual outcome prediction | MIMIC-III semi-synthetic (N=3000) (test) | RMSE0.24 | 35 | |
| Causal outcome forecasting | MIMIC-IV sepsis SOFA-SCORE | RMSE0.09 | 24 | |
| Causal outcome forecasting | MIMIC-IV sepsis (CREATININE) | RMSE0.09 | 24 | |
| Causal outcome forecasting | MIMIC sepsis BILIRUBIN-TOTAL IV | RMSE0.07 | 24 | |
| Causal outcome forecasting | MIMIC-IV sepsis (ALT) | RMSE0.06 | 24 |