# PheCode-guided multi-modal topic modeling of electronic health records improves disease incidence prediction and GWAS discovery from UK Biobank

## TL;DR

A new method called MixEHR-SAGE improves disease prediction and genetic discovery by combining multiple types of health data.

## Contribution

MixEHR-SAGE is a novel multi-modal topic model that enhances phenotyping from EHRs using probabilistic inference and expert priors.

## Key findings

- MixEHR-SAGE identified over 1000 interpretable phenotype topics from UK Biobank data.
- The model accurately predicted type 2 diabetes and leukemia diagnoses.
- Genome-wide association studies using MixEHR-SAGE found new disease-associated loci missed by traditional methods.

## Abstract

Phenome-wide association studies rely on disease definitions derived from diagnostic codes, often failing to leverage the full richness of electronic health records (EHR). We present MixEHR-SAGE, a PheCode-guided multi-modal topic model that integrates diagnoses, procedures, and medications to enhance phenotyping from large-scale EHRs. By combining expert-informed priors with probabilistic inference, MixEHR-SAGE identifies over 1000 interpretable phenotype topics from UK Biobank data. Applied to 350 000 individuals with high-quality genetic data, MixEHR-SAGE-derived risk scores accurately predict incident type 2 diabetes (T2D) and leukemia diagnoses. Subsequent genome-wide association studies using these continuous risk scores uncovered novel disease-associated loci, including PPP1R15A for T2D and JMJD6/SRSF2 for leukemia, that were missed by traditional binary case definitions. These results highlight the potential of probabilistic phenotyping from multi-modal EHRs to improve genetic discovery. The MixEHR-SAGE software is publicly available at: https://github.com/li-lab-mcgill/MixEHR-SAGE.