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 |
+
|