Brain Tumor Classification Using Transfer Learning

Naim Rochmawati, Hanik Badriyah Hidayati, Yuni Yamasari, Wiyli Yustanti, I. Made Suartana, Agus Prihanto, Aditya Prapanca

Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

4 Citations (Scopus)

Abstract

One type of deadly disease is a brain tumor. To determine the presence of a brain tumor, it can be seen from an MRI image. In this research, we classified brain tumor MRI. The classification system uses transfer learning because only a few datasets are used. The Pre-Trained models used to extract features are VGG-16 and ResNet-50. Tests are carried out using several different parameters such as different batch sizes, optimizers, and learning rates. We evaluate the results using the confusion matrix. VGG-16 got the best accuracy of 0.96 using the Adam optimizer and ResNet-50 got the best accuracy of 0.94 using the RMSprop optimizer. From several different parameter variations, there is a relationship between parameter selection and accuracy results.

Original languageEnglish
Title of host publication2022 5th International Conference on Vocational Education and Electrical Engineering
Subtitle of host publicationThe Future of Electrical Engineering, Informatics, and Educational Technology Through the Freedom of Study in the Post-Pandemic Era, ICVEE 2022 - Proceeding
PublisherInstitute of Electrical and Electronics Engineers Inc.
Pages95-99
Number of pages5
ISBN (Electronic)9781665475815
DOIs
Publication statusPublished - 2022
Event5th International Conference on Vocational Education and Electrical Engineering, ICVEE 2022 - Virtual, Surabaya, Indonesia
Duration: 10 Sept 202211 Sept 2022

Publication series

Name2022 5th International Conference on Vocational Education and Electrical Engineering: The Future of Electrical Engineering, Informatics, and Educational Technology Through the Freedom of Study in the Post-Pandemic Era, ICVEE 2022 - Proceeding

Conference

Conference5th International Conference on Vocational Education and Electrical Engineering, ICVEE 2022
Country/TerritoryIndonesia
CityVirtual, Surabaya
Period10/09/2211/09/22

Keywords

  • Adam
  • RMSProp
  • ResNet-50
  • SGD
  • Vgg-16
  • brain tumor
  • optimizer
  • pre-trained model
  • transfer learning

Fingerprint

Dive into the research topics of 'Brain Tumor Classification Using Transfer Learning'. Together they form a unique fingerprint.

Cite this