55 lines
1.6 KiB
Markdown
55 lines
1.6 KiB
Markdown
|
---
|
|||
|
|
license: apache-2.0
|
|||
|
|
tags:
|
|||
|
|
- unsloth
|
|||
|
|
- trl
|
|||
|
|
- sft
|
|||
|
|
datasets:
|
|||
|
|
- suayptalha/Diagnose-Instructions
|
|||
|
|
language:
|
|||
|
|
- en
|
|||
|
|
base_model:
|
|||
|
|
- Qwen/Qwen3-0.6B
|
|||
|
|
pipeline_tag: text-generation
|
|||
|
|
library_name: transformers
|
|||
|
|
---
|
|||
|
|
|
|||
|
|
# Qwen3-0.6B-Diagnosis-Expert
|
|||
|
|
|
|||
|
|
This project performs full fine-tuning on the **Qwen3-0.6B** language model to enhance its clinical diagnosis interpretation and reasoning capabilities. The model was optimized using the bfloat16 (bf16) data type.
|
|||
|
|
|
|||
|
|
## Training Procedure
|
|||
|
|
|
|||
|
|
1. **Dataset Preparation**
|
|||
|
|
|
|||
|
|
* Dataset: Containing paired clinical patient histories and step-by-step diagnostic conclusions.
|
|||
|
|
|
|||
|
|
2. **Model Loading and Configuration**
|
|||
|
|
|
|||
|
|
* Base model: **Qwen3-0.6B**, loaded with the `unsloth` library in bf16 precision.
|
|||
|
|
* Full fine-tuning (`full_finetuning=True`) applied to all layers to adapt the model for medical diagnostic tasks.
|
|||
|
|
|
|||
|
|
3. **Supervised Fine-Tuning (SFT)**
|
|||
|
|
|
|||
|
|
* Utilized the Hugging Face TRL library with the Supervised Fine-Tuning approach.
|
|||
|
|
* The model was trained to generate both intermediate reasoning steps and final diagnostic statements.
|
|||
|
|
* Training hyperparameters:
|
|||
|
|
|
|||
|
|
* Epochs: 2
|
|||
|
|
* Learning rate: 2e-5
|
|||
|
|
* Batch size: 8
|
|||
|
|
|
|||
|
|
## Purpose and Outcome
|
|||
|
|
|
|||
|
|
* Significantly improved the model’s ability to interpret clinical information and propose accurate, structured diagnoses.
|
|||
|
|
|
|||
|
|
## Evaluation
|
|||
|
|
|
|||
|
|
* Performance was measured on a held-out validation set with the following metric:
|
|||
|
|
|
|||
|
|
* **Diagnostic Similarity:** 71.68% similarity compared to DeepSeek V3-0324 baseline.
|
|||
|
|
|
|||
|
|
## License
|
|||
|
|
|
|||
|
|
This project is licensed under the Apache License 2.0. See the [LICENSE](./LICENSE) file for details.
|