Detection and classification on MRI images of brain tumor using YOLO NAS deep learning model

Abstract

If a brain tumor is not properly diagnosed, it might result in fatal consequences and major health issues. As a result, a key component of diagnosis is the early identification of brain tumors and the precise categorization of brain tumor types. This research work focusses on detection and classification such as pituitary, meningioma, glioma, and non-tumorous tumors from brain tumor MRI image using YOLO NAS model. Advances in deep learning have led to an increasing awareness of computer-aided diagnosis technology. To improve classification performance, the input data set is acquired from the REMBRANDT repository using the Digital Database of Brain Tumor Magnetic Resonance Images. The primary phases of this task include segmentation, classification, and pre-processing. The researcher preprocessed the RGB image to exclude any undesired spots before locating the ROI. For each brain tumor image, we used the hybrid anisotropic diffusion filtering (HADF) technique to remove noise. The Encoder-Decoder Network (En–DeNet), which uses a deep neural network based on U-Net as the encoder and a pre-trained EfficientNet as the decoder, is then used to segment the MRI images. A proposed model was developed using a deep learning-based YOLO NAS technique to identify brain cancers from MRI images such as meningioma tumors, non-tumor, glioma tumors, and pituitary tumors. The suggested model's classification performance is weighed based on parameters such as precision, F1-score, sensitivity, accuracy, and specificity. The proposed fusion method YOLO NAS has improved classification results of accuracy (AC = 0.997%), specificity (SP = 0.985%), precision (PR = 0.982%), F1-score (F1 = 0.992%), and sensitivity (SE = 0.985%) using 2570 MRI data for training and 630 data for testing and validation of brain tumor cases. The results show that the brain tumor type classification system based on the proposed YOLO NAS technique works better than the DNN, PDCNN, DenseNet-161, and DCNN-SGD model classifiers.