Downloads · 30 days
3
8% of all-time downloads
xyxing/dermai-vit-base-patch16-224
dermai-vit-base-patch16-224 is a machine learning model from xyxing. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
Fine-tuned google/vit-base-patch16-224 for 7-class skin lesion classification on HAM10000. Part of the DermAI explainability project comparing CNN and Vision Transformer explanations.
Downloads · 30 days
3
8% of all-time downloads
All-time downloads
36
Public
Parameters
85.8M
343 MB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors343 MB · 100%
From the Hugging Face model README
Fine-tuned google/vit-base-patch16-224 for 7-class skin lesion classification on HAM10000. Part of the DermAI explainability project comparing CNN and Vision Transformer explanations.
Hyperparameter tuned: phase1 lr = 1e-3, phase2 lr = 4e-5, weight decay = 0.1, batch size = 8, warmup ratio = 0.1
Best validation Macro F1 = 0.8049