skin-lesion-classifiers

Skin Lesion Classification: CNN and KAN-ViT Hybrids

Trained checkpoints for six image classifiers on a 14-class skin condition dataset. Two are custom CNNs trained from scratch. Four are KAN-ViT hybrids that pair a pretrained backbone with Kolmogorov-Arnold Network (KAN) spline layers.

Code, training notebooks, and full documentation: https://github.com/deettoh/skin-lesion-classification

Checkpoints

Val acc is best validation accuracy from the training logs. Test acc is held-out test accuracy, with test-time augmentation for the CNN.

Usage

from huggingface_hub import hf_hub_download

path = hf_hub_download(
    "diddoe/skin-lesion-classifiers",
    "custom_cnn_v1/v1_custom_cnn_skin_lesion_100_epochs.pth",
)

Rebuild the matching architecture from the GitHub repo, then load with model.load_state_dict(torch.load(path)). All models output 14 classes.

Dataset

Trained on the Kaggle dataset ahmedxc4/skin-ds, a 14-class set of dermoscopy and clinical skin images. The distribution is long-tailed, so every run uses a weighted sampler and Custom CNN V2 uses focal loss.

License

MIT for the model weights. The training images carry their own upstream terms, including ISIC archive sources. Check those before redistributing data.