Mol-JEPA: A multimodal Joint Embedding Predictive Architecture for Molecules
4 affiliations
- 1Boehringer Ingelheim
- 2University of Tübingen
- 3The University of Texas at Austin
- 4Brown University
Abstract
Despite recent advances in molecular foundation models, several limitations remain, such as chemically invalid augmentations, modality collapse, and incomplete representation of biochemical environments. To address these challenges, we present \textbf{Mol-JEPA}, a scalable framework for learning molecular world models. Rather than relying on suboptimal molecular perturbations, our model uses modality masking to exploit information from molecular structures, cellular phenotypes, binding affinities, ADMET profiles, quantum chemistry simulations and other drug discovery data. Across various benchmarks, we show that the representations learned by Mol-JEPA deliver strong performance, demonstrating the value of incorporating biochemical context through latent space prediction.
Related
- Multimodal computer vision and machine learning for predictive monitoring of planar-flow casting of amorphous steel foils
- From Static Material Design to Predictive Open-Loop Control via Chemistry–Architecture Coupling in a Moving-Boundary Reaction–Diffusion Framework
- Scalable Density-Based Bootstrap Embedding with Perturbative Triples for Large Molecular Systems
- Generalized local potential functional embedding theory of localized orbitals
- POLEN: An OpenMM Framework for ML/MM Simulations with Mutually Polarizable AMOEBA Embedding
BibTeX
@misc{rottach2026moljepa,
title = {{Mol-JEPA: A multimodal Joint Embedding Predictive Architecture for Molecules}},
author = {Florian Rottach and Sebastian Schieferdecker and William Rudman and Randall Balestriero and Carsten Eickhoff},
howpublished = {arXiv (Cornell University)},
year = {2026},
month = aug,
doi = {10.48550/arxiv.2608.22642},
url = {https://arxiv.org/abs/2608.22642},
note = {CC BY},
}
RIS
TY - UNPB
TI - Mol-JEPA: A multimodal Joint Embedding Predictive Architecture for Molecules
AU - Florian Rottach
AU - Sebastian Schieferdecker
AU - William Rudman
AU - Randall Balestriero
AU - Carsten Eickhoff
PY - 2026
DA - 2026/08/23
PB - arXiv (Cornell University)
DO - 10.48550/arxiv.2608.22642
UR - https://arxiv.org/abs/2608.22642
AB - Despite recent advances in molecular foundation models, several limitations remain, such as chemically invalid augmentations, modality collapse, and incomplete representation of biochemical environments. To address these challenges, we present \textbf{Mol-JEPA}, a scalable framework for learning molecular world models. Rather than relying on suboptimal molecular perturbations, our model uses modality masking to exploit information from molecular structures, cellular phenotypes, binding affinities, ADMET profiles, quantum chemistry simulations and other drug discovery data. Across various benchmarks, we show that the representations learned by Mol-JEPA deliver strong performance, demonstrating the value of incorporating biochemical context through latent space prediction.
SN - 2331-8422
ER -