<scp>AlexNet‐NDTL</scp>: Classification of <scp>MRI</scp> brain tumor images using modified <scp>AlexNet</scp> with deep transfer learning and Lipschitz‐based data augmentation
Sreedhar Kollem, Katta Ramalinga Reddy, Ch. Rajendra Prasad, Avishek Chakraborty, J. Ajayan, S. Sreejith, Sandip Bhattacharya, L. M. I. Leo Joseph, Ravichander Janapati
Abstract
Abstract Deep learning is frequently used to classify medical images. Surgeons may know the type of tumor before doing surgery on a patient. Transfer learning was used to alleviate the overfitting issue of deep networks in classification since the training samples, such as a brain MRI dataset, were insufficient. To overcome this issue, We introduce a new deep‐learning methodology for the categorization of MRI brain tumor images. This method combines a unique data augmentation model with modified AlexNet and network‐based deep transfer learning. We used Lipschitz‐based data augmentation on a dataset, and the output of the augmentation model was fed into a modified AlexNet that uses network‐based deep transfer learning to extract features from a dataset. The proposed model is trained and tested using the BraTS 2020 and Figshare datasets. The proposed model's performance is assessed using sensitivity, specificity, accuracy, precision, F1‐score, and MCC. The proposed model yields superior results.