AI Navigate

SNPgen: Phenotype-Supervised Genotype Representation and Synthetic Data Generation via Latent Diffusion

arXiv cs.LG / 3/12/2026

📰 NewsIdeas & Deep AnalysisModels & Research

Key Points

  • SNPgen introduces a two-stage conditional latent diffusion framework for generating phenotype-supervised synthetic genotypes by combining GWAS-guided variant selection with a variational autoencoder and a latent diffusion model conditioned on binary disease labels.
  • The method achieves train-on-synthetic, test-on-real predictive performance across four diseases on UK Biobank data, narrowing the gap to genome-wide PRS methods that use substantially more variants.
  • Privacy analysis indicates zero identical matches, near-random membership inference (AUC ~ 0.50), preserved linkage disequilibrium structure, and high allele frequency correlation with source data, supporting strong privacy properties.
  • Controlled simulations verify faithful recovery of the specified genetic association structure, demonstrating utility for downstream genetic analyses while preserving privacy.

Abstract

Polygenic risk scores and other genomic analyses require large individual-level genotype datasets, yet strict data access restrictions impede sharing. Synthetic genotype generation offers a privacy-preserving alternative, but most existing methods operate unconditionally, producing samples without phenotype alignment, or rely on unsupervised compression, creating a gap between statistical fidelity and downstream task utility. We present SNPgen, a two-stage conditional latent diffusion framework for generating phenotype-supervised synthetic genotypes. SNPgen combines GWAS-guided variant selection (1,024-2,048 trait-associated SNPs) with a variational autoencoder for genotype compression and a latent diffusion model conditioned on binary disease labels via classifier-free guidance. Evaluated on 458,724 UK Biobank individuals across four complex diseases (coronary artery disease, breast cancer, type 1 and type 2 diabetes), models trained on synthetic data matched real-data predictive performance in a train-on-synthetic, test-on-real protocol, approaching genome-wide PRS methods that use 2-6\times more variants. Privacy analysis confirmed zero identical matches, near-random membership inference (AUC \approx 0.50), preserved linkage disequilibrium structure, and high allele frequency correlation (r \geq 0.95) with source data. A controlled simulation with known causal effects verified faithful recovery of the imposed genetic association structure.