Finemedlm O1

🚀 FineMedLM-o1

FineMedLM-o1は、高度な医療推論のために設計された専門の医療大規模言語モデル（LLM）です。このモデルは多段階の推論プロセスを採用しており、最終的な回答を出す前に繰り返し思考プロセスを見直し、洗練します。

🚀 クイックスタート

FineMedLM-o1を使用するには、Llama-3.1-8B-Instructと同じ方法で利用できます。

基本的な使用法

from transformers import AutoModelForCausalLM, AutoTokenizer

model = AutoModelForCausalLM.from_pretrained("hongzhouyu/FineMedLM-o1")
tokenizer = AutoTokenizer.from_pretrained("hongzhouyu/FineMedLM-o1")

prompt = "How do the interactions between neuronal activity, gonadal hormones, and neurotrophins influence axon regeneration post-injury, and what are the potential therapeutic implications of this research? Please think step by step."
messages = [
    {"role": "system", "content": "You are a helpful professional doctor."},
    {"role": "user", "content": prompt}
]
text = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
model_inputs = tokenizer([text], return_tensors="pt")

generated_ids = model.generate(
    model_inputs.input_ids,
    max_new_tokens=4096
)
generated_ids = [
    output_ids[len(input_ids):] for input_ids, output_ids in zip(model_inputs.input_ids, generated_ids)
]
response = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]

print(response)

FineMedLM-o1はslow-thinkingアプローチを採用しており、出力は以下のように整形されます。

**Thought**
[Reasoning process]

**Summarization**
[Output]

📚 ドキュメント

モデル情報

リンク

• GitHub
• 論文

引用

@misc{yu2025finemedlmo1enhancingmedicalreasoning,
    title={FineMedLM-o1: Enhancing the Medical Reasoning Ability of LLM from Supervised Fine-Tuning to Test-Time Training}, 
    author={Hongzhou Yu and Tianhao Cheng and Ying Cheng and Rui Feng},
    year={2025},
    eprint={2501.09213},
    archivePrefix={arXiv},
    primaryClass={cs.CL},
    url={https://arxiv.org/abs/2501.09213}, 
}