Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Learning the Neighborhood: Contrast-Free Multimodal Self-Supervised Molecular Graph Pretraining

About

High-quality molecular representations are essential for property prediction and molecular design, yet large labeled datasets remain scarce. While self-supervised pretraining on molecular graphs has shown promise, many existing approaches either depend on hand-crafted augmentations or complex generative objectives, and often rely solely on 2D topology, leaving valuable 3D structural information underutilized. To address this gap, we introduce C-FREE (Contrast-Free Representation learning on Ego-nets), a simple framework that integrates 2D graphs with ensembles of 3D conformers. C-FREE learns molecular representations by predicting subgraph embeddings from their complementary neighborhoods in the latent space, using fixed-radius ego-nets as modeling units across different conformers. This design allows us to integrate both geometric and topological information within a hybrid Graph Neural Network (GNN)-Transformer backbone, without negatives, positional encodings, or expensive pre-processing. Pretraining on the GEOM dataset, which provides rich 3D conformational diversity, C-FREE achieves state-of-the-art results on MoleculeNet, surpassing contrastive, generative, and other multimodal self-supervised methods. Fine-tuning across datasets with diverse sizes and molecule types further demonstrates that pretraining transfers effectively to new chemical domains, highlighting the importance of 3D-informed molecular representations.

Boshra Ariguib, Mathias Niepert, Andrei Manolache• 2025

Related benchmarks

TaskDatasetResultRank
Molecular Property Prediction (Classification)MoleculeNet (test)
BBBP88.9
40
ClassificationMoleculeNet
BBBP Accuracy73.8
34
Molecular Property ClassificationMoleculeNet Linear Probe
BBBP Accuracy73.8
11
Molecular Property RegressionQM9 full end-to-end fine-tuning
MU0.077
8
Molecular Property RegressionZINC
MAE0.204
2
Showing 5 of 5 rows

Other info

Follow for update