Kashif-Mahmood007/chestx6-ssl-checkpoints

ChestX6-SSL-Benchmark Checkpoints

This repository hosts the pretrained and fine-tuned model checkpoints accompanying the manuscript:

"A Deployment Risk Score Framework for Self-Supervised Chest X-Ray Classification: Calibrated Multi-Objective Evaluation Under Annotation Scarcity and Scanner Heterogeneity"

Authors: Kashif Mahmood, Romana Aziz, Muhammad Ramzan, Mahwish Ilyas, and Ala Saleh Alluhaidan

Overview

The checkpoints correspond to the experiments performed on the ChestX6 dataset and include:

• Four supervised baseline models
• SimCLR self-supervised pretrained encoder
• MAE self-supervised pretrained encoder
• Fine-tuned SimCLR models
• Fine-tuned MAE models

Models are provided for 10%, 20%, and 100% labeled-data settings with three random seeds for each configuration.

Repository Structure

supervised/
    ResNet50 (Scratch)
    ResNet50 (ImageNet)
    EfficientNet-B0
    MobileViT-XS

ssl/
    SimCLR pretrained encoder (.pth)
    MAE pretrained encoder (.pth)

finetuned/
    SimCLR fine-tuned models (.pt)
    MAE fine-tuned models (.pt)

Dataset

The experiments use the ChestX6: Multi-Class X-ray Dataset.

Original dataset:

ChestX6 Dataset on Kaggle

The original dataset contains 18,036 images. During preprocessing, 48 duplicate images were removed, resulting in 17,988 unique images used in all experiments.

Cross-dataset evaluation uses ChestMNIST through the MedMNIST package.

Loading

The checkpoints store PyTorch model weights.

Refer to the accompanying GitHub repository for the complete training pipeline, model definitions, configuration files, and loading examples.

GitHub Repository

The complete source code, training pipeline, notebooks, configuration files, dataset split definitions, and reproduction instructions are available on GitHub:

Repository: ChestX6-SSL-Benchmark

The GitHub repository includes:

• End-to-end training and evaluation notebooks
• Fixed train/validation/test split definitions
• Experiment results and manuscript figures
• Instructions for reproducing all experiments reported in the accompanying manuscript

Citation

If you use these checkpoints, please cite the associated manuscript.

@unpublished{mahmood2026drs,
  title  = {A Deployment Risk Score Framework for Self-Supervised Chest X-Ray Classification: Calibrated Multi-Objective Evaluation Under Annotation Scarcity and Scanner Heterogeneity},
  author = {Kashif Mahmood and Romana Aziz and Muhammad Ramzan and Mahwish Ilyas and Ala Saleh Alluhaidan},
  note   = {Manuscript under revision},
  year   = {2026}
}

License

MIT License.