AmareshHebbar/hindi-medical-qwen25-3b

<div align="center">

🕉️ Hindi Medical Reasoning

Qwen2.5-3B fine-tuned for hindi medical reasoning

     

Part of the [Medical AI Fine-tuned Model Suite](https://huggingface.co/AmareshHebbar/medical-ai-model-suite) — 16 specialist models, one per task

</div>

TL;DR

Answers medical questions with detailed chain-of-thought clinical reasoning for Hindi-language and Indic medical AI.

INPUT:  A 45-year-old presents with fatigue, weight gain, cold intolerance and bradycardia for 6 months.
OUTPUT: Chain-of-thought reasoning through hypothyroidism differential leading to TSH and free T4 workup, then diagnosis and management plan.

Architecture

                  +-------------------------+
  user prompt --> |  Qwen2.5-3B-Instruct  | --> base weights (frozen, 4-bit NF4)
                  |  + LoRA adapter (r=16)  | --> hindi-medical-qwen25-3b
                  +-------------------------+
                              |
                              v
                     structured output
                  (code / JSON / classification)

This repo contains only the LoRA adapter (~60MB), not the full merged weights. Load it on top of the base model as shown below — this keeps the download small and lets you swap adapters on one base model in memory.

Intended use

Fine-tune for Hindi-language medical Q&A and ABDM-compatible clinical assistants.

Direct use

Describe symptoms, get back step-by-step clinical reasoning toward a likely diagnosis.

Downstream use

Power a multilingual symptom-checker chatbot or an Indic-language clinical education tool.

Out of scope

Definitive diagnosis or treatment without physician oversight. The training data is primarily English-sourced reasoning chains; verify Hindi-specific clinical terminology accuracy before production use.

This model is not a substitute for a certified medical professional's judgment. Output should be reviewed by a qualified person before being used in a clinical or billing decision.

Quickstart

Option A — Transformers + PEFT

from transformers import AutoModelForCausalLM, AutoTokenizer
from peft import PeftModel
import torch

base_model = "unsloth/Qwen2.5-3B-Instruct"
adapter    = "AmareshHebbar/hindi-medical-qwen25-3b"

tokenizer = AutoTokenizer.from_pretrained(base_model)
model = AutoModelForCausalLM.from_pretrained(
    base_model,
    torch_dtype=torch.bfloat16,
    device_map="auto",
)
model = PeftModel.from_pretrained(model, adapter)

messages = [
    {"role": "system", "content": "आप एक चिकित्सा सहायक हैं। रोगी के लक्षणों के आधार पर ICD-10 कोड और संभावित निदान प्रदान करें।"},
    {"role": "user", "content": "A 45-year-old presents with fatigue, weight gain, cold intolerance and bradycardia for 6 months."},
]
inputs = tokenizer.apply_chat_template(messages, return_tensors="pt", add_generation_prompt=True).to(model.device)
outputs = model.generate(inputs, max_new_tokens=128, temperature=0.1, do_sample=True)
print(tokenizer.decode(outputs[0][inputs.shape[1]:], skip_special_tokens=True))

Expected output:

Chain-of-thought reasoning through hypothyroidism differential leading to TSH and free T4 workup, then diagnosis and management plan.

Option B — Unsloth (2x faster load + inference)

from unsloth import FastLanguageModel

model, tokenizer = FastLanguageModel.from_pretrained(
    model_name="AmareshHebbar/hindi-medical-qwen25-3b",
    max_seq_length=512,
    load_in_4bit=True,
)
FastLanguageModel.for_inference(model)

messages = [
    {"role": "system", "content": "आप एक चिकित्सा सहायक हैं। रोगी के लक्षणों के आधार पर ICD-10 कोड और संभावित निदान प्रदान करें।"},
    {"role": "user", "content": "A patient reports sudden chest pain radiating to the left arm."},
]
prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
inputs = tokenizer(prompt, return_tensors="pt").to("cuda")
outputs = model.generate(**inputs, max_new_tokens=128, temperature=0.1, do_sample=True)
print(tokenizer.decode(outputs[0][inputs["input_ids"].shape[1]:], skip_special_tokens=True))

Option C — vLLM (production serving, OpenAI-compatible)

vllm serve unsloth/Qwen2.5-3B-Instruct \
    --enable-lora \
    --lora-modules hindi-medical-qwen25-3b=AmareshHebbar/hindi-medical-qwen25-3b \
    --host 0.0.0.0 --port 8000 --dtype bfloat16

from openai import OpenAI

client = OpenAI(base_url="http://localhost:8000/v1", api_key="not-needed")
response = client.chat.completions.create(
    model="hindi-medical-qwen25-3b",
    messages=[
        {"role": "system", "content": "आप एक चिकित्सा सहायक हैं। रोगी के लक्षणों के आधार पर ICD-10 कोड और संभावित निदान प्रदान करें।"},
        {"role": "user", "content": "A child presents with high fever and a rash for five days."},
    ],
    temperature=0.1,
)
print(response.choices[0].message.content)

Option D — GGUF / llama.cpp (CPU / edge inference)

This repo ships LoRA adapter weights, not a pre-merged GGUF. To run on llama.cpp, merge first:

pip install unsloth
python -c "
from unsloth import FastLanguageModel
model, tokenizer = FastLanguageModel.from_pretrained('AmareshHebbar/hindi-medical-qwen25-3b', load_in_4bit=False)
model.save_pretrained_gguf('hindi-medical-qwen25-3b-gguf', tokenizer, quantization_method='q4_k_m')
"

Training details

Data

Trained on 19,704 examples extracted from FreedomIntelligence/medical-o1-reasoning-SFT — medical chain-of-thought reasoning pairs (source). No synthetic or LLM-generated training data — every example pairs real-world input with its authoritative output.

Full extraction pipeline documented on the dataset card.

Hyperparameters

Training compute

Fine-tuned with Unsloth for 2x faster training and reduced VRAM, using TRL's SFTTrainer. Full project: wandb.ai/amareshhebbar-/axiomapper.

Bias, risks & limitations

Data recency. Training data reflects a specific snapshot in time (CMS FY2026 / dataset publish date). Codes, rates, and rules referenced may become outdated as source authorities issue updates — always cross-check against the live authoritative source before high-stakes use.

Failure mode. Like any LLM, this model can produce a plausible-sounding but incorrect output, especially on rare, ambiguous, or highly compound real-world cases that fall outside the training distribution. It does not know when it's wrong.

Language. English-language input only (Hindi-medical model excepted, where Hindi system prompts are used but underlying clinical reasoning data is largely English-sourced).

Not a regulated medical device. This model has not been validated, cleared, or approved by any regulatory body (FDA, CDSCO, or equivalent) as a medical device or clinical decision support tool. It is a research/engineering artifact.

Misapplication risk. Do not use this model as the sole basis for a clinical, billing, or compliance decision affecting a real patient or claim. Do not deploy in an emergency triage context without a human-in-the-loop and clear escalation paths.

FAQ

Q: Can I merge the adapter into the base model for faster inference? Yes — use model.merge_and_unload() after loading with PEFT, or use Unsloth's save_pretrained_merged() method.

Q: Why QLoRA instead of full fine-tuning? The base model already has strong language and medical knowledge from pretraining. QLoRA adapts only ~0.5-1% of parameters, which is enough to specialize the output format and domain without the cost or overfitting risk of full fine-tuning.

Q: Can I fine-tune this further on my own data? Yes, this adapter can be used as a starting checkpoint for continued fine-tuning. Note this may require merging first depending on your training framework.

Q: Why is the output format so strict? Each task was trained on a fixed system prompt and consistent output structure. Following the documented system prompt closely (see Quickstart above) gives the most reliable results — deviating from it may produce inconsistent formatting.

Q: Does this model store or transmit my input data? No. Like any open-weight model, all inference happens locally on your own infrastructure (or wherever you deploy it) — nothing is sent back to the model author.

Troubleshooting

Related models in this suite

Full suite overview: AmareshHebbar/medical-ai-model-suite

Changelog

Citation

@misc{medicalai2026,
  author    = {Hebbar, Amaresh},
  title     = {Medical AI Fine-tuning Suite},
  year      = {2026},
  publisher = {HuggingFace},
  url       = {https://huggingface.co/AmareshHebbar}
}

Contact