Ensemble Technique for Brain Tumor Patient Survival Prediction

• Published in: IEEE access Vol. 12; pp. 19285 - 19298
• Main Authors: Vinod, D. S., Prakash, S. P. Shiva, AlSalman, Hussain, Muaad, Abdullah Y., Heyat, Md. Belal Bin
• Format: Journal Article
• Language: English
• Published: Piscataway IEEE 2024; The Institute of Electrical and Electronics Engineers, Inc. (IEEE)
• Subjects: Accuracy; Artificial neural networks; Automated brain tumor detection; Automation; Brain; Brain cancer; Brain modeling; BRATS 2020 dataset; Complexity theory; convolutional neural network (CNN); Convolutional neural networks; Datasets; Feature extraction; Feature maps; Image segmentation; Machine learning; Magnetic resonance imaging; modified self-organizing feature map (MSOFM); Nutrients; Predictive models; Segmentation; Self organizing maps; Self-organizing feature maps; Survival; survival information; survival prediction; tumor identification; Tumors; U-Net
• ISSN: 2169-3536; 2169-3536
• DOI: 10.1109/ACCESS.2024.3360086

Brain Tumours pose a significant health challenge, demanding the immediate development of reliable and automated detection methods within the medical sphere. Swift and accurate identification of these Tumours are crucial for effective treatment and the well-being of patients. These growths stem from uncontrolled cell multiplication, depleting vital nutrients from healthy brain tissue and leading to organ dysfunction. Presently, the conventional method involves a manual examination of brain MRI scans by medical professionals, but this is hindered by the varied shapes and sizes of Tumours, resulting in time-consuming and occasionally imprecise evaluations. The emergence of automation holds immense potential, promising to bolster efficiency and allow medical practitioners more time for direct patient care. Traditional machine learning approaches have historically depended on labor-intensive feature engineering. In our research, we introduce an innovative approach: a combination of the U-Net model, a Convolutional Neural Network (CNN), and Self Organizing Feature Map (SOFM) in an ensemble technique for precise brain Tumour segmentation using the BRATS 2020 dataset. Our evaluation not only focuses on segmentation accuracy but also utilizes valuable survival data from the dataset to predict patient survival rates. The proposed model resulted in average training accuracy, mean Intersection over Union (mIoU), and dice coefficient scores of 0.967, 0.521, and 0.990 respectively for different epochs. Also, average validation accuracy, mIoU, and dice coefficient scores of 0.965, 0.546, and 0.992 respectively. The proposed model showcases a 98.28% accuracy in the segmentation of brain Tumours. The proposed methodology has the potential to revolutionize the landscape of brain Tumour diagnosis and treatment.