GenoBERT: A Language Model for Accurate Genotype Imputation

arXiv cs.AI / 4/2/2026

💬 OpinionIdeas & Deep AnalysisModels & Research

Key Points

  • GenoBERT is a transformer-based, reference-free genotype imputation framework that tokenizes phased genotypes and uses self-attention to model short- and long-range linkage disequilibrium (LD).
  • Benchmarks on the Louisiana Osteoporosis Study (LOS) and 1000 Genomes Project (1KGP) across ancestries and varying missingness (5–50%) show GenoBERT outperforms four baselines, with very high overall accuracy at practical sparsity levels.
  • The method maintains strong performance even at high missingness (50%) and continues to deliver consistent gains across ancestry groups, including settings with weak LD and limited sample sizes.
  • A 128-SNP context window (about 100 kb) is supported by LD-decay analyses as sufficient to capture local correlation structure for accurate imputation.
  • By removing dependence on reference panels while retaining high accuracy, GenoBERT is positioned as a scalable approach that could improve downstream genomic risk prediction and association analyses.

Abstract

Genotype imputation enables dense variant coverage for genome-wide association and risk-prediction studies, yet conventional reference-panel methods remain limited by ancestry bias and reduced rare-variant accuracy. We present Genotype Bidirectional Encoder Representations from Transformers (GenoBERT), a transformer-based, reference-free framework that tokenizes phased genotypes and uses a self-attention mechanism to capture both short- and long-range linkage disequilibrium (LD) dependencies. Benchmarking on two independent datasets including the Louisiana Osteoporosis Study (LOS) and the 1000 Genomes Project (1KGP) across ancestry groups and multiple genotype missingness levels (5-50%) shows that GenoBERT achieves the highest overall accuracy compared to four baseline methods (Beagle5.4, SCDA, BiU-Net, and STICI). At practical sparsity levels (up to 25% missing), GenoBERT attains high overall imputation accuracy (r^2 approx 0.98) across datasets, and maintains robust performance (r^2 > 0.90) even at 50% missingness. Experimental results across different ancestries confirm consistent gains across datasets, with resilience to small sample sizes and weak LD. A 128-SNP (single-nucleotide polymorphism) context window (approximately 100 Kb) is validated through LD-decay analyses as sufficient to capture local correlation structures. By eliminating reference-panel dependence while preserving high accuracy, GenoBERT provides a scalable and robust solution for genotype imputation and a foundation for downstream genomic modeling.