Use machine learning models to identify and assess risk factors for coronary artery disease

Mingyang Zhang, Hongnian Wang and Ju Zhao

PLOS ONE, 2024, vol. 19, issue 9, 1-15

Abstract: Accurate prediction of coronary artery disease (CAD) is crucial for enabling early clinical diagnosis and tailoring personalized treatment options. This study attempts to construct a machine learning (ML) model for predicting CAD risk and further elucidate the complex nonlinear interactions between the disease and its risk factors. Employing the Z-Alizadeh Sani dataset, which includes records of 303 patients, univariate analysis and the Boruta algorithm were applied for feature selection, and nine different ML techniques were subsequently deployed to produce predictive models. To elucidate the intricate pathogenesis of CAD, this study harnessed the analytical capabilities of Shapley values, alongside the use of generalized additive models for curve fitting, to probe into the nonlinear interactions between the disease and its associated risk factors. Furthermore, we implemented a piecewise linear regression model to precisely pinpoint inflection points within these complex nonlinear dynamics. The findings of this investigation reveal that logistic regression (LR) stands out as the preeminent predictive model, demonstrating remarkable efficacy, it achieved an Area Under the Receiver Operating Characteristic curve (AUROC) of 0.981 (95% CI: 0.952–1), and an Area Under the Precision-Recall Curve (AUPRC) of 0.993. The utilization of the 14 most pivotal features in constructing a dynamic nomogram. Analysis of the Shapley smoothing curves uncovered distinctive “S”-shaped and “C”-shaped relationships linking age and triglycerides to CAD, respectively. In summary, machine learning models could provide valuable insights for the early diagnosis of CAD. The SHAP method may provide a personalized risk assessment of the relationship between CAD and its risk factors.

Date: 2024
References: Add references at CitEc
Citations:

Downloads: (external link)
https://journals.plos.org/plosone/article?id=10.1371/journal.pone.0307952 (text/html)
https://journals.plos.org/plosone/article/file?id= ... 07952&type=printable (application/pdf)

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:plo:pone00:0307952

DOI: 10.1371/journal.pone.0307952

Access Statistics for this article

More articles in PLOS ONE from Public Library of Science
Bibliographic data for series maintained by plosone ().