Journal of Advanced Health Informatics Research
Vol. 2 No. 2 (2024)

Comparison of Transfer Learning Performance in Lung and Colon Classification with Knowledge Distillation

Elsa Wulandari, Annastasya Nabila (Unknown)
Yudhistira , Aimar (Unknown)
Purwono (Unknown)
Sharkawy , Abdel-Nasser (Unknown)



Article Info

Publish Date
31 Aug 2024

Abstract

This research aims to apply the knowledge distillation method to medical image classification, specifically in the case of lung and colon image classification using various transfer learning models. Knowledge distillation allows the transfer of knowledge from a larger model (teacher) to a smaller model (student), which enables more efficient model building without sacrificing accuracy. In this research, the DenseNet169 model is used as the teacher model. The student model uses several alternative transfer learning architectures such as DenseNet121, MobileNet, ResNet50, InceptionV3, and Xception. The data used consists of 25,000 histopathology images that have been processed and divided into training, validation, and test data. Data augmentation was performed to enlarge the dataset from 750 to 25,000 images, which helped improve the performance of the model. Model performance evaluation was performed by measuring the accuracy and loss value of each student model compared to the teacher model. The results showed that the student models generated through the knowledge distillation process performed close to or even exceeded the teacher model in some cases, with the Xception model showing the highest accuracy of 96.95%. In conclusion, knowledge distillation is effective in reducing model complexity without compromising performance, which is particularly beneficial for implementation on resource-constrained devices.

Copyrights © 2024






Journal Info

Abbrev

jahir

Publisher

Subject

Computer Science & IT Control & Systems Engineering Engineering Medicine & Pharmacology Public Health

Description

Journal of Advanced Health Informatics Research (JAHIR) is a scientific journal that focuses on the application of computer science to the health field. JAHIR is a peer-reviewed open-access journal that is published three times a year (April, August and December). The scientific journal is published ...