Downloads · 30 days
0
medsiglip-tb-classifier
hongtai91/medsiglip-tb-classifier
medsiglip-tb-classifier is a image classification model from hongtai91. Use it when you need a label for an image. The card lists the license as apache-2.0.
Fine-tuned MLP classifier head on top of Google's MedSigLIP-448 vision encoder for tuberculosis detection from chest X-rays.
Access
Public
Updated Feb 24, 2026
Repo size
2.6 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.pt2.6 MB · 100%
At a glance
- Task
- Image Classification
- License
- apache-2.0
- Access
- Public
- Created
- Feb 24, 2026
- Updated
- Feb 24, 2026
- SHA
- f5012f95
Base models
Model card
From the Hugging Face model README
MedSigLIP TB Classifier
Fine-tuned MLP classifier head on top of Google's MedSigLIP-448 vision encoder for tuberculosis detection from chest X-rays.
Performance
| Metric | Test Set |
|---|---|
| AUC | 0.9975 |
| Sensitivity | 96.2% |
| Specificity | 98.5% |
| F1 Score | 0.944 |
| PPV | 92.7% |
| NPV | 99.3% |
Architecture
- Encoder: MedSigLIP-448 (frozen, 1152-dim embeddings)
- Classifier: 3-layer MLP (1152 → 512 → 128 → 2) with ReLU, dropout=0.3, batch norm
- Training: 20 epochs, Adam (lr=1e-3), CrossEntropyLoss
- Training time: 7.6 seconds on RTX 4090
Training Data
- TBX11K: 8,812 images
- Shenzhen CXR: 662 images
- Montgomery County: 138 images
- Total: 9,342 chest X-rays (7,426 train / 957 val / 959 test)
3-Way Benchmark
| Method | AUC | Sensitivity | Specificity |
|---|---|---|---|
| Random baseline | 0.231 | 0.0% | 100% |
| MedSigLIP zero-shot | 0.837 | 77.2% | 75.7% |
| Fine-tuned (this model) | 0.9975 | 96.2% | 98.5% |
Usage
from tbscan.classifier import TBClassifier
from PIL import Image
classifier = TBClassifier(
model_id="google/medsiglip-448",
classifier_path="classifier.pt"
)
classifier.load()
image = Image.open("chest_xray.png").convert("RGB")
result = classifier.classify(image)
print(f"TB Probability: {result['prob_tb']:.1%}")
Part of TBScan
Built for the MedGemma Impact Challenge 2026. TBScan is an AI-powered TB screening system for low-resource clinics.
⚠️ Research and educational purposes only. Not validated for clinical use.
- Task
- Image Classification
- License
- apache-2.0
- Created
- Feb 24, 2026
- Updated
- Feb 24, 2026