HELICS: Biobank-scale Conditional Synthetic Genome Generation via Latent Flow Matching
Organizations: Department of Biostatistics, Harvard University, Boston, MA 02135 · Department of Statistics, Biostatistics, Harvard University, Boston, MA 02135
Abstract
Individual-level genotype data underpin genetic discovery, disease-risk modeling, and therapeutic development, yet many critical settings remain data-limited, including underrepresented ancestries, rare or imbalanced disease cohorts, and restricted-access biobanks. Conditional synthetic genome generation would support privacy-aware benchmarking, simulation, and data augmentation, but requires learning an ultra-high-dimensional discrete distribution over hundreds of thousands to tens of millions of correlated single-nucleotide polymorphisms (SNPs), or sites of DNA variation. We introduce HELICS: High-fidelity gEneration via Latent Interpolant Conditional Synthesis, a phasing-free latent flow matching framework for genome-wide SNP-level generation from biobank-scale genotype data. HELICS tokenizes chromosomes with convolutional autoencoders, reducing >600K SNPs to 4,096 continuous latent tokens, and trains a transformer parameterized conditional flow matching model over the concatenated multi-chromosome latent representation. Trained on over 417K genomes in the UK Biobank, HELICS generates genome-wide common-variant synthetic cohorts without haplotype phasing and achieves stronger fidelity to real genotypes than existing haplotype-based simulators, including high allele-frequency agreement in external evaluation and superior preservation of local and long-range SNP correlation structure. Conditioning on ancestry and genetic liability summaries enables targeted generation across genetic risk profiles; in perturbation analyses, by increasing the conditioned disease trait liability, HELICS induces SNP-level dosage changes in generated genomes that are highly correlated with published association weights. HELICS provides a scalable generative modeling framework for SNP-level genome data and a path toward conditional synthetic cohorts for data-limited genetic studies.
Figures & tables
| Approach | GW | No. SNPs | MAF | Low MAF | MAF MAE | LD diff. | Membership | Gen. NN IBS |
|---|---|---|---|---|---|---|---|---|
| HAPNEST [ 56 ] | ✓ | 10,509 | 0.942 | 0.207 | 0.025 | 0.101 | 0.030 | 0.914 |
| Sim1000G [ 19 ] | ✗ | 9,899 | 0.988 | 0.637 | 0.011 | 0.122 | 0.011 | 0.839 |
| HAPGEN2 [ 50 ] | ✗ | 17,117 | 0.976 | 0.444 | 0.016 | 0.069 | 0.005 | 0.879 |
| GAN-based [ 60 ] | ✗ | 6,513 | 0.947 | 0.129 | 0.023 | 0.129 | 0.060 | 0.923 |
| RBM-based [ 60 ] | ✗ | 7,531 | 0.829 | 0.191 | 0.032 | 0.110 | 0.001 | 0.927 |
| HCLT-based [ 18 ] | ✗ | 6,827 | 0.846 | 0.140 | 0.036 | 0.152 | 0.029 | 0.906 |
| Approach | Scope | No. SNPs | MAF | Low MAF | MAF MAE | LD diff. | Membership | Gen. NN IBS |
|---|---|---|---|---|---|---|---|---|
| HAPNEST [ 56 ] | Genome-wide | 349,564 | ||||||
| Sim1000G [ 19 ] | Regional | |||||||
| HELICS | Genome-wide | 600,336 | ||||||
| HELICS-EUR | Genome-wide | 600,336 |
| Trait | Variants | Gen-Obs. | Bin 10 1 | Top-20 |
|---|---|---|---|---|
| HDL | 168 | 0.66 | 0.98 | 0.98 |
| LDL | 606 | 0.75 | 1.00 | 0.94 |
| T1D | 16,570 | 0.63 | 1.00 | 1.00 |
| RA | 14,728 | 0.71 | 0.99 | 1.00 |
| ApoB | 43,001 | 0.71 | 0.95 | 0.80 |
Appendix figures & tables24 assets
Supplementary material from the paper’s appendix.
Appendix
| Chromosome | |||||
|---|---|---|---|---|---|
| 1 | 47,555 | 49,152 | 3 | 256 | 6 |
| 2 | 48,530 | 49,152 | 3 | 256 | 6 |
| 3 | 40,846 | 40,960 | 5 | 256 | 5 |
| 4 | 38,621 | 40,960 | 5 | 256 | 5 |
| 5 | 36,643 | 40,960 | 5 | 256 | 5 |
| 6 | 42,638 | 49,152 | 3 | 256 | 6 |
| Hyperparameter | Value |
|---|---|
| Optimizer | AdamW, |
| Learning rate | |
| Learning-rate schedule | None |
| Batch size | |
| Number of epochs | 50 |
| Embedding dimension |
| Parameter | Value |
|---|---|
| Hardware | NVIDIA H200 GPUs |
| Training budget | 72 Hours |
| Input latent shape | |
| Backbone | USiT |
| Transformer hidden width | |
| Transformer layers |
| Trait | PGS Catalog accession |
|---|---|
| Standing height | PGS002332 |
| Coronary artery disease | PGS003725 |
| Alzheimer’s disease (dementia) | PGS003958 |
| Major depression | PGS003333 |
| Body mass index | PGS005198 |
| High-density lipoprotein cholesterol | PGS004914 |
| PGS002960 (external) variant set | Variants scored | Generated-observed | Real cross-score |
|---|---|---|---|
| All matched variants | 2,627 | 0.189 | 0.300 |
| After LD exclusions | 2,608 | 0.135 | 0.194 |
| Training | Training chr. | MAF | Low-MAF | MAF MAE | LD diff. |
|---|---|---|---|---|---|
| 2,500 | 22 | 0.978 | 0.187 | 0.0199 | 0.205 |
| 5,000 | 22 | 0.975 | 0.169 | 0.0220 | 0.221 |
| 10,000 | 22 | 0.968 | 0.179 | 0.0284 | 0.226 |
| 100,000 | 22 | 0.983 | 0.368 | 0.0173 | 0.210 |
| 251,465 | 1-22 | 0.991 | 0.678 | 0.018 | 0.120 |
| Flow model (parameter count) | MAF | Low-MAF | MAF MAE | LD diff. |
|---|---|---|---|---|
| Tiny (1.18M) | 0.221 | 0.002 | 0.187 | 0.251 |
| Small (7.98M) | 0.668 | 0.013 | 0.112 | 0.257 |
| Medium (25.5M) | 0.978 | 0.219 | 0.027 | 0.249 |
| Large (58.9M) | 0.983 | 0.260 | 0.024 | 0.248 |
| Reference | MAF | MAF MAE | Low-MAF | LD diff. |
|---|---|---|---|---|
| 1000G-EUR | ||||
| UKB-EUR |
| Synthetic samples vs. training reference | ||||
|---|---|---|---|---|
| Relatedness category | Sim1000G | HELICS | HAPGEN2 | HAPNEST |
| MZ twins | 0 | 0 | 0 | 0 |
| 1st degree | 0 | 0 | 0 | 0 |
| 2nd degree | 0 | 0 | 0 | 0 |
| 3rd degree | 15,656 | 21,645 | 1,779 | 29,114 |
| Unrelated | 4,984,344 | 24,978,355 | 24,998,221 | 24,970,886 |
| Chr | MAF | Low-MAF | MAF | MAF MAE | LD diff. | LD blocks |
|---|---|---|---|---|---|---|
| 1 | 0.9805 | 0.4967 | 0.9205 | 0.0248 | 0.1414 | 753 |
| 2 | 0.9735 | 0.5144 | 0.9037 | 0.0229 | 0.1452 | 808 |
| 3 | 0.9765 | 0.5307 | 0.9117 | 0.0269 | 0.1393 | 639 |
| 4 | 0.9774 | 0.5203 | 0.9106 | 0.0258 | 0.1327 | 631 |
| 5 | 0.9739 | 0.4972 | 0.8955 | 0.0262 | 0.1327 | 597 |
| 6 | 0.9793 | 0.6279 | 0.9197 | 0.0261 | 0.1362 | 636 |
| Chr | MAF | Low-MAF | MAF | MAF MAE | LD diff. | LD blocks |
|---|---|---|---|---|---|---|
| 1 | 0.9790 | 0.2032 | 0.8503 | 0.0330 | 0.1616 | 753 |
| 2 | 0.9876 | 0.5716 | 0.9566 | 0.0179 | 0.1082 | 808 |
| 3 | 0.9871 | 0.8617 | 0.9830 | 0.0170 | 0.0908 | 639 |
| 4 | 0.9750 | 0.7342 | 0.9585 | 0.0232 | 0.0909 | 631 |
| 5 | 0.9886 | 0.3162 | 0.9159 | 0.0240 | 0.1428 | 597 |
| 6 | 0.9868 | 0.7083 | 0.9546 | 0.0168 | 0.1113 | 636 |
| Ancestry | MAF | MAF MAE | Low-MAF | LD diff. | |
|---|---|---|---|---|---|
| EUR | 391,125 | ||||
| SAS | 9,033 | ||||
| EAS | 5,576 | ||||
| AFR | 7,291 |
| Ancestry | Sliced | RBF MMD 2 | |
|---|---|---|---|
| White | 391,125 | 0.002 0.0001 | 0.002 0.0003 |
| South Asian | 9,033 | 0.042 0.001 | 0.315 0.016 |
| SE Asian | 5,576 | 0.078 0.003 | 0.331 0.016 |
| Black | 4,230 | 0.075 0.002 | 0.846 0.037 |
| African | 3,061 | 0.085 0.003 | 0.564 0.016 |
| Trait | Ancestry | Matched PGS variants | Gen.-obs. | Bin 10 bin 1 mean | Top-20 alignment | ||
|---|---|---|---|---|---|---|---|
| HDL | EUR | 9,366 | 950 | 168 | 0.684 | 0.977 | 0.950 |
| HDL | SAS | 213 | 28 | 168 | 0.230 | 1.000 | 0.950 |
| HDL | AFR | 172 | 23 | 168 | 0.244 | 0.913 | 0.900 |
| LDL | EUR | 9,366 | 950 | 606 | 0.762 | 0.997 | 0.800 |
| LDL | SAS | 213 | 28 | 606 | 0.221 | 1.000 | 0.750 |
| LDL | AFR | 172 | 23 | 606 | 0.308 | 0.957 | 0.850 |
Synthetic Data Generation
Synthetic data generation creates artificial examples, simulations, and datasets to train, test, or protect machine-learning systems.
See all 625 papers