# ModelAuditor Pre-trained Models

Pre-trained models for medical image classification, used with the [ModelAuditor](https://github.com/MLO-lab/ModelAuditor) framework for AI-powered model auditing and robustness evaluation.

## Models

| Model | Architecture | Domain | Task | Input Size |
|-------|-------------|--------|------|------------|
| `camelyon17_resnet50_1_224.pt` | ResNet50 | Pathology | Tumor detection in lymph node sections | 224x224 |
| `chexpert_resnet50_1_224.pt` | ResNet50 | Radiology | Chest X-ray classification | 224x224 |
| `ham10000_resnet50_1_224.pt` | ResNet50 | Dermatology | Skin lesion classification (melanoma vs. benign keratosis) | 224x224 |
| `cifar10.pth` | ResNet50 | General | CIFAR-10 image classification | 224x224 |
| `DermaMNIST_resnet18.pth` | ResNet18 | Dermatology | Skin lesion classification (7 classes) | 224x224 |

## Usage

### Download Models

```bash
pip install huggingface_hub

# Download all models
huggingface-cli download lukaskuhndkfz/ModelAuditor --local-dir models

# Or download individually
huggingface-cli download lukaskuhndkfz/ModelAuditor ham10000_resnet50_1_224.pt --local-dir models
```

### Use with ModelAuditor

```bash
git clone https://github.com/lukaskuhndkfz/ModelAuditor
cd ModelAuditor
pip install -e ".[medical]"

# Run auditing
python main.py --model resnet50 --dataset ham10000 --weights models/ham10000_resnet50_1_224.pt
```

### Load in PyTorch

```python
import torch
from torchvision.models import resnet50

model = resnet50(num_classes=2)
model.load_state_dict(torch.load("ham10000_resnet50_1_224.pt", map_location="cpu"))
model.eval()
```

For DermaMNIST (ResNet18):

```python
import torch
from torchvision.models import resnet18

model = resnet18(num_classes=7)
model.load_state_dict(torch.load("DermaMNIST_resnet18.pth", map_location="cpu"))
model.eval()
```

## Training

Training scripts for all ResNet50 models are available in this repository as well (click on Files and Versions in the menu above). 

## Datasets

- Camelyon17: https://wilds.stanford.edu/datasets/#camelyon17
- CheXpert: https://stanfordmlgroup.github.io/competitions/chexpert/
- HAM10000: https://dataverse.harvard.edu/dataset.xhtml?persistentId=doi:10.7910/DVN/DBW86T
- CIFAR-10: https://www.cs.toronto.edu/~kriz/cifar.html
- DermaMNIST: https://medmnist.com/