Unknown

Dataset Information

0

Applying machine learning to predict real-world individual treatment effects: insights from a virtual patient cohort.


ABSTRACT: Objective:We aimed to investigate bias in applying machine learning to predict real-world individual treatment effects.

Materials and methods:Using a virtual patient cohort, we simulated real-world healthcare data and applied random forest and gradient boosting classifiers to develop prediction models. Treatment effect was estimated as the difference between the predicted outcomes of a treatment and a control. We evaluated the impact of predictors (ie, treatment predictors [X1], confounders [X2], treatment effects modifiers [X3], and other outcome risk factors [X4]) with known effects on treatment and outcome using real-world data, and outcome imbalance on predicting individual outcome. Using counterfactuals, we evaluated percentage of patients with biased predicted individual treatment effects.

Results:The X4 had relatively more impact on model performance than X2 and X3 did. No effects were observed from X1. Moderate-to-severe outcome imbalance had a significantly negative impact on model performance, particularly among subgroups in which an outcome occurred. Bias in predicting individual treatment effects was significant and persisted even when the models had a 100% accuracy in predicting health outcome.

Discussion:Inadequate inclusion of the X2, X3, and X4 and moderate-to-severe outcome imbalance may affect model performance in predicting individual outcome and subsequently bias in predicting individual treatment effects. Machine learning models with all features and high performance for predicting individual outcome still yielded biased individual treatment effects.

Conclusions:Direct application of machine learning might not adequately address bias in predicting individual treatment effects. Further method development is needed to advance machine learning to support individualized treatment selection.

SUBMITTER: Fang G 

PROVIDER: S-EPMC7647181 | biostudies-literature | 2019 Oct

REPOSITORIES: biostudies-literature

altmetric image

Publications

Applying machine learning to predict real-world individual treatment effects: insights from a virtual patient cohort.

Fang Gang G   Annis Izabela E IE   Elston-Lafata Jennifer J   Cykert Samuel S  

Journal of the American Medical Informatics Association : JAMIA 20191001 10


<h4>Objective</h4>We aimed to investigate bias in applying machine learning to predict real-world individual treatment effects.<h4>Materials and methods</h4>Using a virtual patient cohort, we simulated real-world healthcare data and applied random forest and gradient boosting classifiers to develop prediction models. Treatment effect was estimated as the difference between the predicted outcomes of a treatment and a control. We evaluated the impact of predictors (ie, treatment predictors [X1], c  ...[more]

Similar Datasets

| S-EPMC7042174 | biostudies-literature
| S-EPMC5998124 | biostudies-literature
| S-EPMC6704548 | biostudies-literature
| S-EPMC6374273 | biostudies-literature
| S-EPMC7592688 | biostudies-literature
| S-EPMC9251754 | biostudies-literature
2018-04-07 | GSE112798 | GEO
| S-EPMC7809249 | biostudies-literature
| S-EPMC7885605 | biostudies-literature
2022-09-26 | PXD037076 |