Update README.md
Browse files
README.md
CHANGED
|
@@ -14,9 +14,9 @@ base_model:
|
|
| 14 |
- Qwen/Qwen3-8B
|
| 15 |
---
|
| 16 |
|
| 17 |
-
# EHR-R1-
|
| 18 |
|
| 19 |
-
This repository contains the **EHR-R1-
|
| 20 |
|
| 21 |
**EHR-R1** is a family of reasoning-enhanced Large Language Models (LLMs) specifically tailored for Electronic Health Record (EHR) analysis. It is developed based on **EHR-Ins**, a large-scale, comprehensive EHR reasoning instruction dataset, and is trained through a multi-stage paradigm including domain adaptation, reasoning enhancement, and reinforcement learning. This approach systematically acquires domain knowledge and diverse reasoning capabilities, enabling accurate and robust EHR analysis. The project also introduces **EHR-Bench**, a new benchmark curated from MIMIC-IV for comprehensive assessment across 42 distinct EHR tasks.
|
| 22 |
|
|
@@ -59,7 +59,7 @@ For any EHR data, keep the EHR input with markdown format as below:
|
|
| 59 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 60 |
import torch
|
| 61 |
|
| 62 |
-
model_name = "BlueZeros/EHR-R1-
|
| 63 |
model = AutoModelForCausalLM.from_pretrained(
|
| 64 |
model_name,
|
| 65 |
torch_dtype="auto",
|
|
|
|
| 14 |
- Qwen/Qwen3-8B
|
| 15 |
---
|
| 16 |
|
| 17 |
+
# EHR-R1-8B: A Reasoning-Enhanced Foundational Language Model for Electronic Health Record Analysis
|
| 18 |
|
| 19 |
+
This repository contains the **EHR-R1-8B** model, part of the **EHR-R1** series, as presented in the paper [EHR-R1: A Reasoning-Enhanced Foundational Language Model for Electronic Health Record Analysis](https://huggingface.co/papers/2510.25628).
|
| 20 |
|
| 21 |
**EHR-R1** is a family of reasoning-enhanced Large Language Models (LLMs) specifically tailored for Electronic Health Record (EHR) analysis. It is developed based on **EHR-Ins**, a large-scale, comprehensive EHR reasoning instruction dataset, and is trained through a multi-stage paradigm including domain adaptation, reasoning enhancement, and reinforcement learning. This approach systematically acquires domain knowledge and diverse reasoning capabilities, enabling accurate and robust EHR analysis. The project also introduces **EHR-Bench**, a new benchmark curated from MIMIC-IV for comprehensive assessment across 42 distinct EHR tasks.
|
| 22 |
|
|
|
|
| 59 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 60 |
import torch
|
| 61 |
|
| 62 |
+
model_name = "BlueZeros/EHR-R1-8B" # This specific EHR-R1-8B model
|
| 63 |
model = AutoModelForCausalLM.from_pretrained(
|
| 64 |
model_name,
|
| 65 |
torch_dtype="auto",
|