Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

BioMamba: Domain-Adaptive Biomedical Language Models

About

Background. Biomedical language models should improve performance on biomedical text while retaining general-language-modeling fluency. For Mamba-based models, this trade-off has not been systematically studied across biomedical literature and clinical text. Methods. We developed BioMamba, a family of biomedical Mamba2 models at five scales obtained by continued pretraining of released public Mamba2 checkpoints on a balanced 80%/10%/10% mixture of PubMed abstracts, the Colossal Clean Crawled Corpus (C4), and Wikipedia. The contribution is the adaptation recipe and the accompanying open-weight checkpoints. Results. Across five scales, BioMamba consistently lowered PubMed perplexity, improved Wikipedia-style held-out perplexity by 1.46-4.72 PPL, and left C4 perplexity essentially unchanged. On six out-of-domain multiple-choice benchmarks, BioMamba stayed within +/-3 percentage points of Mamba2 with no systematic regression. After supervised fine-tuning, BioMamba+SFT matched or exceeded Mamba2+SFT on MIMIC-IV note completion and discharge summary generation at every evaluated scale, and improved PubMedQA at every scale. The strongest model (BioMamba-2.7B) reached a PubMed perplexity of 5.28 and accuracies of 90.24% and 73.00% on BioASQ and PubMedQA, respectively. Conclusions. A balanced domain-adaptive continued pretraining recipe strengthens Mamba2 language models on biomedical literature and clinical text while preserving general-language-modeling fluency.

Ling Yue, Mingzhi Zhu, Sixue Xing, Yunning Cao, Yanbo Wang, Shimin Shan, Jinfei Liu, Vijil Chenthamarakshan, Shaowu Pan, Payel Das, Tianfan Fu• 2024

Related benchmarks

TaskDatasetResultRank
Language ModelingC4
Perplexity14.91
1688
Question AnsweringPubMedQA (test)--
170
Language ModelingPubmed
Perplexity6.52
59
Language ModelingWikipedia
Perplexity9.71
43
Discharge summary generationMIMIC-IV (test)
ROUGE-110.11
21
Note completionMIMIC-IV (test)
ROUGE-18.11
21
Biomedical Natural Language ProcessingBiomedical NLP Benchmarks
F1 Score88
6
Question AnsweringBioASQ
Accuracy90.24
5
Question AnsweringPubMedQA
Accuracy73
5
Showing 9 of 9 rows

Other info

Follow for update