ToxMPNN: A deep learning model for small molecule toxicity prediction

作者：Yini Zhou, Chao Ning, Yijun Tan, Yaqi Li, Jia-Xu Wang, Yuan-Yuan Shu, Song-Ping Liang, Zhonghua Liu, Ying Wang · 发表于：Journal of Applied Toxicology · 年份：2024 · DOI：10.1002/jat.4591 · 被引用次数：24 · 研究领域：Medicine

Machine learning (ML) has shown a great promise in predicting toxicity of small molecules. However, the availability of data for such predictions is often limited. Because of the unsatisfactory performance of models trained on a single toxicity endpoint, we collected toxic small molecules with multiple toxicity endpoints from previous study. The dataset comprises 27 toxic endpoints categorized into seven toxicity classes, namely, carcinogenicity and mutagenicity, acute oral toxicity, respiratory toxicity, irritation and corrosion, cardiotoxicity, CYP450, and endocrine disruption. In addition, a binary classification Common‐Toxicity task was added based on the aforementioned dataset. To improve the performance of the models, we added marketed drugs as negative samples. This study presents a toxicity predictive model, ToxMPNN, based on the message passing neural network (MPNN) architecture, aiming to predict the toxicity of small molecules. The results demonstrate that ToxMPNN outperforms other models in capturing toxic features within the molecular structure, resulting in more precise predictions with the ROC_AUC testing score of 0.886 for the Toxicity_drug dataset. Furthermore, it was observed that adding marketed drugs as negative samples not only improves the predictive performance of the binary classification Common‐Toxicity task but also enhances the stability of the model prediction. It shows that the graph‐based deep learning (DL) algorithms in this study can be used as...