Abstract:Predictive modeling of toxicity is a crucial step in the drug discovery pipeline. It can help filter out molecules with a high probability of failing in the early stages of de novo drug design. Thus, several machine learning (ML) models have been developed to predict the toxicity of molecules by combining classical ML techniques or deep neural networks with well-known molecular representations such as fingerprints or 2D graphs. But the more natural, accurate representation of molecules is expected to be defined in physical 3D space like in ab initio methods. Recent studies successfully used equivariant graph neural networks (EGNNs) for representation learning based on 3D structures to predict quantum-mechanical properties of molecules. Inspired by this, we investigated the performance of EGNNs to construct reliable ML models for toxicity prediction. We used the equivariant transformer (ET) model in TorchMD-NET for this. Eleven toxicity data sets taken from MoleculeNet, TDCommons, and ToxBenchmark have been considered to evaluate the capability of ET for toxicity prediction. Our results show that ET adequately learns 3D representations of molecules that can successfully correlate with toxicity activity, achieving good accuracies on most data sets comparable to state-of-the-art models. We also test a physicochemical property, namely, the total energy of a molecule, to inform the toxicity prediction with a physical prior. However, our work suggests that these two properties can not be related. We also provide an attention weight analysis for helping to understand the toxicity prediction in 3D space and thus increase the explainability of the ML model. In summary, our findings offer promising insights considering 3D geometry information via EGNNs and provide a straightforward way to integrate molecular conformers into ML-based pipelines for predicting and investigating toxicity prediction in physical space. We expect that in the future, especially for larger, more diverse data sets, EGNNs will be an essential tool in this domain.

Graph Kernel Learning for Predictive Toxicity Models

Equivariant Graph Neural Networks for Toxicity Prediction

Drug Toxicity Prediction by Machine Learning Approaches

Mining Toxicity Information from Large Amounts of Toxicity Data

A deep learning based multi-model approach for predicting drug-like chemical compound's toxicity

An Adaptive Graph Learning Method for Automated Molecular Interactions and Properties Predictions

Multimodal Representation Learning via Graph Isomorphism Network for Toxicity Multitask Learning

Hi-MGT: A hybrid molecule graph transformer for toxicity identification

GCLmf: A Novel Molecular Graph Contrastive Learning Framework Based on Hard Negatives and Application in Toxicity Prediction

Predictive Systems Toxicology

Chemical toxicity prediction based on semi-supervised learning and graph convolutional neural network

Novel Approach of Deep Learning in Toxicity Prediction

deepFPlearn +: enhancing toxicity prediction across the chemical universe using graph neural networks

Review of machine learning and deep learning models for toxicity prediction

TOP: Towards Better Toxicity Prediction by Deep Molecular Representation Learning

KG4SL: knowledge graph neural network for synthetic lethality prediction in human cancers

Efficient Toxicity Prediction via Simple Features Using Shallow Neural Networks and Decision Trees

Co‐Model for Chemical Toxicity Prediction Based on Multi‐task Deep Learning

Toxicity Prediction using Deep Learning

Deep learning for predicting toxicity of chemicals: a mini review

Integrating structure annotation and machine learning approaches to develop graphene toxicity models