A Hybrid Deep Learning Model with Data Augmentation to Improve Tumor Classification Using MRI Images

一种结合数据增强的混合深度学习模型,用于改进基于MRI图像的肿瘤分类。

阅读:1

Abstract

BACKGROUND: Cancer ranks second among the causes of mortality worldwide, following cardiovascular diseases. Brain cancer, in particular, has the lowest survival rate of any form of cancer. Brain tumors vary in their morphology, texture, and location, which determine their classification. The accurate diagnosis of tumors enables physicians to select the optimal treatment strategies and potentially prolong patients' lives. Researchers who have implemented deep learning models for the diagnosis of diseases in recent years have largely focused on deep neural network optimization to enhance their performance. This involves implementing models with the best performance and incorporating various network architectures by configuring their hyperparameters. METHODS: This paper presents a novel hybrid approach for improved brain tumor classification by combining CNNs and EfficientNetV2B3 for feature extraction, followed by K-nearest neighbors (KNN) for classification, which has been described as one of the simplest machine learning algorithms based on supervised learning techniques. The KNN algorithm assumes similarities between new cases and available cases and assigns new cases to the category that most closely resembles the available categories. RESULTS: To evaluate the recommended method's efficacy, two widely known benchmark MRI datasets were utilized in the experiments. The initial dataset consisted of 3064 MRI images depicting meningiomas, gliomas, and pituitary tumors. Images from two classes, consisting of healthy brains and brain tumors, were included in the second dataset, which was obtained from Kaggle. CONCLUSIONS: In order to enhance the performance even further, this study concatenated the CNN and EfficientNetV2B3's flattened outputs before feeding them into the KNN classifier. The proposed framework was run on these two different datasets and demonstrated outstanding performance, with accuracy of 99.51% and 99.8%, respectively.

特别声明

1、本页面内容包含部分的内容是基于公开信息的合理引用;引用内容仅为补充信息,不代表本站立场。

2、若认为本页面引用内容涉及侵权,请及时与本站联系,我们将第一时间处理。

3、其他媒体/个人如需使用本页面原创内容,需注明“来源:[生知库]”并获得授权;使用引用内容的,需自行联系原作者获得许可。

4、投稿及合作请联系:info@biocloudy.com。