Causal Effect Estimation under Informative Censoring: A Multiple-Imputation Approach with Shadow Calibration
Yingrong Wang, Baohong Li, Dian Jin, Yi He, Anpeng Wu, Yujie Shao, Dong Shen, Kun Kuang
Abstract
In many data-driven decision-making scenarios, it is crucial to accurately estimate causal effects on time-to-event outcomes, e.g., time-to-purchase. However, these outcomes are often right-censored: due to events occurring beyond the observation period or loss to follow-up, the observed durations for censored samples are the last follow-up times rather than the actual event times. Critically, this censoring is usually informative : the censoring mechanism is intrinsically linked to the event time, which may be partially missing. This dependency violates the non-informative censoring assumption in conventional survival analysis. Consequently, the latent information in censored data is non-ignorable for causal effect estimation, inducing a censoring bias that remains unresolved by current methodologies. To overcome these challenges, we establish a theoretical framework for identifying causal effects on time-to-event outcomes under informative censoring. Based on this, we propose Multiple Imputation with Shadow Calibration (MISC), a principled framework leveraging shadow variables --- fully observed covariates correlated with event time but conditionally independent of the censoring mechanism --- to enable unbiased estimation. Specifically, MISC (1) automatically learns valid shadow representations from observed covariates, (2) utilizes these representations to identify the conditional distribution of event times to impute censored units, and (3) estimates causal effects using the imputed data. Experimental results on semi-synthetic benchmarks and real-world datasets from medical scenarios and a large-scale e-commerce platform show that MISC achieves state-of-the-art performance in causal effect estimation and counterfactual prediction. Codes are available at: https://github.com/suikateiou/MISC.
Ask about this paper
Ask your agent about it.
Lune has read the top-tier papers around this one, so every answer names the papers it rests on.
Related papers
- Deep Copula-Based Survival Analysis for Dependent Censoring with Identifiability GuaranteesWeijia Zhang, Chun Kai Ling, Xuanhui ZhangAAAI 2024 · 12 citations
- Doubly Robust Conformalized Survival Analysis with Right-Censored DataMatteo Sesia, Vladimir SvetnikICML 2025
- Two-Stage Shadow Inclusion Estimation: An IV Approach for Causal Inference under Latent Confounding and Collider BiasBaohong Li, Anpeng Wu, Ruoxuan Xiong, Kun KuangICML 2024 · 5 citations
- Explicitly Modeling Censoring Produces Superior Survival PredictorsShi-ang Qi, Yakun Yu, Russell GreinerICML 2026
- MIRACLE: Causally-Aware Imputation via Learning Missing Data MechanismsTrent Kyono, Yao Zhang, Alexis Bellot, Mihaela van der SchaarNeurIPS 2021 · 105 citations
