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

Therapeutics Data Commons: Machine Learning Datasets and Tasks for Drug Discovery and Development

About

Therapeutics machine learning is an emerging field with incredible opportunities for innovatiaon and impact. However, advancement in this field requires formulation of meaningful learning tasks and careful curation of datasets. Here, we introduce Therapeutics Data Commons (TDC), the first unifying platform to systematically access and evaluate machine learning across the entire range of therapeutics. To date, TDC includes 66 AI-ready datasets spread across 22 learning tasks and spanning the discovery and development of safe and effective medicines. TDC also provides an ecosystem of tools and community resources, including 33 data functions and types of meaningful data splits, 23 strategies for systematic model evaluation, 17 molecule generation oracles, and 29 public leaderboards. All resources are integrated and accessible via an open Python library. We carry out extensive experiments on selected datasets, demonstrating that even the strongest algorithms fall short of solving key therapeutics challenges, including real dataset distributional shifts, multi-scale modeling of heterogeneous data, and robust generalization to novel data points. We envision that TDC can facilitate algorithmic and scientific advances and considerably accelerate machine-learning model development, validation and transition into biomedical and clinical implementation. TDC is an open-science initiative available at https://tdcommons.ai.

Kexin Huang, Tianfan Fu, Wenhao Gao, Yue Zhao, Yusuf Roohani, Jure Leskovec, Connor W. Coley, Cao Xiao, Jimeng Sun, Marinka Zitnik• 2021

Related benchmarks

TaskDatasetResultRank
ADMET Properties PredictionTDC Caco2 Wang
MAE0.256
12
ADMET Properties PredictionTDC PPBR AZ
MAE7.44
12
ADMET Properties PredictionTDC Half Life Obach
Spearman Correlation0.576
12
ADMET Properties PredictionTDC DILI
AUROC95.6
12
ADMET Properties PredictionTDC Solubility AqSolDB
MAE0.741
7
ADMET Properties PredictionTDC VDss Lombardo
Spearman Correlation0.713
7
ADMET Properties PredictionTDC Clearance Microsome AZ
Spearman Correlation0.63
7
ADMET Properties PredictionTDC Clearance Hepatocyte AZ
Spearman Correlation0.536
7
ADMET Properties PredictionTDC Pgp Broccatelli
AUROC93.8
7
ADMET Properties PredictionTDC HIA Hou
AUROC99.3
7
Showing 10 of 22 rows

Other info

Code

Follow for update