Identifying subtypes of heart failure with machine learning: external, prognostic and genetic validation in three electronic health record sources with 320,863 individuals

Background Reliable identification of heart failure (HF) subtypes might allow targeted management. Machine learning (ML) has been used to explore HF subtypes, but neither across large, independent, population-based datasets, nor across the full spectrum of causes and presentations, nor with clinical and non-clinical validation by different ML methods. Using our published framework, we identified and validated HF subtypes to address these gaps.

Methods We analysed individuals ≥30 years with incident HF from two population-based electronic health records resources (1998-2018; Clinical Practice Research Datalink, CPRD: n=188,799 HF cases; The Health Improvement Network, THIN: n=124,263 HF cases). Pre-and post-HF factors (n=645) included demography, history, examination, blood laboratory values and medications. We identified subtypes using four unsupervised ML methods (K-means, hierarchical, K-Medoids and mixture model clustering) with 87 (from 645) factors in each dataset. We evaluated subtypes for: (i) external validity (across independent datasets); (ii) prognostic validity (predictive accuracy for 1-year mortality); and (iii) uniquely, genetic validity (in UK Biobank; n=9573 cases): association with polygenic risk score (PRS) for 11 HF related traits, and direct association with 12 reported HF single nucleotide polymorphisms (SNPs).

Findings After identifying five clusters, we labelled HF subtypes: 1.Early-onset, 2.Late-onset, 3.AF-related, 4.Metabolic, and 5.Cardiometabolic. External validity: Subtypes were similar across datasets (c-statistic: 0.94, 0.80, 0.79, 0.83, 0.92 for the THIN model in CPRD and 0.79, 0.92, 0.90, 0.89, 0.92 for the CPRD model in THIN for subtypes 1-5, respectively). Prognostic validity: One-year all-cause mortality, risk of non-fatal cardiovascular diseases and all-cause hospitalisation (before and after HF diagnosis) differed across subtypes in CPRD and THIN data. Genetic validity: The AF-related subtype showed associations with PRS for related traits. Late-onset and Cardiometabolic subtypes were most comparable and strongly associated with PRS for Hypertension, Myocardial Infarction and Obesity (p-value < 9.09 × 10⁻⁴). We developed a prototype for clinical use, which could enable evaluation of effectiveness and cost-effectiveness.

Interpretation Across four methods and three datasets, and including genetic data, in the largest HF study to-date, ML algorithms identified five subtypes in individuals with incident HF. These subtypes may inform aetiologic research, clinical risk prediction and the design of HF trials.

Funding European Union Innovative Medicines Initiative.

Evidence before this study In a systematic review until December 2019, we showed that studies of machine learning in subtyping and risk prediction in cardiovascular diseases are limited by small population size, relatively few factors and poor generalisability of findings due to lack of external validation. We further searched PubMed, medRxiv, bioRxiv, arXiv, for relevant peer-reviewed articles and preprints, focusing on machine learning studies in heart failure. Studies remain focused on single diseases, limited risk factors, often single method of machine learning, rarely use subtyping and risk prediction together, and have not been externally validated across datasets. For heart failure, all subtype discovery studies have identified subtypes based on clustering, but so far with no application to clinical practice.

Added value of this study Across two independent, population-based datasets, we used four machine learning methods for subtyping and risk prediction with 89 aetiologic factors as well as 556 further factors for heart failure. We identified and validated five subtypes in incident heart failure, which differentially predicted outcomes. In addition, we externally validated clinical cluster differences by exploring corresponding genetic differences in a large-scale genetic cohort. Our methods and results highlight potential value of electronic health records and machine learning in understanding disease subtypes. Moreover, our approach to external, prognostic, and genetic validity provides a framework for validation of machine learning approaches for disease subtype discovery.

Implications of all the available evidence Our analyses support coordinated use of large-scale, linked electronic health records to identify and validate disease subtypes with relevance for clinical risk prediction, patient selection for trials, and future genetic research.