Data Complexity Based Evaluation of the Model Dependence of Brain MRI Images for Classification of Brain Tumor and Alzheimer's Disease

Abstract

The convolutional neural networks (CNN) have shown promising results for various classification problems over the past years. However, selecting various CNN architectures is still challenging as each architecture performs differently with the same dataset. This research aims to evaluate the dependence of brain MRI on various predictive models of CNN based on the complexity of the data for Brain Tumor and Alzheimer's Disease. Our proposed approach has three parts. First part is the pre-processing of the data which mainly focuses on class balancing and the estimation of data complexity. The second part uses stratified k-fold cross-validation for more reliable results. The last part corresponds to the implementation of four CNN models applying described methods. This paper compares the classification performance of rigorous experimentation on four CNN variants namely S-CNN (CNN trained from scratch), ResNet50, InceptionV3, and Xception over two brain MRI image datasets evaluated with and without the use of Principal Component Analysis (PCA). The work benchmarks CNN models by comparing the average scores of Accuracy, Precision, Recall, F1 score, and AUC score from the stratified five-fold cross-validation.