medical_chest_xray_classifier

Overview

This model is a ResNet-50 based image classifier fine-tuned on clinical chest radiography. It is designed to categorize X-ray images into three distinct classes: Normal, Bacterial Pneumonia, and Viral Pneumonia. The model assists in rapid triage and secondary diagnostic validation in clinical settings.

Model Architecture

The model utilizes a ResNet-50 architecture, leveraging deep residual learning to extract complex thoracic features.

  • Backbone: 50 layers of convolutional blocks with shortcut connections to mitigate vanishing gradients.
  • Input: Normalized 224x224 grayscale-converted images.
  • Optimization: Trained using cross-entropy loss with a focus on maximizing the macro-averaged $F_1$ score to handle class imbalances.

Intended Use

  • Radiology Triage: Prioritizing scans that show high probabilities of infection for immediate human review.
  • Medical Education: Aiding students in identifying visual markers of different pneumonia types.
  • Research: Serving as a baseline for more complex multi-modal diagnostic systems.

Limitations

  • Diagnostic Tool Only: This model is a supportive tool and must not be used for final medical diagnosis without professional oversight.
  • Hardware Variation: Performance may vary based on the manufacturer of the X-ray equipment and exposure settings.
  • Demographics: Validation was primarily conducted on adult datasets; pediatric accuracy may be significantly lower.
Downloads last month
2
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support