arXiv cs.AI· Burak Bakkar, Hana Moalla, Mohamed Salem Jedidi, Haifa Ben Messaoud, Nicolas Lopez Carranza·· 1 天前AI 评分27
用领域专用大语言模型推进临床试验结果预测
Advancing Clinical Trial Outcome Prediction with Domain-Specific Large Language Models
AI 导读
BioBERT(108M 参数)与 BioMistral(7B 参数)在 TOP 基准(12,477 项试验)上做分期结果预测,四种语言模型变体的 ROC-AUC(0.613–0.772)和 PR-AUC(0.649–0.901)均超过已发表的 HINT 结果,PR-AUC 相对提升 10–20%。
正文 · 原文
Abstract:Clinical trial failures are a major barrier to efficient drug development, incurring substantial costs and delaying patient access to effective therapies. Structured approaches such as the Hierarchical Interaction Network (HINT) predict trial outcomes by explicitly modelling interactions between drugs, diseases and protocols, but make limited use of the free text in trial registrations. We evaluate two domain-specific language models, BioBERT (108M parameters) and BioMistral (7B parameters), for phase-level outcome prediction on the TOP benchmark (12,477 trials), representing each trial as text built from ten this http URL fields. We compare fine-tuning (full fine-tuning for BioBERT, low-rank adaptation for BioMistral) with probing, in which the pretrained model is frozen and only a lightweight classifier is trained on its embeddings. On the held-out test set of each phase, all four language-model variants achieved higher ROC-AUC (0.613--0.772) and PR-AUC (0.649--0.901) than the published HINT results, corresponding to relative PR-AUC gains of 10--20\%. F1 differences were small and mixed, and no model, including HINT, clearly exceeded the F1 of a trivial classifier that predicts success for every trial. Under the configurations tested, probing performed comparably to fine-tuning: fine-tuned models scored highest on ROC-AUC and PR-AUC in Phases I and II, and frozen BioMistral embeddings in Phase III (ROC-AUC 0.772; PR-AUC 0.901), although most differences between variants were below 0.03 and were not tested for significance. These results suggest that pretrained biomedical representations already encode signal relevant to trial outcome, and that probing is a computationally cheaper alternative to fine-tuning for this task.
| Subjects: | Artificial Intelligence (cs.AI) |
| Cite as: | arXiv:2610.41710 [cs.AI] |
| (or arXiv:2610.41710v1 [cs.AI] for this version) | |
| https://doi.org/10.48550/arXiv.2610.41710 arXiv-issued DOI via DataCite (pending registration) |
Submission history
From: Burak Bakkar [view email]
[v1]
Fri, 30 Oct 2026 13:17:00 UTC (1,947 KB)
来源:arXiv cs.AI · arxiv.org