Transfer Learning Using Convolutional Neural Network Architectures for Brain Tumor Classification from MRI Images - Artificial Intelligence Applications and Innovations
Conference Papers Year : 2020

Transfer Learning Using Convolutional Neural Network Architectures for Brain Tumor Classification from MRI Images

Abstract

Brain tumor classification is very important in medical applications to develop an effective treatment. In this paper, we use brain contrast-enhanced magnetic resonance images (CE-MRI) benchmark dataset to classify three types of brain tumor (glioma, meningioma and pituitary). Due to the small number of training dataset, our classification systems evaluate deep transfer learning for feature extraction using nine deep pre-trained convolutional Neural Networks (CNNs) architectures. The objective of this study is to increase the classification accuracy, speed the training time and avoid the overfitting. In this work, we trained our architectures involved minimal pre- processing for three different epoch number in order to study its impact on classification performance and consuming time. In addition, the paper benefits acceptable results with small number of epoch in limited time. Our interpretations confirm that transfer learning provides reliable results in the case of small dataset. The proposed system outperforms the state-of-the-art methods and achieve 98.71% classification accuracy

Dates and versions

hal-03125885 , version 1 (29-01-2021)

Identifiers

Cite

Rayene Chelghoum, Ameur Ikhlef, Amina Hameurlaine, Sabir Jacquir. Transfer Learning Using Convolutional Neural Network Architectures for Brain Tumor Classification from MRI Images. 16th IFIP International Conference on Artificial Intelligence Applications and Innovations (AIAI), Jun 2020, Neos Marmaras, Greece. pp.189-200, ⟨10.1007/978-3-030-49161-1_17⟩. ⟨hal-03125885⟩
88 View
0 Download

Altmetric

Share

More