Drug Toxicity Prediction by Machine Learning Approaches
Document Type
Article
Publication Date
8-1-2023
Abstract
Drug property prediction, especially toxicity, helps reduce risks in a range of real-world applications. In this paper, we aim to apply various machine-learning models for solving the drug toxicity prediction problem. Among various machine-learning approaches, we select five suitable representatives: random forest, multi-layer perceptron, logistic regression, graph convolutional neural network, and graph isomorphism network (GIN) for conducting experiments on six datasets for toxicity prediction, including Tox 21, ClinTox, ToxCast, SIDER, HIV, and BACE. We design the GIN with four hidden layers and select the Adam optimizer with the learning rate 10-4 and the batch size 256. Furthermore, we use a batch norm layer inside each of the GIN hidden layers. Experimental results show that the designed GIN model is most efficient in distinguishing between safe and toxic drugs and outperforms the others under the supervision of ROC AUC score and recall.
Identifier
85170286726 (Scopus)
Publication Title
International Journal of Pattern Recognition and Artificial Intelligence
External Full Text Location
https://doi.org/10.1142/S0218001423510138
e-ISSN
17936381
ISSN
02180014
Issue
10
Volume
37
Recommended Citation
Shen, Yucong; Shih, Frank Y.; and Chen, Hao, "Drug Toxicity Prediction by Machine Learning Approaches" (2023). Faculty Publications. 1543.
https://digitalcommons.njit.edu/fac_pubs/1543