Skin Lesion Classification in Low-Resource Settings Using Lightweight CNNs with Uncertainty Estimation

Dermoscopic skin lesion classification is a task of major clinical importance but is computationally expensive, making it inaccessible in resource-constrained healthcare settings. In this paper, we introduce a computationally efficient skin lesion classification framework for seven classes using EfficientNet-B0, complemented by Monte Carlo (MC) Dropout for uncertainty quantification. Our approach was trained and tested on the HAM10000 dataset containing 10,015 dermoscopic images across seven classes. To address the severe 67:1 class imbalance, we employ WeightedRandomSamplerand class-weighted cross-entropy loss as complementary corrections acting at the batch-composition level and the gradient-magnitude level respectively. By performing T=50 stochastic forward passes during inference, we decompose predictive uncertainty into aleatoric and epistemic components and apply an entropy-based referral threshold that flags uncertain predictions for specialist review. To validate spatial interpretability, Gradient-weighted Class Activation Mapping (Grad-CAM) is applied and quantitatively evaluated via Intersection over Union (IoU) against ISIC segmentation masks, yielding a mean IoU of 0.61 across all accepted predictions. Our experiments achieve a test macro AUROC of 0.9404and macro F1-score of 0.7308, with six of seven classes exceeding 70% per-class accuracy (melanocytic nevi: 69.8%). Referring the 30% most uncertain predictions to a clinician raises accepted-subset AUROC from 0.9404 to 0.9568 (+1.64%). The framework is competitive with ResNet-50 and DenseNet-121 at one-fifth the parameter count, and the only lightweight method in the comparison providing calibrated uncertainty estimates. Inference latency benchmarks on an NVIDIA Jetson Nano (edge CPU mode) are reported to contextualize deployment feasibility.

Authors

Institutions

Publication Details

Journal
BioMedInformatics
Published
2026-08-03
DOI
https://doi.org/10.3390/biomedinformatics6040053
Primary Topic
Cutaneous Melanoma Detection and Management
Type
article
Field-Weighted Citation Impact
0.00
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

Skin Lesion Classification in Low-Resource Settings Using Lightweight CNNs with Uncertainty Estimation

Princy Randhawa, Nithesh Naik, Malathy Batumalay, Mohammad Mahmudur Rahman‬ et al.
BioMedInformatics
Cutaneous Melanoma Detection and Management
article

Skin Lesion Classification in Low-Resource Settings Using Lightweight CNNs with Uncertainty Estimation

Princy Randhawa, Nithesh Naik, Malathy Batumalay, Mohammad Mahmudur Rahman‬, Surya Teja Meesala, Md Abu Kawsar Prodhan Hemal, Debabrata Biswas, Swathi Suddala
article en

Abstract

Dermoscopic skin lesion classification is a task of major clinical importance but is computationally expensive, making it inaccessible in resource-constrained healthcare settings. In this paper, we introduce a computationally efficient skin lesion classification framework for seven classes using EfficientNet-B0, complemented by Monte Carlo (MC) Dropout for uncertainty quantification. Our approach was trained and tested on the HAM10000 dataset containing 10,015 dermoscopic images across seven classes. To address the severe 67:1 class imbalance, we employ WeightedRandomSamplerand class-weighted cross-entropy loss as complementary corrections acting at the batch-composition level and the gradient-magnitude level respectively. By performing T=50 stochastic forward passes during inference, we decompose predictive uncertainty into aleatoric and epistemic components and apply an entropy-based referral threshold that flags uncertain predictions for specialist review. To validate spatial interpretability, Gradient-weighted Class Activation Mapping (Grad-CAM) is applied and quantitatively evaluated via Intersection over Union (IoU) against ISIC segmentation masks, yielding a mean IoU of 0.61 across all accepted predictions. Our experiments achieve a test macro AUROC of 0.9404and macro F1-score of 0.7308, with six of seven classes exceeding 70% per-class accuracy (melanocytic nevi: 69.8%). Referring the 30% most uncertain predictions to a clinician raises accepted-subset AUROC from 0.9404 to 0.9568 (+1.64%). The framework is competitive with ResNet-50 and DenseNet-121 at one-fifth the parameter count, and the only lightweight method in the comparison providing calibrated uncertainty estimates. Inference latency benchmarks on an NVIDIA Jetson Nano (edge CPU mode) are reported to contextualize deployment feasibility.

BioMedInformaticsVol. 6(4)
INTI International University (MY), Manipal Academy of Higher Education (IN), University of Wisconsin–Milwaukee (US), University of Houston - Clear Lake (US), Pacific States University (US), Dr. Hari Singh Gour University (IN)
Decent work and economic growth
Openalex Percentile: Top 12%
Cutaneous Melanoma Detection and Management
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.