High-performance visual geometric group deep learning architectures for MRI brain tumor classification

N. Veni,J. Manjula
DOI: https://doi.org/10.1007/s11227-022-04384-9
IF: 3.3
2022-03-11
The Journal of Supercomputing
Abstract:Magnetic resonance imaging (MRI) of the brain is one of the most common imaging technologies used for brain cancer detection. Manual classification leads to more biopsies to ensure that there are no missed diagnoses. Recently, convolutional neural networks have achieved high accuracy in many image classification challenges. This study analyzes four different architectures from the Visual Geometric Group (VGG) for brain image classification using transfer learning. First, the feature space is generated using several 3 × 3 convolution filters and then reduced by the pooling layers in a block. These operations are repeated with different numbers of convolution filters in the subsequent blocks. After a predefined number of blocks, a fully connected layer is employed with an activation function to classify the given input. Four VGG architectures with different numbers of layers, 11, 13, 16 and 19, are developed to classify MRI images. The results prove that transfer learning on VGG architectures has good potential for brain cancer classification for REpository of Molecular BRAin Neoplasia DaTa (REMBRANDT) database MRI images. The results show that VGG-16 has the best performance, with an accuracy of 96% for brain cancer classification, followed by the VGG-19 architecture with 94.5% accuracy.
What problem does this paper attempt to address?