BioMamba: Domain-Adaptive Biomedical Language Models
Background: Biomedical language models should improve performance on biomedical text while retaining general-domain language ability. For Mamba-based models, this trade-off has not been clearly studied across biomedical literature and clinical text. Methods: We developed BioMamba, a family of biomedical models obtained by continued pretraining of public Mamba2 checkpoints on PubMed, with small amounts of general-domain data from the Colossal Clean Crawled Corpus (C4) and Wikipedia included to help preserve general-domain language ability. We evaluated language modeling and three downstream tasks across multiple model scales: clinical note completion, discharge summary generation, and biomedical yes/no question answering. Results: BioMamba consistently improved PubMed modeling, improved Wikipedia modeling, and left C4 performance largely unchanged. After supervised fine-tuning, BioMamba transferred well to both biomedical literature and clinical text, yielding strong results on completion, summarization, and question answering. On MIMIC-IV, BioMamba+SFT consistently matched or exceeded SFT from the corresponding base checkpoints across note completion and discharge summary generation. The strongest model achieved a PubMed perplexity of 5.28 and accuracies of 90.24% and 73.00% on BioASQ and PubMedQA, respectively. Conclusion: Balanced domain-adaptive pretraining strategy strengthens Mamba language models for both biomedical literature and clinical text, while preserving general-domain language capabilities, establishing BioMamba as a practical foundation for biomedical NLP applications.
Authors
- Tianfan Fu (ORCID: https://orcid.org/0000-0002-5574-2541)
- Ling Yue
- Sixue Xing
Institutions
- IBM (United States) (US)
- North University of China (CN)
- Rensselaer Polytechnic Institute (US)
- Jiaxing University (CN)
- First Hospital of Jiaxing (CN)
Publication Details
- Journal
- Health Data Science
- Published
- 2026-05-12
- DOI
- https://doi.org/10.34133/hds.0454
- Citations
- 2
- Primary Topic
- Topic Modeling
- Type
- preprint
- Field-Weighted Citation Impact
- 0.00