chest-xray-classifier

chest-xray-classifier · ConvNeXt-V2-Tiny pneumonia classifier

Production-grade 3-class chest X-ray classifier distinguishing normal, bacterial pneumonia, and viral pneumonia on pediatric frontal radiographs.

Test-set confusion matrix · ConvNeXt-V2-Tiny · 91.3% accuracy (n=624)

Overview


Task	Multiclass image classification (3 classes)
Dataset	Kaggle Chest X-Ray Images (Pneumonia) — 5,856 radiographs
Main model	ConvNeXt-V2-Tiny (`facebook/convnextv2-tiny-22k-224`) fine-tuned
Baseline	DINOv2 ViT-S linear probe (`facebook/dinov2-small`)
Stack	PyTorch Lightning · Hydra · MLflow · DVC · FastAPI · Docker · GitHub Actions · MkDocs
License	MIT

Sections

Architecture — data flow, model choices, metrics rationale
Training — running experiments, logging, overrides
Serving — FastAPI endpoints, Docker deployment
Benchmarks — vs literature, trade-offs
Reproducibility — pinned environment, one-command re-run
Limitations — failure modes, dataset bias
Model card — HF Hub card template

Links

Code: GitHub
Model: Hugging Face

Disclaimer

Research/educational artifact only — not intended for clinical use.