🚀 BioMistral:医療分野向けのオープンソース事前学習大規模言語モデルのコレクション
BioMistralは、医療分野に特化したオープンソースの大規模言語モデルです。Mistralを基礎モデルとし、PubMed Central上でさらに事前学習を行っています。このモデルは、医療分野の質問応答タスクで優れた性能を発揮し、既存のオープンソース医療モデルや独自のモデルと比較しても競争力があります。
🚀 クイックスタート
BioMistralを使うには、Hugging FaceのTransformersライブラリを使用できます。以下は、モデルとトークナイザーをロードするコード例です。
基本的な使用法
from transformers import AutoModel, AutoTokenizer
tokenizer = AutoTokenizer.from_pretrained("BioMistral/BioMistral-7B")
model = AutoModel.from_pretrained("BioMistral/BioMistral-7B")
✨ 主な機能
- 医療分野特化:Mistralをベースに、PubMed Central上で事前学習された医療分野向けの大規模言語モデル。
- 多言語対応:英語以外の7つの言語での評価が行われ、多言語汎化能力を持つ。
- 軽量化モデル:量子化やモデルマージアプローチを用いた軽量化モデルが提供される。
📦 インストール
READMEに具体的なインストール手順が記載されていないため、このセクションをスキップします。
💻 使用例
基本的な使用法
from transformers import AutoModel, AutoTokenizer
tokenizer = AutoTokenizer.from_pretrained("BioMistral/BioMistral-7B")
model = AutoModel.from_pretrained("BioMistral/BioMistral-7B")
📚 ドキュメント
概要
大規模言語モデル(LLM)は近年、医療や医学などの専門分野での応用が期待されています。しかし、汎用的なLLMを医療分野に適応させることは大きな課題です。BioMistralは、この問題を解決するために開発されたオープンソースのLLMです。
モデルの詳細
BioMistralモデル
量子化モデル
ベースモデル |
方法 |
q_group_size |
w_bit |
バージョン |
VRAM GB |
時間 |
ダウンロード |
BioMistral-7B |
FP16/BF16 |
|
|
|
15.02 |
x1.00 |
HuggingFace |
BioMistral-7B |
AWQ |
128 |
4 |
GEMM |
4.68 |
x1.41 |
HuggingFace |
BioMistral-7B |
AWQ |
128 |
4 |
GEMV |
4.68 |
x10.30 |
HuggingFace |
BioMistral-7B |
BnB.4 |
|
4 |
|
5.03 |
x3.25 |
HuggingFace |
BioMistral-7B |
BnB.8 |
|
8 |
|
8.04 |
x4.34 |
HuggingFace |
BioMistral-7B-DARE |
AWQ |
128 |
4 |
GEMM |
4.68 |
x1.41 |
HuggingFace |
BioMistral-7B-TIES |
AWQ |
128 |
4 |
GEMM |
4.68 |
x1.41 |
HuggingFace |
BioMistral-7B-SLERP |
AWQ |
128 |
4 |
GEMM |
4.68 |
x1.41 |
HuggingFace |
教師付き微調整ベンチマーク
|
臨床知識グラフ |
医療遺伝学 |
解剖学 |
専門医療 |
大学生物学 |
大学医学 |
MedQA |
MedQA 5選択肢 |
PubMedQA |
MedMCQA |
平均 |
BioMistral 7B |
59.9 |
64.0 |
56.5 |
60.4 |
59.0 |
54.7 |
50.6 |
42.8 |
77.5 |
48.1 |
57.3 |
Mistral 7B Instruct |
62.9 |
57.0 |
55.6 |
59.4 |
62.5 |
57.2 |
42.0 |
40.9 |
75.7 |
46.1 |
55.9 |
|
|
|
|
|
|
|
|
|
|
|
|
BioMistral 7B Ensemble |
62.8 |
62.7 |
57.5 |
63.5 |
64.3 |
55.7 |
50.6 |
43.6 |
77.5 |
48.8 |
58.7 |
BioMistral 7B DARE |
62.3 |
67.0 |
55.8 |
61.4 |
66.9 |
58.0 |
51.1 |
45.2 |
77.7 |
48.7 |
59.4 |
BioMistral 7B TIES |
60.1 |
65.0 |
58.5 |
60.5 |
60.4 |
56.5 |
49.5 |
43.2 |
77.5 |
48.1 |
57.9 |
BioMistral 7B SLERP |
62.5 |
64.7 |
55.8 |
62.7 |
64.8 |
56.3 |
50.8 |
44.3 |
77.8 |
48.6 |
58.8 |
|
|
|
|
|
|
|
|
|
|
|
|
MedAlpaca 7B |
53.1 |
58.0 |
54.1 |
58.8 |
58.1 |
48.6 |
40.1 |
33.7 |
73.6 |
37.0 |
51.5 |
PMC-LLaMA 7B |
24.5 |
27.7 |
35.3 |
17.4 |
30.3 |
23.3 |
25.5 |
20.2 |
72.9 |
26.6 |
30.4 |
MediTron-7B |
41.6 |
50.3 |
46.4 |
27.9 |
44.4 |
30.8 |
41.6 |
28.1 |
74.9 |
41.3 |
42.7 |
BioMedGPT-LM-7B |
51.4 |
52.0 |
49.4 |
53.3 |
50.7 |
49.1 |
42.5 |
33.9 |
76.8 |
37.6 |
49.7 |
|
|
|
|
|
|
|
|
|
|
|
|
GPT-3.5 Turbo 1106* |
74.71 |
74.00 |
65.92 |
72.79 |
72.91 |
64.73 |
57.71 |
50.82 |
72.66 |
53.79 |
66.0 |
BioMistral 7Bモデルの教師付き微調整(SFT)性能をベースラインと比較した結果です。精度(↑)で測定され、3ショットの3つのランダムシードで平均化されています。DARE、TIES、およびSLERPは、BioMistral 7BとMistral 7B Instructを結合するモデルマージ戦略です。最良のモデルは太字、2番目に良いモデルは下線で示されています。*GPT-3.5 Turboの性能は、SFTなしの3ショットの結果から報告されています。
🔧 技術詳細
READMEに具体的な技術詳細が記載されていないため、このセクションをスキップします。
📄 ライセンス
このプロジェクトは、Apache 2.0ライセンスの下で公開されています。
引用
Arxiv : https://arxiv.org/abs/2402.10373
@misc{labrak2024biomistral,
title={BioMistral: A Collection of Open-Source Pretrained Large Language Models for Medical Domains},
author={Yanis Labrak and Adrien Bazoge and Emmanuel Morin and Pierre-Antoine Gourraud and Mickael Rouvier and Richard Dufour},
year={2024},
eprint={2402.10373},
archivePrefix={arXiv},
primaryClass={cs.CL}
}
⚠️ 重要提示
BioMistralは、高品質の証拠に基づく医療知識をカプセル化することを目的としていますが、専門的なパラメータ内でこの知識を効果的、安全、または適切に伝達するようには調整されていません。特定のユースケースに完全にアライメントされ、実世界の医療環境でのランダム化比較試験を含むさらなるテストを受けるまで、医療コンテキストでのBioMistralの使用は避けることをお勧めします。BioMistral 7Bには、まだ十分に評価されていない固有のリスクとバイアスが存在する可能性があります。また、このモデルの性能は実世界の臨床環境で評価されていません。したがって、BioMistral 7Bは厳密に研究ツールとしてのみ使用することをお勧めし、自然言語生成や専門的な健康医療目的のための本番環境での展開は避けるようにしてください。
💡 使用建议
このモデルは研究目的での使用を想定しています。医療現場での使用は避け、十分なテストとアライメントが行われるまでは、本番環境での展開は行わないでください。