Testing Dependent Censoring via Mutual-Information Uncertainty and Deep Latent Imputation
Abstract
Learning a survival prediction model can be viewed as regression with the added complication of censoring. Each subject has a true event time and a censoring time , yet we only observe and . Many standard methods assume conditional independence , which may not hold in practice. Assessing this dependence is challenging, as for each subject we observe either or , but never both. To address this, for each we define indicator variables and , taking values in , where "?" denotes censoring-induced missingness. Using across multiple time points, we develop a model-assisted diagnostic for testing . Our statistic is the Conditional Mutual Information (CMI) uncertainty range, ΔIₜ = Iᵗₘₐₓ − Iᵗₘᵢₙ, defined as the width of the CMI values attainable over all feasible completions of the missing "?" entries. Here, Iᵗₘₐₓ and Iᵗₘᵢₙ are the corresponding maximum and minimum feasible CMI values. Under independence, ΔIₜ follows a characteristic null distribution, whereas dependent censoring induces structured deviations. To compute these bounds efficiently, we formulate the problem as a decomposable integer program and solve it exactly via a polynomial-time algorithm. To calibrate the statistic, we combine this uncertainty measure with a deep variational frailty model that imputes latent event and censoring marginals, followed by stratified permutation to generate a model-assisted null distribution under conditional independence. Synthetic experiments show calibrated behavior under independent censoring and increasing power for several dependent-censoring mechanisms, while real-data experiments reveal heterogeneous evidence of possible dependent censoring.
est. 32% chance this paper gets accepted at ICLR 2027.
What do you think this paper will get?
All positions stay anonymous.