Polygenic risk scores outperform machine learning methods in predicting coronary artery disease status

Genet Epidemiol. 2020 Mar;44(2):125-138. doi: 10.1002/gepi.22279. Epub 2020 Jan 10.

Abstract

Coronary artery disease (CAD) is the leading global cause of mortality and has substantial heritability with a polygenic architecture. Recent approaches of risk prediction were based on polygenic risk scores (PRS) not taking possible nonlinear effects into account and restricted in that they focused on genetic loci associated with CAD, only. We benchmarked PRS, (penalized) logistic regression, naïve Bayes (NB), random forests (RF), support vector machines (SVM), and gradient boosting (GB) on a data set of 7,736 CAD cases and 6,774 controls from Germany to identify the algorithms for most accurate classification of CAD status. The final models were tested on an independent data set from Germany (527 CAD cases and 473 controls). We found PRS to be the best algorithm, yielding an area under the receiver operating curve (AUC) of 0.92 (95% CI [0.90, 0.95], 50,633 loci) in the German test data. NB and SVM (AUC ~ 0.81) performed better than RF and GB (AUC ~ 0.75). We conclude that using PRS to predict CAD is superior to machine learning methods.

Keywords: classification; coronary artery disease; machine learning; polygenic risk scores; prediction.

Publication types

  • Research Support, Non-U.S. Gov't

MeSH terms

  • Bayes Theorem
  • Benchmarking
  • Coronary Artery Disease / genetics*
  • Databases, Genetic
  • Genetic Predisposition to Disease*
  • Humans
  • Machine Learning*
  • Models, Genetic
  • Multifactorial Inheritance / genetics*
  • Probability
  • ROC Curve
  • Support Vector Machine