MoleculeNet: a benchmark for molecular machine learning
Zhenqin WuBharath RamsundarEvan N. FeinbergJoseph GomesCaleb GeniesseAneesh S. PappuKarl LeswingVijay Pande
Presents MoleculeNet, a standardized open-source benchmark integrated into DeepChem that provides common datasets, evaluation metrics, and baseline implementations to systematically compare machine learning algorithms across diverse molecular, biophysical, and quantum mechanical tasks.
The article addresses the challenge that molecular machine learning lacks standardized benchmarks, making it difficult to compare new algorithms fairly. Molecular datasets are typically small, heterogeneous, and expensive to obtain, while input molecules vary widely in structure and the properties to predict range from quantum mechanics to physiological effects on the human body. Without common datasets, metrics, and evaluation protocols, progress has been slow despite growing interest in data-driven methods for chemistry.
The work introduces MoleculeNet, a benchmark that curates 17 public datasets covering over 700,000 compounds across four categories—quantum mechanics, physical chemistry, biophysics, and physiology—along with defined metrics, data splits, and open-source implementations of featurization and learning methods integrated into the DeepChem library.
The authors evaluated conventional models such as random forests, kernel SVMs, and multitask networks alongside graph-based models that use learnable featurizations. They tested multiple splitting strategies including random, scaffold, stratified, and time-based splits, and reported results using appropriate regression and classification metrics such as MAE, RMSE, ROC-AUC, and PRC-AUC.
Graph-based models with learnable representations delivered the strongest results on 11 of the 17 datasets, often outperforming conventional approaches by clear margins when sufficient data were available. However, these models showed reduced robustness on smaller datasets and under high class imbalance. Physics-informed featurizations proved more critical than the choice of learning algorithm for quantum and biophysical tasks. Data-driven methods also matched or exceeded traditional ab-initio calculations on physical chemistry properties once moderate amounts of training data were provided.
These findings indicate that learnable featurizations offer a scalable path forward for molecular property prediction, yet additional data, improved handling of imbalance, and task-specific features remain necessary for biophysical and physiological applications. The benchmark establishes a common platform that can accelerate method development in both chemistry and machine learning, similar to the role ImageNet played in computer vision.
The authors recommend extending MoleculeNet to cover protein structure prediction and DNA modeling, encouraging community contributions of new datasets and algorithms to DeepChem, and collecting more high-quality data for physiology tasks. Limitations include the computational cost of some graph models, potential overfitting on small datasets, variability across data splits, and the fact that reported numbers reflect fixed training-time constraints rather than fully optimized runs from the original papers.
- Paper: Convolutional Networks on Graphs for Learning Molecular Fingerprints, David Duvenaud et al. (2015). Introduces differentiable neural molecular fingerprints directly from graph structures, establishing the learnable molecular representation paradigm benchmarked in MoleculeNet.
- Paper: Neural Message Passing for Quantum Chemistry, Justin Gilmer et al. (2017). Formulates the Message Passing Neural Network (MPNN) framework for quantum chemistry on molecular graphs, providing one of the core graph-based model architectures evaluated across MoleculeNet.
- Paper: The Graph Neural Network Model, Franco Scarselli et al. (2009). Establishes the foundational graph neural network architecture that operates directly on irregular molecular graph structures without fixed-size vector flattening.
- Paper: Weisfeiler-Lehman Graph Kernels, N. Shervashidze et al. (2011). Presents the Weisfeiler-Lehman subtree graph kernel framework, which forms the theoretical and methodological basis for discrete structural comparisons and baseline graph featurizations.
- Paper: Automatic Chemical Design Using a Data-Driven Continuous Representation of Molecules, R. Gómez-Bombarelli et al. (2016). Demonstrates data-driven continuous latent representations of molecules using variational autoencoders on standard datasets like QM9 and ZINC, framing the representation learning goals MoleculeNet evaluates.
- Paper: Learning Convolutional Neural Networks for Graphs, Mathias Niepert et al. (2016). Develops PATCHY-SAN for convolutional learning over arbitrary graphs via neighborhood normalization, providing key context for non-spectral spatial graph deep learning baselines.
- Paper: An Overview of Multi-Task Learning in Deep Neural Networks, Sebastian Ruder (2017). Synthesizes hard and soft multi-task deep learning mechanisms, explaining the multi-task molecular property prediction formulations evaluated throughout MoleculeNet.
- Paper: XGBoost: A Scalable Tree Boosting System, Tianqi Chen et al. (2016). Details the scalable tree boosting system XGBoost, which serves as a standard non-deep learning baseline for tabular and fingerprint-based molecular property prediction.
- Paper: Open Graph Benchmark: Datasets for Machine Learning on Graphs, Weihua Hu et al. (2020). Generalizes standardized molecular and graph-level benchmarking to a broader suite of large-scale graph machine learning challenges and realistic evaluation protocols.
- Paper: Strategies for Pre-training Graph Neural Networks, Weihua Hu et al. (2020). Addresses the data scarcity and out-of-distribution challenges identified in MoleculeNet by introducing node- and graph-level pre-training strategies for molecular GNNs.
- Paper: How Powerful are Graph Neural Networks?, Keyulu Xu et al. (2019). Provides the foundational theoretical analysis and the Graph Isomorphism Network (GIN) architecture that explains the representational boundaries of the GNNs benchmarked in MoleculeNet.
- Paper: Relational inductive biases, deep learning, and graph networks, Peter W. Battaglia et al. (2018). Synthesizes graph networks and relational inductive biases into a unifying framework covering message passing and molecular learning architectures.
- Paper: E(n) Equivariant Graph Neural Networks, Victor Garcia Satorras et al. (2021). Extends molecular graph representations by enforcing E(n) rotational and translational equivariance directly in physical 3D coordinate space for quantum property prediction.
- Paper: Do Transformers Really Perform Bad for Graph Representation?, Chengxuan Ying et al. (2021). Develops the Graphormer transformer architecture with structural encodings, significantly advancing beyond standard GNN baselines on large-scale molecular property prediction tasks.
- Paper: Weisfeiler and Leman Go Neural: Higher-Order Graph Neural Networks, Christopher Morris et al. (2019). Extends standard molecular message-passing neural networks to higher-order k-GNNs based on multi-dimensional Weisfeiler-Leman heuristics to capture complex molecular subgraphs.
- Paper: Graph Contrastive Learning with Augmentations, Yuning You et al. (2020). Presents a self-supervised graph contrastive learning framework with structural augmentations to improve generalization on molecular and biochemical property benchmarks.
- Paper: GNNExplainer: Generating Explanations for Graph Neural Networks, Rex Ying et al. (2019). Introduces a model-agnostic explanation method to interpret and identify critical substructures in graph neural network predictions on molecular datasets.
- Paper: Graph Neural Networks: A Review of Methods and Applications, Jie Zhou et al. (2018). Provides a comprehensive survey of graph neural network variants, training paradigms, and downstream scientific applications developed following initial molecular benchmarking efforts.
