baotramduong / brain-tumor-classification-with-efficient-net-and-grad-cam-visualization

Brain Tumor Classification with Efficient Net Convolutional Neural Network (CNNs)

Brain Tumor Classification with Efficient Net Convolutional Neural Network (CNNs) and Grad-CAM Visualization

Blog

Medium Blog

Introduction

In this project we will build and train an Efficient Net model and apply it to the Brain Tumor MRI Dataset to classify tumors: glioma_tumor, meningioma_tumor, pituitary_tumor, and no_tumor.

In addition, we will also use a technique known as Gradient-Weighted Class Activation Mapping (Grad-CAM) to visualize the regions of the inputs and help us explain how our CNN models think and make decision.

Data Source

The data set which we are going to use has 3,285 images of brain MRI scans Which are categorized in four different classes namely glioma_tumor, meningioma_tumor, pituitary_tumor, and no_tumor. 

The dataset can be accessed on Kaggle Brain Tumor MRI Dataset or you can clone the dataset from this github repository.

Exploratory Data Analysis

Modeling

Model Evaluation

Prediction

Grad-CAM

Glioma Tumor

Meningioma Tumor

No Tumor

Pituitary Tumor

Reference

Agarwal, V. (2020, May 23). Complete architectural details of all efficientnet models. Medium. Retrieved September 19, 2021, from https://towardsdatascience.com/complete-architectural-details-of-all-efficientnet-models-5fd5b736142.

Arya, A. (n.d.). Brain tumor classification using Keras [MOOC]. Coursera. https://www.coursera.org/projects/brain-tumor-classification-using-keras-jbek2?courseSlug=brain-tumor-classification-using-keras-jbek2&showOnboardingModal=check.

Kermany, D. S., Goldbaum, M., Cai, W., Valentim, C., Liang, H., Baxter, S. L., McKeown, A., Yang, G., Wu, X., Yan, F., Dong, J., Prasadha, M. K., Pei, J., Ting, M., Zhu, J., Li, C., Hewett, S., Dong, J., Ziyar, I., Shi, A., … Zhang, K. (2018). Identifying Medical Diagnoses and Treatable Diseases by Image-Based Deep Learning. Cell, 172(5), 1122–1131.e9. https://doi.org/10.1016/j.cell.2018.02.010.

Quick brain tumor facts. National Brain Tumor Society. (2021, March 22). Retrieved September 20, 2021, from https://braintumor.org/brain-tumor-information/brain-tumor-facts/.

Siddhartha. (2019, June 5). CAM visualization OF EFFIECIENT. Machine Leaning Blog. Retrieved September 20, 2021, from https://sidml.github.io/efficientnet-gradcam-comparison-to-other-models/.

Tan, M.; Le, Q.. (2019). EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks. Proceedings of the 36th International Conference on Machine Learning, in Proceedings of Machine Learning Research 97:6105–6114 Available from https://proceedings.mlr.press/v97/tan19a.html.

Rosebrock, A. (2020, March 9). Grad-CAM: Visualize Class Activation Maps with Keras, TensorFlow, and Deep Learning. PyImageSearch. Retrieved September 10, 2021, from https://www.pyimagesearch.com/2020/03/09/grad-cam-visualize-class-activation-maps-with-keras-tensorflow-and-deep-learning/.