Unsupervised feature extraction using deep learning empowers discovery of genetic determinants of the electrocardiogram.

Abstract

Background: Electrocardiograms (ECGs) are widely used to assess cardiac health, but traditional clinical interpretation relies on a limited set of human-defined parameters. While advanced data-driven methods can outperform analyses of conventional ECG features for some tasks, they often lack interpretability. Variational autoencoders (VAEs), a form of unsupervised machine learning, can address this limitation by extracting ECG features that are both comprehensive and interpretable, known as latent factors. These latent factors provide a low-dimensional representation optimised to capture the full informational content of the ECG. The aim of this study was to develop a deep learning model to learn these latent ECG features, and to use this optimised feature set in genetic analyses to identify fundamental determinants of cardiac electrical function. This approach has the potential to expand our understanding of cardiac electrophysiology by uncovering novel phenotypic and genetic relationships.

Methods: Our novel VAE model was trained on a dataset comprising over one million secondary care median beat ECGs, with external validation in the UK Biobank (UKB). We performed common and rare variant association studies for VAE latent factors and conventional ECG traits on quality-controlled UKB data. Associated genetic variants were compared to loci for conventional ECG parameters available in the UKB and literature. Loci were considered novel if they were not previously associated with ECG traits in the GWAS Catalog and showed no known associations in nearby genes based on literature review. Novel GWAS associations were validated in a withheld subset of the UKB cohort. Additionally, we compared the associations of the VAE latent factors and conventional ECG traits with phenotypic traits, disease codes, and echocardiographic traits.

Results: The VAE identified 20 independent latent factors that captured ECG morphology with high accuracy (mean Pearson correlation: 0.95). GWAS of latent factors identified 65 unique loci, including 27 novel regions not associated with conventional ECG parameters in the same dataset. Six novel loci were not associated with the ECG in previous larger GWAS studies, including genes implicated in cardiac function and remodelling. Rare variant analysis identified seven additional genes with links to cardiac electrophysiology and remodelling. Phenotypic analyses revealed stronger and more comprehensive associations for latent factors compared to conventional traits, particularly for echocardiographic measures and cardiac phenotypes. Visualisations of latent factor alterations highlighted the interpretability of this approach.

Conclusions: Our study shows that the VAE provides a valuable tool for advancing our understanding of cardiac function and its genetic underpinnings, outperforming traditional approaches in genetic and phenotypic discovery.