English

Combining Deep Learning and Explainable AI for Toxicity Prediction of Chemical Compounds

Machine Learning 2025-10-28 v1 Artificial Intelligence

Abstract

The task here is to predict the toxicological activity of chemical compounds based on the Tox21 dataset, a benchmark in computational toxicology. After a domain-specific overview of chemical toxicity, we discuss current computational strategies, focusing on machine learning and deep learning. Several architectures are compared in terms of performance, robustness, and interpretability. This research introduces a novel image-based pipeline based on DenseNet121, which processes 2D graphical representations of chemical structures. Additionally, we employ Grad-CAM visualizations, an explainable AI technique, to interpret the model's predictions and highlight molecular regions contributing to toxicity classification. The proposed architecture achieves competitive results compared to traditional models, demonstrating the potential of deep convolutional networks in cheminformatics. Our findings emphasize the value of combining image-based representations with explainable AI methods to improve both predictive accuracy and model transparency in toxicology.

Keywords

Cite

@article{arxiv.2510.22572,
  title  = {Combining Deep Learning and Explainable AI for Toxicity Prediction of Chemical Compounds},
  author = {Eduard Popescu and Adrian Groza and Andreea Cernat},
  journal= {arXiv preprint arXiv:2510.22572},
  year   = {2025}
}