Update README.md
Browse files
README.md
CHANGED
@@ -5,7 +5,7 @@ license: apache-2.0
|
|
5 |
# MedReason: Eliciting Factual Medical Reasoning Steps in LLMs via Knowledge Graphs
|
6 |
|
7 |
<p align="center">
|
8 |
-
π <a href="https://arxiv.org/" target="_blank">Paper</a> ο½π€ <a href="https://huggingface.co/UCSC-VLAA/MedReason-8B" target="_blank">MedReason-8B</a> | π <a href="https://huggingface.co/UCSC-VLAA/MedReason
|
9 |
</p>
|
10 |
|
11 |
|
@@ -47,4 +47,23 @@ messages = [{"role": "user", "content": input_text}]
|
|
47 |
inputs = tokenizer(tokenizer.apply_chat_template(messages, tokenize=False,add_generation_prompt=True), return_tensors="pt").to(model.device)
|
48 |
outputs = model.generate(**inputs, max_new_tokens=2048)
|
49 |
print(tokenizer.decode(outputs[0], skip_special_tokens=True))
|
50 |
-
```
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
5 |
# MedReason: Eliciting Factual Medical Reasoning Steps in LLMs via Knowledge Graphs
|
6 |
|
7 |
<p align="center">
|
8 |
+
π <a href="https://arxiv.org/abs/2504.00993" target="_blank">Paper</a> ο½π€ <a href="https://huggingface.co/UCSC-VLAA/MedReason-8B" target="_blank">MedReason-8B</a> | π <a href="https://huggingface.co/datasets/UCSC-VLAA/MedReason" target="_blank">MedReason Data</a>
|
9 |
</p>
|
10 |
|
11 |
|
|
|
47 |
inputs = tokenizer(tokenizer.apply_chat_template(messages, tokenize=False,add_generation_prompt=True), return_tensors="pt").to(model.device)
|
48 |
outputs = model.generate(**inputs, max_new_tokens=2048)
|
49 |
print(tokenizer.decode(outputs[0], skip_special_tokens=True))
|
50 |
+
```
|
51 |
+
|
52 |
+
## ππΌ Acknowledgement
|
53 |
+
|
54 |
+
We gratefully acknowledge the inspiring work of [HuatuoGPT-o1](https://github.com/FreedomIntelligence/HuatuoGPT-o1), which laid important groundwork for this research. We also thank the developers of the excellent tools [curator](https://github.com/bespokelabsai/curator/), [trl](https://github.com/huggingface/trl), and [sglang](https://github.com/sgl-project/sglang) for making this work possible.
|
55 |
+
|
56 |
+
## π Citation
|
57 |
+
|
58 |
+
```
|
59 |
+
@misc{wu2025medreasonelicitingfactualmedical,
|
60 |
+
title={MedReason: Eliciting Factual Medical Reasoning Steps in LLMs via Knowledge Graphs},
|
61 |
+
author={Juncheng Wu and Wenlong Deng and Xingxuan Li and Sheng Liu and Taomian Mi and Yifan Peng and Ziyang Xu and Yi Liu and Hyunjin Cho and Chang-In Choi and Yihan Cao and Hui Ren and Xiang Li and Xiaoxiao Li and Yuyin Zhou},
|
62 |
+
year={2025},
|
63 |
+
eprint={2504.00993},
|
64 |
+
archivePrefix={arXiv},
|
65 |
+
primaryClass={cs.CL},
|
66 |
+
url={https://arxiv.org/abs/2504.00993},
|
67 |
+
}
|
68 |
+
```
|
69 |
+
|