D3: A Small Language Model for Drug-Drug Interaction prediction and comparison with Large Language Models
Year: 2025
Type: article
Abstract: Large Language Models (LLMs) have significantly advanced Natural Language Processing (NLP) applications, including healthcare. However, their high computational demands pose challenges for deployment in resource-constrained settings. Small Language Models (SLMs) offer a promising alternative, balancing performance and efficiency. In this study, we introduce D3, a compact SLM with approximately 70 million parameters, designed for Drug-Drug Interaction (DDI) prediction. Trained on a curated DrugBank dataset, D3 was compared against fine-tuned state-of-the-art LLMs, Qwen 2.5, Gemma 2, Mistral v0.3, and LLaMA 3.1, ranging from 1.5 billion to 70 billion parameters. Despite being 1000 times smaller than LLaMA 3.1, D3 achieved an F1 score of 0.86, comparable to larger models (Mistral v0.3: 0.88, LLaMA 3.1: 0.89), with no statistically significant performance difference. Expert evaluations further confirmed that D3’s predictions were clinically relevant and closely aligned with those of larger models. Our findings demonstrate that SLMs can effectively compete with LLMs in DDI prediction, achieving strong performance while significantly reducing computational requirements. Beyond DDI prediction, this work highlights the broader potential of small models in healthcare, where balancing accuracy and efficiency is critical. • D3 delivers DDI prediction on par with LLMs using 1000 × fewer parameters. • Our findings highlight the potential of SLMs in biomedical applications. • Optimized SLMs can rival LLMs on specialized tasks at lower computational cost.
Authors: Ahmed Ibrahim, Abdullah Hosseini, Salma Ibrahim, Aamenah Sattar, Ahmed Serag
Language: English
Subfield: Computational Theory and Mathematics
Field: Computer Science
Domain: Physical Sciences
Open Access status: gold
Estimated APC paid: $2,340
OpenAlex ID: W4409920175