Aiti: Jurnal Teknologi Informasi
Vol 23 No 3 (2026)

Optimizing CNN for burn severity classification using hyperband

Vallerie Britania Sepang (Universitas Sam Ratulangi)
Winsy Christo Deilan Weku (Universitas Sam Ratulangi)
Aditya Lapu Kalua (Universitas Sam Ratulangi)



Article Info

Publish Date
01 Oct 2026

Abstract

Burn injuries require accurate severity diagnosis to determine appropriate treatment and reduce mortality. However, conventional clinical assessment is susceptible to subjectivity and diagnostic errors. This study aims to optimize a Convolutional Neural Network (CNN) model for classifying burn severity using the Hyperband hyperparameter optimization algorithm. A dataset of 1,040 burn images, collected from primary and secondary sources, was categorized as first-, second-, or third-degree burns. A MobileNet-based CNN model was trained using both default and Hyperband-optimized hyperparameters and evaluated using 10-fold cross-validation. The results demonstrate that hyperparameter optimization substantially improved classification performance. The optimized model achieved an accuracy of 93.57%, compared with 82.14% for the baseline model. Improvements were also observed in precision, recall, and F1-score, indicating better classification performance and reduced prediction errors. These findings demonstrate that Hyperband can effectively optimize CNN hyperparameters for burn severity classification. The proposed approach has the potential to support more consistent and accurate burn assessment. Future research should investigate advanced optimization strategies and larger, more diverse datasets to improve model generalization and robustness further.

Copyrights © 2026






Journal Info

Abbrev

aiti

Publisher

Subject

Computer Science & IT

Description

AITI: Jurnal Teknologi Informasi is a peer-review journal focusing on information system and technology issues. AITI invites academics and researchers who do original research in information system and technology, including but not limited to: Cryptography Networking Internet of Things Big Data Data ...