MoleculeNet: a benchmark for molecular machine learning.

Journal: Chemical science
Published Date: Oct 31, 2017

Abstract

Molecular machine learning has been maturing rapidly over the last few years. Improved methods and the presence of larger datasets have enabled machine learning algorithms to make increasingly accurate predictions about molecular properties. However, algorithmic progress has been limited due to the lack of a standard benchmark to compare the efficacy of proposed methods; most new algorithms are benchmarked on different datasets making it challenging to gauge the quality of proposed methods. This work introduces MoleculeNet, a large scale benchmark for molecular machine learning. MoleculeNet curates multiple public datasets, establishes metrics for evaluation, and offers high quality open-source implementations of multiple previously proposed molecular featurization and learning algorithms (released as part of the DeepChem open source library). MoleculeNet benchmarks demonstrate that learnable representations are powerful tools for molecular machine learning and broadly offer the best performance. However, this result comes with caveats. Learnable representations still struggle to deal with complex tasks under data scarcity and highly imbalanced classification. For quantum mechanical and biophysical datasets, the use of physics-aware featurizations can be more important than choice of particular learning algorithm.

Authors

  • Zhenqin Wu
    Department of Chemistry , Stanford University , Stanford , CA 94305 , USA . Email: pande@stanford.edu.
  • Bharath Ramsundar
    Department of Computer Science , Stanford University , Stanford , CA 94305 , USA.
  • Evan N Feinberg
    Program in Biophysics , Stanford School of Medicine , Stanford , CA 94305 , USA.
  • Joseph Gomes
    Department of Chemistry , Stanford University , Stanford , CA 94305 , USA . Email: pande@stanford.edu.
  • Caleb Geniesse
    Program in Biophysics , Stanford School of Medicine , Stanford , CA 94305 , USA.
  • Aneesh S Pappu
    Department of Computer Science , Stanford University , Stanford , CA 94305 , USA.
  • Karl Leswing
    Schrodinger Inc. , USA.
  • Vijay Pande
    Department of Chemistry , Stanford University , Stanford , CA 94305 , USA . Email: pande@stanford.edu.

Keywords

No keywords available for this article.

External Resources

Popular Topics
Recent Journals