xyxing's picture
Create README.md
95629ad verified
|
Raw
History Blame Contribute Delete
329 Bytes

Fine-tuned google/efficientnet-b0 for 7-class skin lesion classification on HAM10000. Part of the DermAI explainability project comparing CNN and Vision Transformer explanations.

Hyperparameter tuned: phase1 lr = 4e-4, phase2 lr = 1e-4, weight decay = 0.01, batch size = 32, warmup ratio = 0.1

Best validation Macro F1 = 0.7073