Papers › auton-survival: an Open-Source Package for Regression, Counterfactual Estimation,...

auton-survival: an Open-Source Package for Regression, Counterfactual Estimation, Evaluation and Phenotyping with Censored Time-to-Event Data

15 Apr 2022arXiv:2204.07276archive 2025-07-28

Chirag Nagpal, Willa Potosnak, Artur Dubrawski

Applications of machine learning in healthcare often require working with time-to-event prediction tasks including prognostication of an adverse event, re-hospitalization or death. Such outcomes are typically subject to censoring due to loss of follow up. Standard machine learning methods cannot be applied in a straightforward manner to datasets with censored outcomes. In this paper, we present auton-survival, an open-source repository of tools to streamline working with censored time-to-event or survival data. auton-survival includes tools for survival regression, adjustment in the presence of domain shift, counterfactual estimation, phenotyping for risk stratification, evaluation, as well as estimation of treatment effects. Through real world case studies employing a large subset of the SEER oncology incidence data, we demonstrate the ability of auton-survival to rapidly support data scientists in answering complex health and epidemiological questions.

PaperPDFCode

Code

autonlab/auton-survival officialmentioned in papermentioned on GitHubpytorch report
chiragnagpal/DeepSurvivalMachines mentioned on GitHubpytorch report
qinzzz/auton-survival-785 mentioned on GitHubpytorch report

Repository list and official/mentioned flags are the archive's, frozen 2025-07-28. Reachability, where shown, is from one Syntology probe window (2026-09-16 to 2026-09-18); repositories not probed show nothing. GitHub stars are not tracked.

Code Syntology ran Syntology

Not run by Syntology. Nothing on this page verifies that the listed code works.

Tasks

BIG-bench Machine LearningTime-to-Event Prediction

1 archive task tag without a task page not shown.

Results from the paper archive 2025-07-28

No leaderboard rows for this paper in the archive.

Methods

1x1 ConvolutionAverage PoolingBatch NormalizationConvolutionCosine AnnealingDense ConnectionsGlobal Average PoolingGrouped ConvolutionLARSReLURegNetYSEERSigmoid ActivationSqueeze-and-Excitation BlockSwAV

Report a problem or propose a change · a person checks every report against the paper or source before anything changes; decisions are listed on /corrections