Enhancing Prediction of Brain Tumor Classification Using Images and Numerical Data Features

Diagnostics (Basel). 2023 Jul 31;13(15):2544. doi: 10.3390/diagnostics13152544.

Abstract

Brain tumors, along with other diseases that harm the neurological system, are a significant contributor to global mortality. Early diagnosis plays a crucial role in effectively treating brain tumors. To distinguish individuals with tumors from those without, this study employs a combination of images and data-based features. In the initial phase, the image dataset is enhanced, followed by the application of a UNet transfer-learning-based model to accurately classify patients as either having tumors or being normal. In the second phase, this research utilizes 13 features in conjunction with a voting classifier. The voting classifier incorporates features extracted from deep convolutional layers and combines stochastic gradient descent with logistic regression to achieve better classification results. The reported accuracy score of 0.99 achieved by both proposed models shows its superior performance. Also, comparing results with other supervised learning algorithms and state-of-the-art models validates its performance.

Keywords: MRI images; MobileNet; UNet; brain tumor prediction; data features; ensemble learning; healthcare; transfer learning.