BiMediX: Bilingual Medical Mixture of Experts LLM

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Pieri, Sara, Mullappilly, Sahal Shaji, Khan, Fahad Shahbaz, Anwer, Rao Muhammad, Khan, Salman, Baldwin, Timothy, Cholakkal, Hisham
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866929623203315712
author Pieri, Sara
Mullappilly, Sahal Shaji
Khan, Fahad Shahbaz
Anwer, Rao Muhammad
Khan, Salman
Baldwin, Timothy
Cholakkal, Hisham
author_facet Pieri, Sara
Mullappilly, Sahal Shaji
Khan, Fahad Shahbaz
Anwer, Rao Muhammad
Khan, Salman
Baldwin, Timothy
Cholakkal, Hisham
contents In this paper, we introduce BiMediX, the first bilingual medical mixture of experts LLM designed for seamless interaction in both English and Arabic. Our model facilitates a wide range of medical interactions in English and Arabic, including multi-turn chats to inquire about additional details such as patient symptoms and medical history, multiple-choice question answering, and open-ended question answering. We propose a semi-automated English-to-Arabic translation pipeline with human refinement to ensure high-quality translations. We also introduce a comprehensive evaluation benchmark for Arabic medical LLMs. Furthermore, we introduce BiMed1.3M, an extensive Arabic-English bilingual instruction set covering 1.3 Million diverse medical interactions, resulting in over 632 million healthcare specialized tokens for instruction tuning. Our BiMed1.3M dataset includes 250k synthesized multi-turn doctor-patient chats and maintains a 1:2 Arabic-to-English ratio. Our model outperforms state-of-the-art Med42 and Meditron by average absolute gains of 2.5% and 4.1%, respectively, computed across multiple medical evaluation benchmarks in English, while operating at 8-times faster inference. Moreover, our BiMediX outperforms the generic Arabic-English bilingual LLM, Jais-30B, by average absolute gains of 10% on our Arabic medical benchmark and 15% on bilingual evaluations across multiple datasets. Our project page with source code and trained model is available at https://github.com/mbzuai-oryx/BiMediX .
format Preprint
id arxiv_https___arxiv_org_abs_2402_13253
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle BiMediX: Bilingual Medical Mixture of Experts LLM
Pieri, Sara
Mullappilly, Sahal Shaji
Khan, Fahad Shahbaz
Anwer, Rao Muhammad
Khan, Salman
Baldwin, Timothy
Cholakkal, Hisham
Computation and Language
In this paper, we introduce BiMediX, the first bilingual medical mixture of experts LLM designed for seamless interaction in both English and Arabic. Our model facilitates a wide range of medical interactions in English and Arabic, including multi-turn chats to inquire about additional details such as patient symptoms and medical history, multiple-choice question answering, and open-ended question answering. We propose a semi-automated English-to-Arabic translation pipeline with human refinement to ensure high-quality translations. We also introduce a comprehensive evaluation benchmark for Arabic medical LLMs. Furthermore, we introduce BiMed1.3M, an extensive Arabic-English bilingual instruction set covering 1.3 Million diverse medical interactions, resulting in over 632 million healthcare specialized tokens for instruction tuning. Our BiMed1.3M dataset includes 250k synthesized multi-turn doctor-patient chats and maintains a 1:2 Arabic-to-English ratio. Our model outperforms state-of-the-art Med42 and Meditron by average absolute gains of 2.5% and 4.1%, respectively, computed across multiple medical evaluation benchmarks in English, while operating at 8-times faster inference. Moreover, our BiMediX outperforms the generic Arabic-English bilingual LLM, Jais-30B, by average absolute gains of 10% on our Arabic medical benchmark and 15% on bilingual evaluations across multiple datasets. Our project page with source code and trained model is available at https://github.com/mbzuai-oryx/BiMediX .
title BiMediX: Bilingual Medical Mixture of Experts LLM
topic Computation and Language
url https://arxiv.org/abs/2402.13253