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.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Language Modeling | C4 | Perplexity14.91 | 1688 | |
| Question Answering | PubMedQA (test) | -- | 170 | |
| Language Modeling | Pubmed | Perplexity6.52 | 59 | |
| Language Modeling | Wikipedia | Perplexity9.71 | 43 | |
| Discharge summary generation | MIMIC-IV (test) | ROUGE-110.11 | 21 | |
| Note completion | MIMIC-IV (test) | ROUGE-18.11 | 21 | |
| Biomedical Natural Language Processing | Biomedical NLP Benchmarks | F1 Score88 | 6 | |
| Question Answering | BioASQ | Accuracy90.24 | 5 | |
| Question Answering | PubMedQA | Accuracy73 | 5 |