An Optimized Balanced-Learning Framework for Malignant Skin Lesion Triage Using Compound-Scaled Neural Networks

Argha Orion Silitonga(1*), Raissa Camilla Maringka(2), Wilsen Grivin Mokodaser(3), George M W Tangka(4), Marchel Timothy Tombeng(5),

(1) Universitas Klabat, Minahasa Utara
(2) Universitas Klabat, Minahasa Utara
(3) Universitas Klabat, Minahasa Utara
(4) Universitas Klabat, Minahasa Utara
(5) Universitas Klabat, Minahasa Utara
(*) Corresponding Author

Abstract


Skin cancer represents a prevalent global health challenge, and early detection is very important to reduce mortality risk. Manual dermoscopic diagnosis risks human bias, making deep learning classification a vital research topic. While several previous studies utilizing the ISIC 2019 dataset have demonstrated high diagnostic capabilities, they primarily focus on complex multi-class classification. However, in real-world clinical workflows, the primary necessity is a swift, dependable triage system that can confidently distinguish dangerous lesions from non-threatening ones. Furthermore, many existing models require substantial computational overhead yet still suffer from imbalanced accuracy when dealing with minority malignant classes. The novelty of this study lies in addressing these gaps by developing a streamlined, clinically practical binary screening framework optimized specifically for malignant-versus-benign triage. The original multi-class labels were transformed into binary classes where malignant lesions consist of melanoma (MEL), basal cell carcinoma (BCC), and squamous cell carcinoma (SCC), while benign lesions consist of nevus (NV), benign keratosis (BKL), dermatofibroma (DF), and vascular lesions (VASC). The experiment applied transfer learning with ImageNet-pretrained weights, data augmentation, class weighting, and fourfold stratified cross-validation. Unlike prior works that rely on resource-heavy architectures, we leverage the compound-scaled EfficientNet-B4 backbone—delivering superior feature representational power with significantly fewer parameters evaluate on a large-scale cohort of 25,331 dermoscopic images. Experimental results show that the proposed model achieved an average accuracy of 89.77% and an average ROC AUC of 96.16%. The best fold obtained 91.49% accuracy with ROC AUC of 97.19%. Simultaneously, the framework maintained an average F1-score of 89.20%

Keywords


Skin Cancer; Dermoscopy; Deep Learning; EfficientNet-B4; Cross-Validation

Full Text:

PDF

References


M. Hameed, A. Zameer, and M. A. Z. Raja, “A Comprehensive Systematic Review: Advancements in Skin Cancer Classification and Segmentation Using the ISIC Dataset,” CMES, vol. 140, no. 3, pp. 2131–2164, 2024, doi: 10.32604/cmes.2024.050124.

N. Nigar, M. Umar, M. K. Shahzad, S. Islam, and D. Abalo, “A Deep Learning Approach Based on Explainable Artificial Intelligence for Skin Lesion Classification,” IEEE Access, vol. 10, pp. 113715–113725, 2022, doi: 10.1109/ACCESS.2022.3217217.

I. Goodfellow, Y. Bengio, and A. Courville, Deep Learning. MIT Press, 2016. [Online]. Available: http://www.deeplearningbook.org

A. K. Sharma et al., “Dermatologist-Level Classification of Skin Cancer Using Cascaded Ensembling of Convolutional Neural Network and Handcrafted Features Based Deep Neural Network,” IEEE Access, vol. 10, pp. 17920–17932, 2022, doi: 10.1109/ACCESS.2022.3149824.

P. Kumar, R. Chauhan, A. Shankar, and T. Stephan, “Role of Artificial Intelligence for Skin Cancer Detection,” in Evolving Role of AI and IoMT in the Healthcare Market, F. Al-Turjman, M. Kumar, T. Stephan, and A. Bhardwaj, Eds., Cham: Springer International Publishing, 2021, pp. 141–174. doi: 10.1007/978-3-030-82079-4_7.

Z. Rahman, Md. S. Hossain, Md. R. Islam, Md. M. Hasan, and R. A. Hridhee, “An approach for multiclass skin lesion classification based on ensemble learning,” Informatics in Medicine Unlocked, vol. 25, p. 100659, 2021, doi: 10.1016/j.imu.2021.100659.

K. M. Hosny and M. A. Kassem, “Refined Residual Deep Convolutional Network for Skin Lesion Classification,” J Digit Imaging, vol. 35, no. 2, pp. 258–280, Apr. 2022, doi: 10.1007/s10278-021-00552-0.

C. Zhao, R. Shuai, L. Ma, W. Liu, D. Hu, and M. Wu, “Dermoscopy Image Classification Based on StyleGAN and DenseNet201,” IEEE Access, vol. 9, pp. 8659–8679, 2021, doi: 10.1109/ACCESS.2021.3049600.

J. Liu, “VGG, MobileNet and AlexNet on Recognizing Skin Cancer Symptoms,” in 2022 3rd International Conference on Electronic Communication and Artificial Intelligence (IWECAI), Zhuhai, China: IEEE, Jan. 2022, pp. 525–528. doi: 10.1109/IWECAI55315.2022.00107.

S. K. Singh, S. Banerjee, A. Chakraborty, and A. Bandyopadhyay, “Classification of Melanoma Skin Cancer Using Inception-ResNet,” in Frontiers of ICT in Healthcare, vol. 519, J. K. Mandal and D. De, Eds., in Lecture Notes in Networks and Systems, vol. 519. , Singapore: Springer Nature Singapore, 2023, pp. 65–74. doi: 10.1007/978-981-19-5191-6_6.

B. Koonce, “EfficientNet,” in Convolutional Neural Networks with Swift for Tensorflow, Berkeley, CA: Apress, 2021, pp. 109–123. doi: 10.1007/978-1-4842-6168-2_10.

P. Kumar, D. Kumar, A. Kumar, and P. S. Rathore, “AttnEffNet-B4: an attention-augmented EfficientNet-B4 framework with fourier transformation for robust multi-disease diagnosis,” Sci Rep, Apr. 2026, doi: 10.1038/s41598-026-49257-w.

P. Tschandl, C. Rosendahl, and H. Kittler, “The HAM10000 dataset, a large collection of multi-source dermatoscopic images of common pigmented skin lesions,” Sci Data, vol. 5, no. 1, p. 180161, Aug. 2018, doi: 10.1038/sdata.2018.161.

S. Rohini and P. Vidhyasaraswathi, “A Deep Learning Approach for High-Accuracy Brain Tumor Classification: Evaluating ResNet-50, U-Net, and EfficientNet-B4,” in 2025 3rd International Conference on Self Sustainable Artificial Intelligence Systems (ICSSAS), Erode, India: IEEE, Jun. 2025, pp. 1366–1373. doi: 10.1109/ICSSAS66150.2025.11080727.

A. Mao, M. Mohri, and Y. Zhong, “Cross-Entropy Loss Functions: Theoretical Analysis and Applications,” in Proceedings of the 40th International Conference on Machine Learning, A. Krause, E. Brunskill, K. Cho, B. Engelhardt, S. Sabato, and J. Scarlett, Eds., in Proceedings of Machine Learning Research, vol. 202. PMLR, Jul. 2023, pp. 23803–23828. [Online]. Available: https://proceedings.mlr.press/v202/mao23b.html

S. Kornblith, T. Chen, H. Lee, and M. Norouzi, “Why Do Better Loss Functions Lead to Less Transferable Features?,” in Advances in Neural Information Processing Systems, M. Ranzato, A. Beygelzimer, Y. Dauphin, P. S. Liang, and J. W. Vaughan, Eds., Curran Associates, Inc., 2021, pp. 28648–28662. [Online]. Available: https://proceedings.neurips.cc/paper_files/paper/2021/file/f0bf4a2da952528910047c31b6c2e951-Paper.pdf

S. Dereich and A. Jentzen, “Convergence rates for the Adam optimizer,” 2024, arXiv. doi: 10.48550/ARXIV.2407.21078.

Q. Yu, J. He, X. Deng, X. Shen, and L.-C. Chen, “Convolutions Die Hard: Open-Vocabulary Segmentation with Single Frozen Convolutional CLIP,” in Advances in Neural Information Processing Systems, A. Oh, T. Naumann, A. Globerson, K. Saenko, M. Hardt, and S. Levine, Eds., Curran Associates, Inc., 2023, pp. 32215–32234. [Online]. Available: https://proceedings.neurips.cc/paper_files/paper/2023/file/661caac7729aa7d8c6b8ac0d39ccbc6a-Paper-Conference.pdf

S. Szeghalmy and A. Fazekas, “A Comparative Study of the Use of Stratified Cross-Validation and Distribution-Balanced Stratified Cross-Validation in Imbalanced Learning,” Sensors, vol. 23, no. 4, p. 2333, Feb. 2023, doi: 10.3390/s23042333.

S. Prusty, S. Patnaik, and S. K. Dash, “SKCV: Stratified K-fold cross-validation on ML classifiers for predicting cervical cancer,” Front. Nanotechnol., vol. 4, p. 972421, Aug. 2022, doi: 10.3389/fnano.2022.972421.

M. Hassanzad and K. Hajian-Tilaki, “Methods of determining optimal cut-point of diagnostic biomarkers with application of clinical data in ROC analysis: an update review,” BMC Med Res Methodol, vol. 24, no. 1, p. 84, Apr. 2024, doi: 10.1186/s12874-024-02198-2.

J. Allgaier and R. Pryss, “Cross-Validation Visualized: A Narrative Guide to Advanced Methods,” MAKE, vol. 6, no. 2, pp. 1378–1388, Jun. 2024, doi: 10.3390/make6020065.

L. Lausser, R. Szekely, F. Schmid, M. Maucher, and H. A. Kestler, “Efficient cross-validation traversals in feature subset selection,” Sci Rep, vol. 12, no. 1, p. 21485, Dec. 2022, doi: 10.1038/s41598-022-25942-4.




DOI: http://dx.doi.org/10.61944/bids.v5i1.165

Refbacks

  • There are currently no refbacks.


Copyright (c) 2026 Argha Orion Silitonga, Raissa Camilla Maringka, Wilsen Grivin Mokodaser, George M. W. Tangka, Marchel Timothy Tombeng

Creative Commons License
This work is licensed under a Creative Commons Attribution 4.0 International License.

Bulletin of Informatics and Data Science
Asosiasi Peneliti Data Science Indonesia
Email: pdsi.bids@gmail.com
This work is licensed under a Creative Commons Attribution 4.0 International License.