Journal Article

A novel Parametric Flatten-p Mish activation function based deep CNN model for brain tumor classification

Mondal, Ayan, Shrivastava, Vimal K.

2022

Overview

The brain tumor is one of the deadliest diseases of all cancers. Influenced by the recent developments of convolutional neural networks (CNNs) in medical imaging, we have formed a CNN based model called BMRI-Net for brain tumor classification. As the activation function is one of the important modules of CNN, we have proposed a novel parametric activation function named Parametric Flatten-p Mish (PFpM) to improve the performance. PFpM can tackle the significant disadvantages of the pre-existing activation functions like neuron death and bias shift effect. The parametric approach of PFpM also offers the model some extra flexibility to learn the complex patterns more accurately from the data. To validate our proposed methodology, we have used two brain tumor datasets namely Figshare and Br35H. We have compared the performance of our model with state-of-the-art deep CNN models like DenseNet201, InceptionV3, MobileNetV2, ResNet50 and VGG19. Further, the comparative performance of PFpM has been presented with various activation functions like ReLU, Leaky ReLU, GELU, Swish and Mish. We have performed record-wise and subject-wise (patient-level) experiments for Figshare dataset whereas only record-wise experiments have been performed in case of Br35H dataset due to unavailability of subject-wise information. Further, the model has been validated using hold-out and 5-fold cross-validation techniques. On Figshare dataset, our model has achieved 99.57% overall accuracy with hold-out validation and 98.45% overall accuracy with 5-fold cross validation in case of record-wise data split. On the other hand, the model has achieved 97.91% overall accuracy with hold-out validation and 97.26% overall accuracy with 5-fold cross validation in case of subject-wise data split. Similarly, for Br35H dataset, our model has attained 99% overall accuracy with hold-out validation and 98.33% overall accuracy with 5-fold cross validation using record-wise data split. Hence, our findings can introduce a secondary procedure in the clinical diagnosis of brain tumors. •A unique CNN architecture for brain tumor classification.•A novel parametric piece-wise activation function named PFpM.•Achieved 99.57% and 99% overall classification accuracy on Figshare and Br35H dataset, respectively.•Comparative performance analysis of proposed model with state-of-the-art models.•Comparative performance analysis of models with different activation functions.

• Publisher: Elsevier Ltd,Elsevier Limited
• Subject: Accuracy / Activation function / Algorithms / Artificial intelligence / Artificial neural networks / BMRI-Net / Brain / Brain cancer / Brain Neoplasms - diagnostic imaging / Brain tumor classification / Brain tumors / Cell death / Classification / CNN / Datasets / Diabetic retinopathy / Humans / Image classification / Image Interpretation, Computer-Assisted - methods / Internal Medicine / Magnetic Resonance Imaging - methods / Medical imaging / Metastasis / MRI / Neural networks / Neural Networks, Computer / Neuroimaging / Other / Performance enhancement / PFpM / Support vector machines / Tumors