JEPA-DNA: Grounding Genomic Foundation Models through Joint-Embedding Predictive Architectures
Not on the map. Why?
Abstract (arXiv)
Genomic Foundation Models (GFMs) typically rely on Masked Language Modeling (MLM) or Next-Token Prediction (NTP) to learn the "Laws of Nature". While effective at capturing local syntax, these generative paradigms prioritize token-level reconstruction over high-level functional context. We introduce JEPA-DNA, a model-agnostic continual training framework that integrates a Joint-Embedding Predictive Architecture (JEPA) with traditional generative objectives. By supervising global sequence embeddings in a latent space, JEPA-DNA forces models to predict the functional representations of masked genomic segments, shifting the learning signal from token recovery to semantic alignment. We evaluate JEPA-DNA on 17 diverse genomic benchmark tasks, demonstrating consistent gains in linear probing and zero-shot performance regardless of the underlying GFM architecture or generative objective, and establishing a new state-of-the-art for GFMs under the linear-probing protocol. Through extensive ablation studies, we further characterize the synergistic interplay between generative and latent objectives. Our code is provided in the supplementary material.
Key facts. Cited by 9 papers in the map (data through 14 Oct 2026). Builds on 11 of its 46 references that are in the map. First posted on arXiv 19 Feb 2026.
Builds on
11 of 46 references.
References · 11
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
- 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 12,049
- 2021VICReg: Variance-Invariance-Covariance Regularization for Self-Supervised Learningcited by 1,057
- 2023Self-Supervised Learning from Images with a Joint-Embedding Predictive Architecturecited by 1,039
- 2020AdaBelief Optimizer: Adapting Stepsizes by the Belief in Observed Gradientscited by 283
- 2025LeJEPA: Provable and Scalable Self-Supervised Learning Without the Heuristicscited by 254
- 2023HyenaDNA: Long-Range Genomic Sequence Modeling at Single Nucleotide Resolutioncited by 194
- 2023DNABERT-2: Efficient Foundation Model and Benchmark For Multi-Species Genomecited by 143
- 2025LLM-JEPA: Large Language Models Meet Joint Embedding Predictive Architecturescited by 38
- 2023BEND: Benchmarking DNA Language Models on biologically meaningful taskscited by 29
- 2024Genomic Language Models: Opportunities and Challengescited by 20
Built on by
Citing papers per year: 2026: 9.
9 papers.
- 2026Learning Clinical Representations Under Systematic Distribution Shiftcited by 2
- 2026Discriminative Representation Learning for Clinical Predictioncited by 2
- 2026Uncertainty-Aware Foundation Models for Clinical Datacited by 1
- 2026ProteinJEPA: Latent prediction improves protein language model pretrainingcited by 0
- 2026Event Fields: Learning Latent Event Structure for Waveform Foundation Modelscited by 0
- 2026WISTERIA: Learning Clinical Representations from Noisy Supervision via Multi-View Consistency in Electronic Health Recordscited by 0
- 2026AURORA: Contextual Orthogonalization for Geometric Representation Learning in Healthcare Foundation Modelscited by 0
- 2026MorphologyFM: A Foundation Model for Morphology-Aware Representation Learning from ECG and Pulse Oximetry Waveformscited by 0
- 2026VANDAM: Viewing a nucleotide sequence with DNA molecular priorscited by 0
- 2026VANDAM: Viewing a nucleotide sequence with DNA molecular priors30 Sep 2026
- 2026MorphologyFM: A Foundation Model for Morphology-Aware Representation Learning from ECG and Pulse Oximetry Waveforms3 Jul 2026
- 2026AURORA: Contextual Orthogonalization for Geometric Representation Learning in Healthcare Foundation Models18 May 2026
- 2026WISTERIA: Learning Clinical Representations from Noisy Supervision via Multi-View Consistency in Electronic Health Records10 May 2026
- 2026Event Fields: Learning Latent Event Structure for Waveform Foundation Models9 May 2026
- 2026ProteinJEPA: Latent prediction improves protein language model pretraining8 May 2026
- 2026Uncertainty-Aware Foundation Models for Clinical Data5 Apr 2026
- 2026Discriminative Representation Learning for Clinical Prediction21 Mar 2026
- 2026Learning Clinical Representations Under Systematic Distribution Shift7 Mar 2026