DrugReasoner: Interpretable Drug Approval Prediction with a Reasoning-augmented Language Model
2508.18579v1
cs.LG, cs.AI, q-bio.QM, I.2.7; J.3; I.2.6
2025-08-28
Авторы:
Mohammadreza Ghaffarzadeh-Esfahani, Ali Motahharynia, Nahid Yousefian, Navid Mazrouei, Jafar Ghaisari, Yousof Gheisari
Резюме на русском
## Контекст
Область исследования, связанная с предсказанием утверждения лекарственных средств, является ключевой для оптимизации научных исследований и сокращения времени получения новых лекарств на рынок. Несмотря на высокую точность методов машинного обучения, они часто сталкиваются с проблемой непрозрачности решений, что ограничивает их применение в реальной практике. Это делает необходимым развитие моделей, которые не только предсказывали бы результаты, но и обеспечивали бы понятное обоснование своих выводов.
## Метод
DrugReasoner — это применительно к лекарственным средствам расширенная модель языка (LLM), основанная на архитектуре LLaMA. Модель была дообучена с использованием групповой политики оптимизации (GRPO) для повышения точности предсказания и улучшения разборчивости решений. Основные инструменты включили методы сравнительного анализа, используя описательные характеристики медикаментов, которые помогали сопоставлять их с утвержденными и отклоненными вариантами. Модель также выдает подробные логические разрешения, подкрепляющие свои выводы.
## Результаты
Проведенные эксперименты показали, что DrugReasoner превысил многие классические модели, такие как логистическая регрессия, суппурт-векторная машина и к-ближайшие соседи, показав AUC в 0.732 и F1-меру в 0.729 на валидационной выборке, а также стабильные результаты на тестовой выборке. Эксперименты на внешней выборке подтвердили высокую стабильность модели, с AUC в 0.728 и F1-мерой в 0.774. Эти результаты становятся ключевыми для предсказания утверждения лекарственных средств с высокой точностью и сохранением логической прозрачности решений.
## Значимость
DrugReasoner может применяться в различных фармацевтических процессах, в том числе в быстром оценивании потенциала новых лекарственных средств. Он улучшает прозрачность, что делает его привлекательным для решений в области AI-поддерживаемого лекарственного развития. Благодаря развитию решений, которые обеспечивают не только точность но и понятные выводы, модель открывает новые пути для интеллектуальной поддержки в научном и практическом лекарственном производстве.
## Выводы
Результаты DrugReasoner показали высокую точность и прозрачность в предсказании утверждения лекарственных средств. На основе этих достижений, будущие исследования будут направлены на улучшение модели, включая расширение размера данных, изучение новых методов оптимизации и расширение применимости модели к другим областям биомедицины.
Abstract
Drug discovery is a complex and resource-intensive process, making early
prediction of approval outcomes critical for optimizing research investments.
While classical machine learning and deep learning methods have shown promise
in drug approval prediction, their limited interpretability constraints their
impact. Here, we present DrugReasoner, a reasoning-based large language model
(LLM) built on the LLaMA architecture and fine-tuned with group relative policy
optimization (GRPO) to predict the likelihood of small-molecule approval.
DrugReasoner integrates molecular descriptors with comparative reasoning
against structurally similar approved and unapproved compounds, generating
predictions alongside step-by-step rationales and confidence scores.
DrugReasoner achieved robust performance with an AUC of 0.732 and an F1 score
of 0.729 on the validation set and 0.725 and 0.718 on the test set,
respectively. These results outperformed conventional baselines, including
logistic regression, support vector machine, and k-nearest neighbors and had
competitive performance relative to XGBoost. On an external independent
dataset, DrugReasoner outperformed both baseline and the recently developed
ChemAP model, achieving an AUC of 0.728 and an F1-score of 0.774, while
maintaining high precision and balanced sensitivity, demonstrating robustness
in real-world scenarios. These findings demonstrate that DrugReasoner not only
delivers competitive predictive accuracy but also enhances transparency through
its reasoning outputs, thereby addressing a key bottleneck in AI-assisted drug
discovery. This study highlights the potential of reasoning-augmented LLMs as
interpretable and effective tools for pharmaceutical decision-making.