Pith. Sign in

REVIEW

Mol-JEPA: A multimodal Joint Embedding Predictive Architecture for Molecules

Machine Learning · Submitted 23 Aug 2026

Pith's verdict: Mol-JEPA is a solid, unusually transparent multimodal pretraining contribution with real ablations and open code, but its headline numbers are inflated by oracle probe selection and the paper never checks whether its 4.7M-molecule corpus contains the benchmark molecules it later evaluates on.

T0 · AI review

How this review was made: An AI referee (deepseek-v4-flash) read the paper on August 27, 2026 and answered a fixed list of questions about it. T0 is the starting level every paper gets. The levels describe how deeply Pith has checked the paper, never the authors.

  1. T0 AI review An AI read the paper and answered a fixed list of questions.
  2. T1 Audit Automatic checks that the references and files the paper relies on exist and match.
  3. T2 Compute The paper's code or calculations re-run with the same results.
  4. T3 Formal statement The claims written out in a formal math language a computer can read.
  5. T4 Proof Those claims proved, and the proof checked by computer.

How Pith checks papers →

· 3 major objections · 6 minor · Ask about this paper · The compiled record · Read the full review →
Challenge this review Comment (0) Read on arXiv ↗
Cite ▾
arXiv ID2608.22642 (version 4)
Pith IDOC55K2EJ, Pith's permanent record of this paper and its review
CitationCarsten Eickhoff, Florian Rottach, Randall Balestriero, Sebastian Schieferdecker, William Rudman. "Mol-JEPA: A multimodal Joint Embedding Predictive Architecture for Molecules." arXiv:2608.22642 (2026). https://pith.science/paper/OC55K2EJ
Cite this reviewBibTeX for Pith's review of this paper
Claim paper

Why Pith says this

Highlight any text to comment on it.

Referee report
The paper

The paper claims

Predicting masked biochemical views beats structure-only molecular models

The reading

This paper tries to establish that a molecular representation is stronger when it is trained to predict the latent embeddings of many biochemical modalities—structure, cell-painting phenotype, binding affinity, ADMET measurements, quantum-chemical descriptors—than when it is trained from molecular structure alone. The proposed model, Mol-JEPA, masks whole modalities during pretraining and reconstructs their embeddings in latent space, sidestepping chemically invalid augmentations and contrastive negative sampling. On open benchmark suites, the learned representations beat standard baselines on average and show their largest edge on small datasets and on molecules far from the training distribution. That matters for drug discovery, where labeled data are scarce, noisy, and expensive to collect.

What carries the argument

The load-bearing mechanism is modality masking inside a Joint Embedding Predictive Architecture. Rather than perturbing a molecule's graph or SMILES string, Mol-JEPA randomly deactivates entire modalities—each modality is one of 14 views, such as a cell-painting embedding, a binding-affinity vector, or an ADMET label vector—and trains a Transformer encoder to predict the masked latent embeddings from the remaining ones. Three learnable encoders (atom, graph, vector) project all modalities into a shared space, and SIGReg applies an Epps-Pulley isotropy test on random 1D projections of the embeddings to prevent representation collapse. The prediction error is the mean squared error $\|\hat{s}^m_i - \hat{s}^m_{i,\mathrm{pred}}\|_2^2$ summed over masked modalities.

Extended reading notes

Core claim

On its own terms, the paper's central claim is that modality-masked latent prediction creates molecular world models. Mol-JEPA combines 14 views of a molecule, from frozen embeddings of upstream models (atomistic, SMILES, cell-painting, binding-affinity predictors), computed descriptors (MOE, ECFP, GFN2-xTB, DFT), and experimental label vectors (ChEMBL, PCBA, TDC), across 4.69 million curated compounds. During training a Transformer predictor sees a random subset of encoded modalities and predicts the mean-squared error of the masked ones; Sketched Isotropic Gaussian Regularization (SIGReg) keeps the latent space from collapsing. The authors report that the best Mol-JEPA variant reaches an average MAE of 0.402 across 22 benchmark tasks, below every baseline (best baseline, TabICLv2 with AlvaDesc descriptors, 0.453), and that the advantage grows on small benchmark families and with Tanimoto distance from the training set. They also report that adding modalities consistently improves downstream performance and that removing the experimental modalities degrades it.

Load-bearing premise

The claim rests on the premise that the frozen embeddings and experimental label vectors used as 'views' of a molecule are trustworthy enough that predicting them teaches real chemistry instead of reproducing upstream artifacts, noise, or redundant biases.

Editorial extensions

If this is right

  • Adding modalities improves downstream performance; training time grows roughly linearly with modality count, so the framework can absorb new data sources predictably.
  • Mol-JEPA's edge is largest on low-data benchmarks (ASAP/Polaris, Biogen ADME) and on molecules with low Tanimoto similarity to the training set, which is the regime drug-discovery teams actually face.
  • Full multimodal pretraining cuts downstream MAE by about 14% (linear probe) and 13% (nonlinear probe) relative to training on molecular graph and ECFP4 fingerprints alone.
  • In leave-one-modality-out training, the graph modality contributes most, but experimental modalities still add value, supporting the paper's thesis that biochemical context is informative beyond structure.
  • A lightweight Transformer probe over modality-specific embeddings and TabICLv2 over CLS tokens are the strongest downstream uses, while LoRA fine-tuning underperforms.

Reading between the lines

Editorial extensions of the paper, not claims the author makes directly.

  • Our inference: because the framework never invents molecular edits, it should transfer to other data-scarce scientific domains where multiple noisy, partially paired readouts of the same object exist, such as materials synthesis or cellular screening.
  • Our inference: Mol-JEPA's ceiling is set by the quality of its frozen upstream embeddings; replacing the atomistic, cell-painting, or binding-affinity encoders with newer models should improve the learned representations with no architectural change.
  • Our inference: the reconstruction objective may act as a denoiser that is robust to label noise, but a direct test—scrambling one modality's embeddings across molecules and measuring downstream degradation—would separate genuine cross-modal signal from mere pretraining regularization.
  • Our inference: the reported gain on small datasets suggests Mol-JEPA embeddings could serve as a fixed feature extractor for active-learning loops, where assay budget, not model capacity, is the constraint.
Original 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.

The compiled record

The paper's own sentences, arranged, kept beside the review and separate from it. Each label is drawn from a closed vocabulary; the number beside it is the probability the classifier gave that label. Numbers in the tables are the authors', checked by code. No sentence in this record was written by a model.

Compiled by classifier-1.13.0 via TypeSafe API, question bank abstract-p1-20260923, 2026-09-23: 1 request, 3,638 input tokens.

What the abstract itself states, as probabilities: claims first 0.04 · states limitation 0.12 · human participants 0.05 · controlled comparison 0.14 · randomized assignment 0.01 · states quantitative result 0.04 · compares with prior results 0.34 · mentions code or data availability 0.03.

The abstract, sentence by sentence

arXiv listing abstract v4. Results and the main claim are underlined; .

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.

background contribution method result interpretation main claim

What the classifier reports for each sentence. The number is the probability it gave the label.
#sentencerolethe authors' own workstates the main conclusionstrength
s0 Despite recent advances in molecular foundation models, several limitations remain, such as chemically invalid augmentations, modality collapse, and incomplete representation of biochemical environments. background 0.61 0.07 0.03 asserted 1.00
s1 To address these challenges, we present \textbf{Mol-JEPA}, a scalable framework for learning molecular world models. contribution 1.00 0.78 0.07 asserted 0.96
s2 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. method 0.98 0.83 0.08 asserted 1.00
s3 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. result 0.94 0.96 0.78 asserted 1.00

What kind of paper the classifier reports this is: method or tool 1.00 · evidence from simulation or computation 0.56.

What was examined, and what was not

sources read
  • arXiv listing abstract v4
not examined
  • full text
  • figures and tables
  • references and citing sentences
  • relations to other papers
levels run
  • P abstract
not run
  • S sections
  • T body sentences
  • C citations
  • R cross-paper relations

Every quoted sentence on this record is the authors'. Every label is from Pith's closed vocabulary, with the probability shown beside it. Every number in the measurements table is the authors', checked by code. Record as JSON

Figures
Figure 1. Mol-JEPA: Rather than representing molecules solely through their structure, our model learns from a spectrum of biochemical and physical modalities. During training, modalities are randomly masked and predicted from the remaining inputs using a Transformer-based predictor. To prevent representation collapse, we employ isotropic regularization through SIGReg. 1 arXiv:2608.22642v1 [cs.LG] 23 Aug 2026
Figure 2. Mol-JEPA evaluation results. 3.7 MOL-JEPA DOWNSTREAM PREDICTION There exist multiple strategies for using Mol-JEPA embeddings for downstream tasks. First, we use the CLS token to extract a global summary of all modalities. We feed this representation of size 512 into a linear and non-linear probe and experiment with TabICLv2 as a predictor. We use default parameters and do not perform any hyperparameter optimization. Secondly, we use the modality-specific predicted embeddings in an aggregative probe. We choose a 2 layer transformer encoder that receives tokens of size 512 for each…
Figure 3. Mol-JEPA training results. (a) Performance increases with more modalities. The lines represent the training time and epochs on a 100k sub￾set dataset, annotated with the average downstream MAE. (b) Leave-one-modality-out analysis. We analyze the ef￾fect of removing modalities during training. The reported scores are the average downstream probe MAEs
Figure 4. Modality analysis nation is that it is the only modality with a trainable GNN backbone, whereas the remaining modalities rely on frozen representations. We also find that the molecular fingerprint modalities MOE and ECFP4 rank among the most im￾portant modalities, indicating that their expert-designed features contain valuable information for downstream tasks. Lastly, we find that modalities capturing chemical knowledge through binding affinities and cellular interaction pro￾files also contribute to the learned representations, highlighting the value of incorporating biological…
Figure 5. Multimodal training dynamics. Effective rank, prediction loss and SIGReg loss for different modalities during training. While effective rank is continuously increasing, the losses exhibit instabilities for some modalities. works Kamai et al. (2026), showing that different modalities might require different approaches and not all might be suitable for joint embedding predictive architectures
Figure 6. Losses and downstream error. 5 CONCLUSION In this work, we presented Mol-JEPA, a multimodal foundation model that enables stable learning from many molec￾ular modalities by predicting their latent representations. Using various high quality benchmarks, we show that Mol-JEPA achieves strong performance across different downstream tasks. Furthermore, we statistically compare the predictions and find that our model is particularly better on smaller datasets and in out-of-distribution settings. Additionally, we analyze the impact of individual modalities on training dynamics and…
Figure 7. Cluster splits t-SNE visualization for ExpansionRx LogD dataset using ECFP4 fingerprints.
Figure 8. Loss parameters performance
Figure 9. Mean absolute error of different probe variants across datasets.
Figure 10. Mean absolute error of different feature variants across datasets.
Figure 11. Win rates on public competition splits light two representative examples on the cluster splits, demonstrating Mol-JEPA’s ability to significantly outperform the evaluated baseline methods in predictive accuracy. We further evaluate model win rates on the public splits of ExpansionRX, ASAP, and PXR. These splits are generated using a temporal strategy, which, according to the competition organizers, better represents realistic drug discovery settings by preserving the chronological order of data generation. As shown in
Figure 13. Pretraining data analysis Molecular space coverage To understand the role of the pretraining data for the downstream tasks, we analyze the molecular similarity using generative topographic mappings (GTM) Bishop et al. (1998). We randomly sample 5,000 molecules from the pretraining corpus and 2,000 molecules from the benchmark datasets, convert them into ECFP4 fingerprints, and project them into a shared GTM space. The resulting visualization, shown in Figure 13a, reveals that while some benchmark datasets are well represented within the pretraining distribution, others occupy more…
Figure 14. Training configurations E PRETRAINING DATA SCALING We analyze the effect of pretraining dataset size on downstream performance. As shown in Figure 14a, performance steadily improves with increasing pretraining data for both linear and nonlinear probes across all downstream datasets (reflected by the confidence bands). To better highlight this trend, performance is reported as 1-MAE, where higher values indicate better performance. The observed scaling behavior suggests that Mol-JEPA benefits from additional pretraining data and further increases in dataset size are expected to…
Figure 15. Multimodal performance gains. We re-train the model on the full dataset using only Graph and ECFP4 modalities. We find that the downstream performance is significantly better when using all modalities
Figure 16. Centered Kernel Alignment specific representations according to their underlying biological and physicochemical relationships while preserving information unique to individual modalities. H IN-CONTEXT LEARNING COMPARISON We assess the effectiveness of Mol-JEPA representations for in-context learning by comparing them with widely used molecular representations. As illustrated in Figures 17-19, Mol-JEPA CLS-token embeddings achieve superior performance across most tasks for different TabICLv2 hyperparameters. This finding indicates that the multimodal architecture effectively…
Figure 17. OpenADMET PXR TabICLv2 comparison
Figure 18. OpenADMET Caco2 Efflux TabICLv2 comparison.
Figure 19. OpenADMET LogD TabICLv2 comparison. 26

Showing 18 of the paper's 19 figures. See all figures in the PDF ↗ Figures reproduced from arXiv:2608.22642 by the authors.

References
How the literature uses it

What this paper does with the arXiv papers it cites

Each citing sentence is taken from the citing paper's LaTeX source and typed by Pith's classifier. A label appears at probability 0.8 or more (0.95 for support), a contest at 0.7 or more; the quoted words are the citing authors'. How each label was checked · JSON

What it cites · 8 sentences without a confident label

Uses the method or tool of 3 papers
Uses data or values from 2 papers
Cites as support 2 papers
  • Joint embedding approaches have been shown to be more robust to variations in the input and target spaces [ref] , making them particularly interesting for molecular data.

    2505.12477 · Introduction · 0.98

  • The experienced complexity of multimodal pretraining is in line with recent works [ref] , showing that different modalities might require different approaches and not all might be suitable for joint embedding predictive architectures.

    When to Align, When to Predict: A Phase Diagram for Multimodal Learning · Results · 0.97

Cites as background 3 papers

Discussion

Disagree with the review, or know something it missed? Leave a comment, highlight any passage above to comment on it, or challenge this review formally.