Improving ADMET prediction with descriptor augmentation of Mol2Vec embeddings
Stratiichuk, R.; Shevchuk, N.; Kyrylenko, R.; Vozniak, V.; Koleiev, I.; Voitsitskyi, T.; Husak, V.; Ostrovsky, Z.; Khropachov, I.; Starosyla, S.; Yesylevsky, S.; Nafiev, A.
Show abstract
The accurate prediction of ADMET (Absorption, Distribution, Metabolism, Excretion, and Toxicity) properties is crucial for early-stage drug development, enabling the reduction of late-stage attrition and guiding compound prioritization. In recent years, machine learning models have emerged as powerful tools for ADMET prediction, leveraging diverse molecular representations ranging from handcrafted descriptors to graph neural networks and language model embeddings. Despite these advances, balancing predictive performance with computational efficiency remains a key challenge, particularly for high-throughput screening scenarios. Among unsupervised embedding methods, Mol2Vec has shown promise by capturing chemical substructure context analogously to word embeddings in natural language processing. However, its performance on comprehensive ADMET benchmarks has not been systematically assessed. In this work, we reimplement Mol2Vec with an expanded training corpus and higher embedding dimensionality, and evaluate its utility across 16 ADMET prediction tasks from the Therapeutics Data Commons (TDC). We show that while Mol2Vec embeddings alone are competitive, combining them with classical molecular descriptors and applying feature selection significantly improves performance. Our final MLP models with enhanced Mol2Vec embeddings achieved top-1 results in 10 of 16 benchmarks, outperforming all previously reported models on the TDC leaderboard in this regard, demonstrating that descriptor-enriched representations, paired even with relatively simple MLPs, can rival or exceed the performance of more complex models.
Matching journals
The top 5 journals account for 50% of the predicted probability mass.
Similar papers in this journal
- MolAI: A Deep Learning Framework for Data-driven Molecular Descriptor Generation and Advanced Drug Discovery Applications 96%
- A Multimodal Deep Learning Framework for Predicting PPI-Modulator Interactions 95%
- Benchmarking of Small Molecule Feature Representations for hERG, Nav1.5, and Cav1.2 Cardiotoxicity Prediction 95%
Similar papers in this journal
- Predicting Inhibitors of OATP1B1 via Heterogeneous OATP-Ligand Interaction Graph Neural Network (HOLIgraph) 95%
- DrugDiff - small molecule diffusion model with flexible guidance towards molecular properties 94%
- Comprehensive machine learning boosts structure-based virtual screening for PARP1 inhibitors 93%
Similar papers in this journal
Similar papers in this journal
- Intense bitterness of molecules: machine learning for expediting drug discovery 95%
- G-PLIP: Knowledge graph neural network for structure-free protein-ligand bioactivity prediction 94%
- Machine learning driven acceleration of biopharmaceutical formulation development using Excipient Prediction Software (ExPreSo) 93%
Similar papers in this journal
"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.