File size: 8,312 Bytes
b0c7556 42ff37e b0c7556 42ff37e 09e8ada 42ff37e 09e8ada 42ff37e f775d2a 09e8ada 42ff37e 09e8ada 42ff37e 09e8ada 42ff37e 09e8ada 42ff37e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 |
---
license: apache-2.0
language:
- en
library_name: transformers
---
# Model Card MELT-Mixtral-8x7B-Instruct-v0.1
The MELT-Mixtral-8x7B-Instruct-v0.1 Large Language Model (LLM) is a pretrained generative text model pre-trained and fine-tuned on using publically avalable medical data.
MELT-Mixtral-8x7B-Instruct-v0.1 is 68.2% accurate across 3 USMLE benchmarks, surpassing the pass mark (>60%) in the U.S. Medical Licensing Examination (USMLE) style questions.
To the best of our understanding our model is 4% less accurate than Google's 540 billion parameter [Med-Palm](https://sites.research.google/med-palm/), which is 10X larger.
## Model Details
The Medical Education Language Transformer (MELT) models have been trained on a wide-range of text, chat, Q/A, and instruction data in the medical domain.
While the model was evaluated using publically avalable [USMLE](https://www.usmle.org/) example questions, its use it intented to be more broadly applicable.
### Model Description
<!-- Provide a longer summary of what this model is. -->
- **Developed by:** [Center for Applied AI](https://caai.ai.uky.edu/)
- **Funded by:** [Institute or Biomedical Informatics](https://www.research.uky.edu/IBI)
- **Model type:** [More Information Needed]
- **Language(s) (NLP):** English
- **License:** Apache 2.0
- **Finetuned from model:** [Mixtral-8x7B-Instruct-v0.1](https://huggingface.co./mistralai/Mixtral-8x7B-Instruct-v0.1)
## Uses
MELT is intended for research purposes only. MELT models are best suited for prompts using a QA or chat format.
### Out-of-Scope Use
<!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->
MELT is intended for research purposes only and should not be used for medical advice.
## Bias, Risks, and Limitations
<!-- This section is meant to convey both technical and sociotechnical limitations. -->
MELT was training using collections publicly available, which likely contain biased and inaccurate information. The training and evaluation datasets have not been evaluated for content or accuracy.
## How to Get Started with the Model
Use this model like you would the Mixtral-8x7B-Instruct-v0.1 model.
## Training Details
### Training Data
<!-- This should link to a Dataset Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->
The following datasets were used for training:
[Expert Med](https://dataverse.harvard.edu/dataset.xhtml?persistentId=doi:10.7910/DVN/Q3A969)
[MedQA train](https://huggingface.co./datasets/bigbio/med_qa)
[MedMCQA train](https://github.com/MedMCQA/MedMCQA?tab=readme-ov-file#data-download-and-preprocessing)
[LiveQA](https://github.com/abachaa/LiveQA_MedicalTask_TREC2017)
[MedicationQA](https://huggingface.co./datasets/truehealth/medicationqa)
[MMLU clinical topics](https://huggingface.co./datasets/Stevross/mmlu)
[Medical Flashcards](https://huggingface.co./datasets/medalpaca/medical_meadow_medical_flashcards)
[Wikidoc](https://huggingface.co./datasets/medalpaca/medical_meadow_wikidoc)
[Wikidoc Patient Information](https://huggingface.co./datasets/medalpaca/medical_meadow_wikidoc_patient_information)
[MEDIQA](https://huggingface.co./datasets/medalpaca/medical_meadow_mediqa)
[MMMLU](https://huggingface.co./datasets/medalpaca/medical_meadow_mmmlu)
[icliniq 10k](https://drive.google.com/file/d/1ZKbqgYqWc7DJHs3N9TQYQVPdDQmZaClA/view?usp=sharing)
[HealthCare Magic 100k](https://drive.google.com/file/d/1lyfqIwlLSClhgrCutWuEe_IACNq6XNUt/view?usp=sharing)
[GenMedGPT-5k](https://drive.google.com/file/d/1nDTKZ3wZbZWTkFMBkxlamrzbNz0frugg/view?usp=sharing)
[Mental Health Conversational](https://huggingface.co./datasets/heliosbrahma/mental_health_conversational_dataset)
### Training Procedure
<!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
#### Training Hyperparameters
- **Lora Rank:** 64
- **Lora Alpha:** 16
- **Lora Targets:** "o_proj","down_proj","v_proj","gate_proj","up_proj","k_proj","q_proj"
- **LR:** 2e-4
- **Epoch:** 3
- **Precision:** bf16 <!--fp32, fp16 mixed precision, bf16 mixed precision, bf16 non-mixed precision, fp16 non-mixed precision, fp8 mixed precision -->
## Evaluation
<!-- This section describes the evaluation protocols and provides the results. -->
MELT-Mixtral-8x7B-Instruct-v0.1 demonstrated a average 4.42% improvement over Mixtral-8x7B-Instruct-v0.1 across 3 USMLE benchmarks.
The base Mixtral-8x7B-Instruct-v0.1 model already performs signifigantly better (65.31%) than the base [llama-2-7b-chat-hf](https://huggingface.co./meta-llama/Llama-2-7b-chat-hf) model (35.26%) and our [MELT-llama-2-7b-chat-v0.1](https://huggingface.co./IBI-CAAI/MELT-llama-2-7b-chat-v0.1) model (46.33%).
While there was limited improvement on the USMLE benchmarks, our training data contained a broad collection of medical text, chats, and multi-choice questions that would not be captured on the USMLE evaluation.
### Mixtral-8x7B-Instruct-v0.1
- **medqa:** {'base': {'Average': 62.0, 'STEP-1': 62.24, 'STEP-2&3': 61.72}}
- **mausmle:** {'base': {'Average': 73.12, 'STEP-1': 68.24, 'STEP-2': 78.16, 'STEP-3': 72.9}}
- **medmcqa:** {'base': {'Average': 60.82, 'MEDICINE': 56.52, 'OPHTHALMOLOGY': 59.52, 'ANATOMY': 62.33, 'PATHOLOGY': 70.93, 'PHYSIOLOGY': 65.91, 'DENTAL': 52.37, 'RADIOLOGY': 66.07, 'BIOCHEMISTRY': 75.21, 'ANAESTHESIA': 73.91, 'GYNAECOLOGY': 55.56, 'PHARMACOLOGY': 74.72, 'SOCIAL': 50.0, 'PEDIATRICS': 60.61, 'ENT': 73.68, 'SURGERY': 61.69, 'MICROBIOLOGY': 61.64, 'FORENSIC': 69.77, 'PSYCHIATRY': 77.78, 'SKIN': 60.0, 'ORTHOPAEDICS': 71.43, 'UNKNOWN': 100.0}}
- **average:** 65.31%
### MELT-Mixtral-8x7B-Instruct-v0.1
- **medqa:** {'base': {'Average': 67.19, 'STEP-1': 67.55, 'STEP-2&3': 66.78}}
- **mausmle:** {'base': {'Average': 73.84, 'STEP-1': 74.12, 'STEP-2': 75.86, 'STEP-3': 71.96}}
- **medmcqa:** {'base': {'Average': 63.58, 'MEDICINE': 63.04, 'OPHTHALMOLOGY': 66.67, 'ANATOMY': 67.12, 'PATHOLOGY': 72.48, 'PHYSIOLOGY': 67.42, 'DENTAL': 54.15, 'RADIOLOGY': 71.43, 'BIOCHEMISTRY': 80.17, 'ANAESTHESIA': 69.57, 'GYNAECOLOGY': 60.13, 'PHARMACOLOGY': 74.16, 'SOCIAL': 56.67, 'PEDIATRICS': 65.15, 'ENT': 65.79, 'SURGERY': 64.92, 'MICROBIOLOGY': 64.38, 'FORENSIC': 65.12, 'PSYCHIATRY': 88.89, 'SKIN': 70.0, 'ORTHOPAEDICS': 78.57, 'UNKNOWN': 100.0}}
- **average:** 68.2%
### Testing Data, Factors & Metrics
#### Testing Data
<!-- This should link to a Dataset Card if possible. -->
[MedQA test](https://huggingface.co./datasets/bigbio/med_qa)
[MedMCQA test](https://github.com/MedMCQA/MedMCQA?tab=readme-ov-file#data-download-and-preprocessing)
[MA USMLE](https://huggingface.co./datasets/medalpaca/medical_meadow_usmle_self_assessment)
## Disclaimer:
The use of large language models, such as this one, is provided without warranties or guarantees of any kind. While every effort has been made to ensure accuracy, completeness, and reliability of the information generated, it should be noted that these models may produce responses that are inaccurate, outdated, or inappropriate for specific purposes. Users are advised to exercise discretion and judgment when relying on the information generated by these models. The outputs should not be considered as professional, legal, medical, financial, or any other form of advice. It is recommended to seek expert advice or consult appropriate sources for specific queries or critical decision-making. The creators, developers, and providers of these models disclaim any liability for damages, losses, or any consequences arising from the use, reliance upon, or interpretation of the information provided by these models. The user assumes full responsibility for their interactions and usage of the generated content. By using these language models, users agree to indemnify and hold harmless the developers, providers, and affiliates from any claims, damages, or liabilities that may arise from their use. Please be aware that these models are constantly evolving, and their capabilities, limitations, and outputs may change over time without prior notice. Your use of this language model signifies your acceptance and understanding of this disclaimer.
|