CoTox: Chain-of-Thought-Based Molecular Toxicity Reasoning and Prediction

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Park, Jueon, Park, Yein, Song, Minju, Park, Soyon, Lee, Donghyeon, Baek, Seungheun, Kang, Jaewoo
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912688089595904
author Park, Jueon
Park, Yein
Song, Minju
Park, Soyon
Lee, Donghyeon
Baek, Seungheun
Kang, Jaewoo
author_facet Park, Jueon
Park, Yein
Song, Minju
Park, Soyon
Lee, Donghyeon
Baek, Seungheun
Kang, Jaewoo
contents Drug toxicity remains a major challenge in pharmaceutical development. Recent machine learning models have improved in silico toxicity prediction, but their reliance on annotated data and lack of interpretability limit their applicability. This limits their ability to capture organ-specific toxicities driven by complex biological mechanisms. Large language models (LLMs) offer a promising alternative through step-by-step reasoning and integration of textual data, yet prior approaches lack biological context and transparent rationale. To address this issue, we propose CoTox, a novel framework that integrates LLM with chain-of-thought (CoT) reasoning for multi-toxicity prediction. CoTox combines chemical structure data, biological pathways, and gene ontology (GO) terms to generate interpretable toxicity predictions through step-by-step reasoning. Using GPT-4o, we show that CoTox outperforms both traditional machine learning and deep learning model. We further examine its performance across various LLMs to identify where CoTox is most effective. Additionally, we find that representing chemical structures with IUPAC names, which are easier for LLMs to understand than SMILES, enhances the model's reasoning ability and improves predictive performance. To demonstrate its practical utility in drug development, we simulate the treatment of relevant cell types with drug and incorporated the resulting biological context into the CoTox framework. This approach allow CoTox to generate toxicity predictions aligned with physiological responses, as shown in case study. This result highlights the potential of LLM-based frameworks to improve interpretability and support early-stage drug safety assessment. The code and prompt used in this work are available at https://github.com/dmis-lab/CoTox.
format Preprint
id arxiv_https___arxiv_org_abs_2508_03159
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle CoTox: Chain-of-Thought-Based Molecular Toxicity Reasoning and Prediction
Park, Jueon
Park, Yein
Song, Minju
Park, Soyon
Lee, Donghyeon
Baek, Seungheun
Kang, Jaewoo
Machine Learning
Artificial Intelligence
Drug toxicity remains a major challenge in pharmaceutical development. Recent machine learning models have improved in silico toxicity prediction, but their reliance on annotated data and lack of interpretability limit their applicability. This limits their ability to capture organ-specific toxicities driven by complex biological mechanisms. Large language models (LLMs) offer a promising alternative through step-by-step reasoning and integration of textual data, yet prior approaches lack biological context and transparent rationale. To address this issue, we propose CoTox, a novel framework that integrates LLM with chain-of-thought (CoT) reasoning for multi-toxicity prediction. CoTox combines chemical structure data, biological pathways, and gene ontology (GO) terms to generate interpretable toxicity predictions through step-by-step reasoning. Using GPT-4o, we show that CoTox outperforms both traditional machine learning and deep learning model. We further examine its performance across various LLMs to identify where CoTox is most effective. Additionally, we find that representing chemical structures with IUPAC names, which are easier for LLMs to understand than SMILES, enhances the model's reasoning ability and improves predictive performance. To demonstrate its practical utility in drug development, we simulate the treatment of relevant cell types with drug and incorporated the resulting biological context into the CoTox framework. This approach allow CoTox to generate toxicity predictions aligned with physiological responses, as shown in case study. This result highlights the potential of LLM-based frameworks to improve interpretability and support early-stage drug safety assessment. The code and prompt used in this work are available at https://github.com/dmis-lab/CoTox.
title CoTox: Chain-of-Thought-Based Molecular Toxicity Reasoning and Prediction
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2508.03159