metadata

language: []
license: mit
tags:
  - chest-xray
  - efficientnet-b0
  - medical-ai
  - radiology
  - deep-learning
datasets:
  - nih-chest-xray
  - nlmcxr
model-index:
  - name: rayz_EfficientNet_B0
    results:
      - task:
          type: image-classification
        dataset:
          name: nih-chest-xray
          type: medical-image
        metrics:
          - name: AUROC Score
            type: accuracy
            value: 0.72 - 0.93

Rayz : AI-Powered Chest X-ray Analysis

🩺 Overview

This model analyzes chest X-rays to detect 14 potential lung conditions using EfficientNet_B0, a lightweight yet high-performing CNN. It was trained on NIH Chest X-ray Dataset & NLMCXR Dataset, providing reliable multi-class classification for various lung diseases.

🚀 Motivation

This project began when I received a false-positive tuberculosis (TB) report and had to wait for delayed X-ray results due to a holiday. Not knowing how to interpret X-rays, I built this AI tool to help others in similar situations.

📜 Model Details

Model type: Image Classification (Chest X-ray Analysis)
Architecture: EfficientNet_B0
Trained on: NIH Chest X-ray & NLMCXR Datasets
Input format: Chest X-ray images (.png, .jpg)
Output: Probabilities for 14 lung conditions
License: MIT
Compute Requirement: Can run on CPU, optimized for GPU (CUDA)

💡 Why EfficientNet_B0?

I tested multiple models, including DenseNet121, ViT, and CNNs, but EfficientNet_B0_best_93.44 outperformed the others in terms of:

High Accuracy (AUROC: 0.72 - 0.93)
Lower Computational Cost
Faster Inference Speed
Better Generalization across datasets

📊 Model Performance

Model	AUROC Score (Avg)
EfficientNet_B0	0.72 - 0.93
DenseNet121	0.55 - 0.95
ViT_Base	0.32 - 0.65

🔧 How to Use the Model

1️⃣ Install Dependencies

pip install torch torchvision transformers pillow numpy matplotlib seaborn