Rigorous Evaluation of Large Language Models for Malaria Drug Discovery: Trade-offs in Performance, Scale, and Resource Utility
Organizations: Magami Open Sciences Initiative
Abstract
We introduce Malaria-Instruct, a curated instruction-following dataset derived from the ChEMBL Legacy Malaria corpus for Malaria virtual screening, and conduct a systematic evaluation of five open-source LLMs; Gemma-2 2B/9B, TxGemma-2B/9B, and LlaSMol-Mistral-7B, on a rigorous out-of-distribution data split. Performance was benchmarked against classical ML models (Random Forest, XGBoost) and frontier proprietary models (Gemini 2.5, OpenAI o3) under few-shot conditions. Fine-tuned LLMs substantially outperformed all baselines: TxGemma-9B achieved the highest ROC-AUC () and LlaSMol-Mistral-7B the best enrichment factor (EF@1% 4.99). Domain-specific fine-tuning proved categorically indispensable with TxGemma-9B collapsing from ROC-AUC 0.731 to 0.499, under its best few-shot condition, and neither Gemini 2.5 (ROC-AUC 0.53) nor o3 (ROC-AUC 0.59) achieved reliable discrimination without fine-tuning. Biomedical pretraining conferred a measurable advantage at equivalent scale, while chemistry-aware pretraining yielded superior prospective enrichment. Fine-tuned open-source LLMs represent a compelling, resource-efficient paradigm for antimalarial VS, outperforming both classical pipelines and proprietary reasoning models under structurally challenging conditions.
Figures & tables
| Category | Task Type | Model | Size | ROC_AUC | MCC | EF (1%) | Inf. Time (s) |
| Classical | Finetuned | RandomForest | – | 0.6828 | 0.2581 | 2.5790 | 0.0004 |
| Finetuned | XGBoost | – | 0.6652 | 0.2164 | 1.7908 | 0.00004 | |
| Open Source | Finetuned | Gemma 2 | 2B | 0.7063 | 0.47735 | 3.66045 | 0.0040 |
| LLMs | Finetuned | TxGemma 2 | 2B | 0.6955 | 0.53275 | 4.6646 | 0.0047 |
| Finetuned | LlaSmol - Mistral | 7B | 0.70225 | 0.56415 | 4.9865 | 0.08015 | |
| Finetuned | Gemma 2 | 9B | 0.6383 | 0.4408 | 4.6347 | 0.0304 |
| Model | Shots | Acc | AUC | MCC | EF |
|---|---|---|---|---|---|
| LlaSMol 7B | 3 | 0.7952 | 0.5140 | 0.0545 | 1.5424 |
| 4 | 0.8011 | 0.5196 | 0.0721 | 1.6980 | |
| 5 | 0.8020 | 0.5128 | 0.0455 | 1.4416 | |
| Gemma 2B | 3 | 0.0061 | 0.5025 | -0.0026 | 0.2971 |
| 4 | 0.0059 | 0.5029 | 0.0053 | 0.7198 | |
| 5 | 0.0009 | 0.4993 | 0.0038 | 1.2486 |
| Model | Shots | ROC_AUC | MCC | EF (1%) |
|---|---|---|---|---|
| Gemini 2.5 | 3 | 0.5182 | 0.0945 | 2.342 |
| 4 | 0.5284 | 0.1689 | 3.788 | |
| 5 | 0.5175 | 0.0883 | 2.254 | |
| OpenAI o3 | 3 | 0.5751 | 0.1285 | 1.411 |
| 4 | 0.5907 | 0.1573 | 1.551 | |
| 5 | 0.5867 | 0.1583 | 1.646 |
| Finetuning | Fewshot (Best) | ||||||
|---|---|---|---|---|---|---|---|
| Size | Model | Train GPU | Inf. Time | EF | Inf. GPU | Inf. Time | EF |
| 2B | Gemma 2 | 16GB | 0.0040 | 3.660 | 24GB | 0.0457 | 0.720 |
| 2B | TxGemma | 16GB | 0.0047 | 4.665 | 24GB | 0.0142 | 0.804 |
| 7B | LlaSmol | 24GB | 0.0802 | 4.987 | 24GB | 0.2337 | 1.698 |
| 9B | Gemma 2 | 24GB | 0.0304 | 4.635 | 24GB | 0.4895 | 1.110 |
| 9B | TxGemma | 24GB | 0.0371 | 4.230 | 24GB | 0.0873 | 1.816 |
Drug Discovery
Drug discovery identifies, designs, and tests molecules that can safely affect disease biology.
See all 165 papers