Jivi-MedX-v2: The Next-Generation Medical Language Model

Jivi-MedX-v2 is a cutting-edge medical language model developed by Jivi AI to support a wide range of clinical applications. Built on the Meta-Llama-3-8B architecture, it has been fine-tuned with Supervised Fine-Tuning (SFT) and Odds Ratio Preference Optimization (ORPO) to enhance its medical reasoning capabilities. The model is designed to generate accurate, domain-specific responses, making it ideal for clinical decision support, medical research, and healthcare automation.

Purpose-Built for the Medical Domain:

Jivi-MedX-v2 is specifically fine-tuned to meet the complex language and knowledge demands of the healthcare industry. Trained on a high-quality, domain-specific dataset, it excels in comprehending and generating precise medical text, making it an invaluable tool for clinical decision support, research, and patient education.

Training Process:

Built on the Meta-Llama-3-8B architecture, Jivi-MedX-v2 has been meticulously refined using Supervised Fine-Tuning (SFT) and Odds Ratio Preference Optimization (ORPO). This ensures that the model aligns effectively with medical terminology and reasoning while maintaining learning efficiency. Hyperparameter tuning strategies have been carefully implemented to prevent catastrophic forgetting, ensuring consistent performance across various tasks.

Data Preparation:

Jivi-MedX-v2 has been trained on a curated dataset of over 1,000,000 data points, covering diverse medical literature, clinical notes, research papers, and diagnostic guidelines. This comprehensive dataset enhances its ability to generate accurate and contextually relevant medical information.

Benchmarks:

With 8 billion parameters, Jivi-MedX-v2 outperforms other models of similar size, delivering best-in-class results on multiple medical benchmarks. It surpasses larger proprietary and open-source models, including GPT-4o and Deepseek-R1, in key performance evaluations, setting a new standard for AI-driven medical intelligence.

How to use

Use with transformers

Please ensure transformers>=4.45.2

import torch
import transformers

model_id = "jiviai/medX_v2"

pipeline = transformers.pipeline(
    "text-generation",
    model=model_id,
    model_kwargs={"torch_dtype": torch.bfloat16},
    device_map="auto",
)

prompt = "Question: A 20-year-old man comes to the physician because of worsening gait unsteadiness and bilateral hearing loss for 1 month. He has had intermittent tingling sensations on both cheeks over this time period. He has no history of serious medical illness and takes no medications. Audiometry shows bilateral sensorineural hearing loss. Genetic evaluation shows a mutation of a tumor suppressor gene on chromosome 22 that encodes merlin. This patient is at increased risk for which of the following conditions?\nA. Renal cell carcinoma\nB. Meningioma\nC. Astrocytoma\nD. Vascular malformations\nAnswer:\n"
gen_kwargs = {
    "return_full_text": False,
    "max_new_tokens": 100,
}
print(pipeline(prompt, **gen_kwargs))

Supported Languages: Currently this model supports English. We are planning to introduce multi-lingual support shortly.

Feedback: To send any feedback/questions please use the community section of the model.

Intended use

The data, code, and model checkpoints are intended to be used solely for:

1. Future research on medical query answering.

Disclaimer: The data, code, and model checkpoints are not intended to be used in clinical care or for any clinical decision-making purposes.

This model has been fine-tuned and optimized for answering medical MCQs. It needs to be further aligned for downstream tasks.