Beta

Explore
Marketplace
Neural Labs
Chat
Wallet
Docs

Beta

Explore Marketplace Neural Labs Wallet Docs

Build

Manage

My assets Manage wallet Organization

Community

Discord X (Twitter)Telegram

Docs Privacy Policy Terms and Conditions

Proud member of

Proud member of

Docs Cookie Policy

Privacy Policy Terms and Conditions

© 2026 AlphaNeural AI. All rights reserved.

medgemma-anatomy-v2.1 – AI Model by krishna195 | AlphaNeural AI

You can deploy this model and start earning money today!

krishna195/
medgemma-anatomy-v2.1

like

0

image-text-to-text

clinical-reasoning

text-generation

google/medgemma-4b-it

text-generation-inference

endpoints_compatible

Views

No views yet

MedGemma-4B Anatomy v2.1 (Optimized)

Improved version with better generalization - fixes overfitting from v2.0.

Key Improvements

✅ Reduced epochs: 6 → 3 (prevents overfitting)
✅ Early stopping: Stops when validation loss plateaus
✅ Stronger regularization: Increased dropout and weight decay
✅ Better convergence: Higher learning rate with more warmup

Model Details

Base Model: google/medgemma-4b-it (4B parameters)
Training Data: 895 anatomy Q&A pairs
Method: LoRA (r=32, α=64, dropout=0.1)
Epochs: 3 (with early stopping)
Training Time: ~0.2 hours
Hardware: A100 40GB GPU
Final Train Loss: 1.3326
Best Val Loss: 1.2016

Training Configuration

python

CONFIG = {
    'max_seq_length': 1024,
    'num_epochs': 3,
    'batch_size': 4 (effective 16),
    'learning_rate': 0.0001,
    'lora_r': 32,
    'lora_alpha': 64,
    'lora_dropout': 0.1,
    'weight_decay': 0.03,
    'early_stopping_patience': 5
}

Usage

python

from transformers import AutoTokenizer, AutoModelForCausalLM
import torch

model_name = "krishna195/medgemma-anatomy-v2.1"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForCausalLM.from_pretrained(
    model_name,
    torch_dtype=torch.bfloat16,
    device_map="auto"
)

question = "What is the carpal tunnel?"
prompt = f"<start_of_turn>user\n{question}<end_of_turn>\n<start_of_turn>model\n"

inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
outputs = model.generate(**inputs, max_new_tokens=512, temperature=0.7)
print(tokenizer.decode(outputs[0], skip_special_tokens=True))

License

Apache 2.0