TasksSotADatasetsPapersMethodsSubmitAbout
Papers With Code 2

A community resource for machine learning research: papers, code, benchmarks, and state-of-the-art results.

Explore

Notable BenchmarksAll SotADatasetsPapersMethods

Community

Submit ResultsAbout

Data sourced from the PWC Archive (CC-BY-SA 4.0). Built by the community, for the community.

Papers/auton-survival: an Open-Source Package for Regression, Cou...

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

Chirag Nagpal, Willa Potosnak, Artur Dubrawski

2022-04-15Time-to-Event PredictionBIG-bench Machine Learning
PaperPDFCodeCode(official)Code

Abstract

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.

Related Papers

Time-to-event prediction for grouped variables using Exclusive Lasso2025-04-02Enhancing Collaborative Filtering-Based Course Recommendations by Exploiting Time-to-Event Information with Survival Analysis2025-02-27Evidential time-to-event prediction with calibrated uncertainty quantification2024-11-12An Introduction to Deep Survival Analysis Models for Predicting Time-to-Event Outcomes2024-10-01An evidential time-to-event prediction model based on Gaussian random fuzzy numbers2024-06-19A personalized Uncertainty Quantification framework for patient survival models: estimating individual uncertainty of patients with metastatic brain tumors in the absence of ground truth2023-11-28DySurv: dynamic deep learning model for survival analysis with conditional variational inference2023-10-28Towards Flexible Time-to-event Modeling: Optimizing Neural Networks via Rank Regression2023-07-16