# Recovering Sparse and Interpretable Subgroups with Heterogeneous Treatment Effects with Censored Time-to-Event Outcomes

arXiv: https://arxiv.org/abs/2302.12504 · HTML: https://mlsanity.com/paper/2302.12504/

## Key facts

- Cited by 1 papers in the map (data through 14 Oct 2026).
- Builds on 13 of its 37 references that are in the map.
- First posted on arXiv 24 Feb 2023.

## Builds on (most cited first, 13 of 13)

- [Estimation and Inference of Heterogeneous Treatment Effects using Random Forests](https://mlsanity.com/paper/1510.04342/) (2015)
- [Causal Effect Inference with Deep Latent-Variable Models](https://mlsanity.com/paper/1705.08821/) (2017)
- [Bayesian Inference of Individualized Treatment Effects using Multi-task Gaussian Processes](https://mlsanity.com/paper/1704.02801/) (2017)
- [Generalization Bounds and Representation Learning for Estimation of Potential Outcomes and Causal Effects](https://mlsanity.com/paper/2001.07426/) (2020)
- [Deep Cox Mixtures for Survival Regression](https://mlsanity.com/paper/2101.06536/) (2021)
- [SurvITE: Learning Heterogeneous Treatment Effects from Time-to-Event Data](https://mlsanity.com/paper/2110.14001/) (2021)
- [auton-survival: an Open-Source Package for Regression, Counterfactual Estimation, Evaluation and Phenotyping with Censored Time-to-Event Data](https://mlsanity.com/paper/2204.07276/) (2022)
- [Causal Rule Sets for Identifying Subgroups with Enhanced Treatment Effect](https://mlsanity.com/paper/1710.05426/) (2017)
- [Benchmarking Heterogeneous Treatment Effect Models through the Lens of Interpretability](https://mlsanity.com/paper/2206.08363/) (2022)
- [Interpretable Subgroup Discovery in Treatment Effect Estimation with Application to Opioid Prescribing Guidelines](https://mlsanity.com/paper/1905.03297/) (2019)
- [Counterfactual Phenotyping with Censored Time-to-Events](https://mlsanity.com/paper/2202.11089/) (2022)
- [Adaptive Hyper-box Matching for Interpretable Individualized Treatment Effect Estimation](https://mlsanity.com/paper/2003.01805/) (2020)
- [Interpretable Personalized Experimentation](https://mlsanity.com/paper/2111.03267/) (2021)

## Built on by (most cited first, 1 of 1)

- [Identifying treatment response subgroups in observational time-to-event data](https://mlsanity.com/paper/2408.03463/) (2024)

## Abstract (arXiv)

Studies involving both randomized experiments as well as observational data typically involve time-to-event outcomes such as time-to-failure, death or onset of an adverse condition. Such outcomes are typically subject to censoring due to loss of follow-up and established statistical practice involves comparing treatment efficacy in terms of hazard ratios between the treated and control groups. In this paper we propose a statistical approach to recovering sparse phenogroups (or subtypes) that demonstrate differential treatment effects as compared to the study population. Our approach involves modelling the data as a mixture while enforcing parameter shrinkage through structured sparsity regularization. We propose a novel inference procedure for the proposed model and demonstrate its efficacy in recovering sparse phenotypes across large landmark real world clinical studies in cardiovascular health.
