A Naive Bayes machine learning approach to risk prediction using censored, time-to-event data
Predicting an individual's risk of experiencing a future clinical outcome is a statistical task with important consequences for both practicing clinicians and public health experts. Modern observational databases such as electronic health records (EHRs) provide an alternative to the longitudinal cohort studies traditionally used to construct risk models, bringing with them both opportunities and challenges. Large sample sizes and detailed covariate histories enable the use of sophisticated machine learning techniques to uncover complex associations and interactions, but observational databases are often ``messy,'' with high levels of missing data and incomplete patient follow-up. In this paper, we propose an adaptation of the well-known Naive Bayes (NB) machine learning approach for classification to time-to-event outcomes subject to censoring. We compare the predictive performance of our method to the Cox proportional hazards model which is commonly used for risk prediction in healthcare populations, and illustrate its application to prediction of cardiovascular risk using an EHR dataset from a large Midwest integrated healthcare system.
Code (0)
등록된 구현이 없습니다.
Tasks
BIG-bench Machine LearningSimilar Papers 제목 키워드 기반
Data mining for censored time-to-event data: A Bayesian network model for predicting cardiovascular risk from electronic health record data
Models for predicting the risk of cardiovascular events based on individual patient characteristics are important tools for managing patient care. Most current and commonly used risk prediction models have been built fro…
BIG-bench Machine LearningType I Tobit Bayesian Additive Regression Trees for Censored Outcome Regression
Censoring occurs when an outcome is unobserved beyond some threshold value. Methods that do not account for censoring produce biased predictions of the unobserved outcome. This paper introduces Type I Tobit Bayesian Addi…
regressionVocal Bursts Type PredictionEnhancing Uncertainty Quantification in Drug Discovery with Censored Regression Labels
In the early stages of drug discovery, decisions regarding which experiments to pursue can be influenced by computational models. These decisions are critical due to the time-consuming and expensive nature of the experim…
Drug DiscoveryregressionSurvival AnalysisUncertainty QuantificationEarly Detection of At-Risk Students Using Machine Learning
This research presents preliminary work to address the challenge of identifying at-risk students using supervised machine learning and three unique data categories: engagement, demographics, and performance data collecte…
Binary ClassificationStudent dropoutDeep Survival Machines: Fully Parametric Survival Regression and Representation Learning for Censored Data with Competing Risks
We describe a new approach to estimating relative risks in time-to-event prediction problems with censored data in a fully parametric manner. Our approach does not require making strong assumptions of constant proportion…
Representation LearningSurvival AnalysisTime-to-Event Prediction