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

Evaluating Self-Supervised Learning for Molecular Graph Embeddings

About

Graph Self-Supervised Learning (GSSL) provides a robust pathway for acquiring embeddings without expert labelling, a capability that carries profound implications for molecular graphs due to the staggering number of potential molecules and the high cost of obtaining labels. However, GSSL methods are designed not for optimisation within a specific domain but rather for transferability across a variety of downstream tasks. This broad applicability complicates their evaluation. Addressing this challenge, we present "Molecular Graph Representation Evaluation" (MOLGRAPHEVAL), generating detailed profiles of molecular graph embeddings with interpretable and diversified attributes. MOLGRAPHEVAL offers a suite of probing tasks grouped into three categories: (i) generic graph, (ii) molecular substructure, and (iii) embedding space properties. By leveraging MOLGRAPHEVAL to benchmark existing GSSL methods against both current downstream datasets and our suite of tasks, we uncover significant inconsistencies between inferences drawn solely from existing datasets and those derived from more nuanced probing. These findings suggest that current evaluation methodologies fail to capture the entirety of the landscape.

Hanchen Wang, Jean Kaddour, Shengchao Liu, Jian Tang, Joan Lasenby, Qi Liu• 2022

Related benchmarks

TaskDatasetResultRank
Molecular property predictionMoleculeNet BBBP (scaffold)
ROC AUC66.4
158
Molecular property predictionMoleculeNet SIDER (scaffold)
ROC-AUC0.591
136
Molecular property predictionMoleculeNet BACE (scaffold)
ROC-AUC67.4
126
Molecular property predictionMoleculeNet MUV (scaffold)
ROC-AUC0.474
107
Molecular property predictionHIV (scaffold split)
ROC-AUC70.2
56
Molecular property predictionTox21 scaffold split
ROC-AUC68.2
56
Molecular property predictionClinTox scaffold split
ROC-AUC0.663
16
Showing 7 of 7 rows

Other info

Code

Follow for update