Text Classification
Transformers
English
medical
triage
emergency-medicine
esi
biomedbert
clinical-nlp
decision-support
not-for-clinical-use
Eval Results (legacy)
Instructions to use vadimbelsky/bert-esi-triage-v57 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use vadimbelsky/bert-esi-triage-v57 with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="vadimbelsky/bert-esi-triage-v57")# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("vadimbelsky/bert-esi-triage-v57", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Rename repo to bert-esi-triage-v57; align with v49/v50 convention; update internal references
Browse files
README.md
CHANGED
|
@@ -15,7 +15,7 @@ tags:
|
|
| 15 |
- not-for-clinical-use
|
| 16 |
base_model: microsoft/BiomedNLP-BiomedBERT-base-uncased-abstract-fulltext
|
| 17 |
model-index:
|
| 18 |
-
- name: esi-triage-v57
|
| 19 |
results:
|
| 20 |
- task:
|
| 21 |
type: text-classification
|
|
@@ -68,11 +68,11 @@ model-index:
|
|
| 68 |
value: 0.943
|
| 69 |
---
|
| 70 |
|
| 71 |
-
# ESI
|
| 72 |
|
| 73 |
## TL;DR
|
| 74 |
|
| 75 |
-
`esi-triage-v57` is a fine-tuned BiomedBERT classifier for the
|
| 76 |
**Emergency Severity Index (ESI 1-5)** triage decision. It is the first
|
| 77 |
model in this line to meet ESI 1 safety recall **≥75% on every primary
|
| 78 |
eval slice** (MIMIC, MC-MED, MIETIC, Lukina), with a mean ESI 1 recall
|
|
@@ -125,7 +125,7 @@ class V57BertESI(nn.Module):
|
|
| 125 |
|
| 126 |
tok = AutoTokenizer.from_pretrained(ENCODER)
|
| 127 |
model = V57BertESI()
|
| 128 |
-
ckpt = torch.load(hf_hub_download("vadimbelsky/esi-triage-v57", "model.pt"),
|
| 129 |
map_location="cpu", weights_only=False)
|
| 130 |
# Many auxiliary heads in the checkpoint are not needed for ESI inference.
|
| 131 |
sd = model.state_dict()
|
|
@@ -317,7 +317,7 @@ prevalence of ESI 1 varies materially across input formats
|
|
| 317 |
engine ensemble safety floor},
|
| 318 |
author = {Belski, Vadim},
|
| 319 |
year = {2026},
|
| 320 |
-
url = {https://huggingface.co/vadimbelsky/esi-triage-v57},
|
| 321 |
note = {Validated on 18-eval suite: MIMIC-IV-ED holdout, MC-MED
|
| 322 |
Stanford clean, MIETIC narrative clean, Lukina v3,
|
| 323 |
7 condition-specific slices, 6 subgroup slices. v57 epoch 3.
|
|
@@ -335,3 +335,4 @@ prevalence of ESI 1 varies materially across input formats
|
|
| 335 |
retired in favor of v57 for new deployments).
|
| 336 |
- **Successor:** v58 (training in progress as of 2026-06-01; targets
|
| 337 |
six corpus fixes for tightened ESI 2 / ESI 5 boundaries).
|
|
|
|
|
|
| 15 |
- not-for-clinical-use
|
| 16 |
base_model: microsoft/BiomedNLP-BiomedBERT-base-uncased-abstract-fulltext
|
| 17 |
model-index:
|
| 18 |
+
- name: bert-esi-triage-v57
|
| 19 |
results:
|
| 20 |
- task:
|
| 21 |
type: text-classification
|
|
|
|
| 68 |
value: 0.943
|
| 69 |
---
|
| 70 |
|
| 71 |
+
# BERT-ESI-Triage v57 — BiomedBERT multi-head triage classifier
|
| 72 |
|
| 73 |
## TL;DR
|
| 74 |
|
| 75 |
+
`bert-esi-triage-v57` is a fine-tuned BiomedBERT classifier for the
|
| 76 |
**Emergency Severity Index (ESI 1-5)** triage decision. It is the first
|
| 77 |
model in this line to meet ESI 1 safety recall **≥75% on every primary
|
| 78 |
eval slice** (MIMIC, MC-MED, MIETIC, Lukina), with a mean ESI 1 recall
|
|
|
|
| 125 |
|
| 126 |
tok = AutoTokenizer.from_pretrained(ENCODER)
|
| 127 |
model = V57BertESI()
|
| 128 |
+
ckpt = torch.load(hf_hub_download("vadimbelsky/bert-esi-triage-v57", "model.pt"),
|
| 129 |
map_location="cpu", weights_only=False)
|
| 130 |
# Many auxiliary heads in the checkpoint are not needed for ESI inference.
|
| 131 |
sd = model.state_dict()
|
|
|
|
| 317 |
engine ensemble safety floor},
|
| 318 |
author = {Belski, Vadim},
|
| 319 |
year = {2026},
|
| 320 |
+
url = {https://huggingface.co/vadimbelsky/bert-esi-triage-v57},
|
| 321 |
note = {Validated on 18-eval suite: MIMIC-IV-ED holdout, MC-MED
|
| 322 |
Stanford clean, MIETIC narrative clean, Lukina v3,
|
| 323 |
7 condition-specific slices, 6 subgroup slices. v57 epoch 3.
|
|
|
|
| 335 |
retired in favor of v57 for new deployments).
|
| 336 |
- **Successor:** v58 (training in progress as of 2026-06-01; targets
|
| 337 |
six corpus fixes for tightened ESI 2 / ESI 5 boundaries).
|
| 338 |
+
|