D3: A Small Language Model for Drug-Drug Interaction prediction and comparison with Large Language Models

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

Institutions

Publication Details

Journal
Machine Learning with Applications
Published
2025-04-29
DOI
https://doi.org/10.1016/j.mlwa.2025.100658
Citations
5
Primary Topic
Computational Drug Discovery Methods
Type
article
Field-Weighted Citation Impact
7.52
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

D3: A Small Language Model for Drug-Drug Interaction prediction and comparison with Large Language Models

Salma Ibrahim, Ahmed I. Abd‐Elhamid, Ahmed Serag, A.H.M. Sarowar Sattar et al.
5 citations
Machine Learning with Applications
Computational Drug Discovery Methods
7.52
article

D3: A Small Language Model for Drug-Drug Interaction prediction and comparison with Large Language Models

Salma Ibrahim, Ahmed I. Abd‐Elhamid, Ahmed Serag, A.H.M. Sarowar Sattar, Abdullah Hosseini
article en
5 citations

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.

Machine Learning with ApplicationsVol. 20
Weill Cornell Medical College in Qatar (QA), Qatar University (QA)
Quality Education
Openalex Percentile: Top 3%
Computational Drug Discovery Methods
7.52
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.