Machine learning prediction of physician-assigned treatment categories from pre-procedural electronic medical record data in cardiac patients: A multiclass classification study

Cardiovascular diseases (CVDs) remain a leading source of morbidity, mortality, and healthcare burden worldwide. In patients with coronary artery disease (CAD), choosing among medical therapy, percutaneous coronary intervention (PCI), and coronary artery bypass grafting (CABG) is clinically complex and depends on multiple pre-procedural factors. In this retrospective multicenter registry study, we developed a multiclass machine learning model to predict physician-assigned treatment categories rather than clinically optimal treatment. The analytic cohort included 2,682 unique patient records from the BioArc Clinical Registry, with class imbalance across medical therapy (n = 1,634), PCI (n = 817), and surgery (n = 231). Semi-structured JSON records were parsed to extract pre-decision demographic, clinical, symptom-related, and risk-factor variables. Missing data handling, encoding, SMOTE, recursive feature elimination (RFE), and hyperparameter tuning were performed within a nested stratified cross-validation workflow to reduce dependence on a single data split and to limit information leakage. The final XGBoost workflow retained 18 predictors after RFE. Across the outer folds of nested cross-validation, the model achieved an overall accuracy of 82.3% ± 1.5% (95% CI: 80.8%-83.8%) and a macro F1-score of 0.775 ± 0.018 (95% CI: 0.757-0.793), with one-vs-rest AUC values of 0.78 ± 0.02, 0.75 ± 0.02, and 0.80 ± 0.03 for medical therapy, PCI, and surgery, respectively. Class-specific performance and confusion patterns were summarized from outer-fold predictions. Model interpretation using SHAP and partial dependence plots identified age, systolic blood pressure, symptom-related features, and opium consumption as influential predictors in this dataset. These findings indicate that pre-procedural EMR data contain signal related to observed treatment-assignment patterns; however, the model does not identify the optimal treatment or demonstrate improved clinical outcomes. External validation, external calibration, and prospective outcome-based evaluation are required before clinical deployment.

Authors

Institutions

Publication Details

Journal
PLOS Digital Health
Published
2026-07-16
DOI
https://doi.org/10.1371/journal.pdig.0001149
Primary Topic
Machine Learning in Healthcare
Type
article
Field-Weighted Citation Impact
0.00
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

Machine learning prediction of physician-assigned treatment categories from pre-procedural electronic medical record data in cardiac patients: A multiclass classification study

Mohammad Tanhaei
PLOS Digital Health
Machine Learning in Healthcare
article

Machine learning prediction of physician-assigned treatment categories from pre-procedural electronic medical record data in cardiac patients: A multiclass classification study

Mohammad Tanhaei
article en

Abstract

Cardiovascular diseases (CVDs) remain a leading source of morbidity, mortality, and healthcare burden worldwide. In patients with coronary artery disease (CAD), choosing among medical therapy, percutaneous coronary intervention (PCI), and coronary artery bypass grafting (CABG) is clinically complex and depends on multiple pre-procedural factors. In this retrospective multicenter registry study, we developed a multiclass machine learning model to predict physician-assigned treatment categories rather than clinically optimal treatment. The analytic cohort included 2,682 unique patient records from the BioArc Clinical Registry, with class imbalance across medical therapy (n = 1,634), PCI (n = 817), and surgery (n = 231). Semi-structured JSON records were parsed to extract pre-decision demographic, clinical, symptom-related, and risk-factor variables. Missing data handling, encoding, SMOTE, recursive feature elimination (RFE), and hyperparameter tuning were performed within a nested stratified cross-validation workflow to reduce dependence on a single data split and to limit information leakage. The final XGBoost workflow retained 18 predictors after RFE. Across the outer folds of nested cross-validation, the model achieved an overall accuracy of 82.3% ± 1.5% (95% CI: 80.8%-83.8%) and a macro F1-score of 0.775 ± 0.018 (95% CI: 0.757-0.793), with one-vs-rest AUC values of 0.78 ± 0.02, 0.75 ± 0.02, and 0.80 ± 0.03 for medical therapy, PCI, and surgery, respectively. Class-specific performance and confusion patterns were summarized from outer-fold predictions. Model interpretation using SHAP and partial dependence plots identified age, systolic blood pressure, symptom-related features, and opium consumption as influential predictors in this dataset. These findings indicate that pre-procedural EMR data contain signal related to observed treatment-assignment patterns; however, the model does not identify the optimal treatment or demonstrate improved clinical outcomes. External validation, external calibration, and prospective outcome-based evaluation are required before clinical deployment.

PLOS Digital HealthVol. 5(7)
Ilam University (IR)
Good health and well-being
Openalex Percentile: Top 6%
Machine Learning in Healthcare
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.