MolPILE -- large-scale, diverse dataset for molecular representation learning
By: Jakub Adamczyk , Jakub Poziemski , Franciszek Job and more
Potential Business Impact:
Builds better computer models for new medicines.
The size, diversity, and quality of pretraining datasets critically determine the generalization ability of foundation models. Despite their growing importance in chemoinformatics, the effectiveness of molecular representation learning has been hindered by limitations in existing small molecule datasets. To address this gap, we present MolPILE, large-scale, diverse, and rigorously curated collection of 222 million compounds, constructed from 6 large-scale databases using an automated curation pipeline. We present a comprehensive analysis of current pretraining datasets, highlighting considerable shortcomings for training ML models, and demonstrate how retraining existing models on MolPILE yields improvements in generalization performance. This work provides a standardized resource for model training, addressing the pressing need for an ImageNet-like dataset in molecular chemistry.
Similar Papers
ChemPile: A 250GB Diverse and Curated Dataset for Chemical Foundation Models
Machine Learning (CS)
Gives AI a huge chemistry library to learn from.
MolTextNet: A Two-Million Molecule-Text Dataset for Multimodal Molecular Learning
Biomolecules
Helps find new medicines by understanding molecule descriptions.
MolMole: Molecule Mining from Scientific Literature
Machine Learning (CS)
Lets computers understand chemistry from papers.