一种使用磁共振成像进行脑肿瘤多分类的新型集成框架。

A Novel Ensemble Framework for Multi-Classification of Brain Tumors Using Magnetic Resonance Imaging.

作者信息

Çetin-Kaya Yasemin, Kaya Mahir

机构信息

Department of Computer Engineering, Faculty of Engineering and Architecture, Tokat Gaziosmanpaşa University, Tokat 60250, Turkey.

出版信息

Diagnostics (Basel). 2024 Feb 9;14(4):383. doi: 10.3390/diagnostics14040383.

Abstract

Brain tumors can have fatal consequences, affecting many body functions. For this reason, it is essential to detect brain tumor types accurately and at an early stage to start the appropriate treatment process. Although convolutional neural networks (CNNs) are widely used in disease detection from medical images, they face the problem of overfitting in the training phase on limited labeled and insufficiently diverse datasets. The existing studies use transfer learning and ensemble models to overcome these problems. When the existing studies are examined, it is evident that there is a lack of models and weight ratios that will be used with the ensemble technique. With the framework proposed in this study, several CNN models with different architectures are trained with transfer learning and fine-tuning on three brain tumor datasets. A particle swarm optimization-based algorithm determined the optimum weights for combining the five most successful CNN models with the ensemble technique. The results across three datasets are as follows: Dataset 1, 99.35% accuracy and 99.20 F1-score; Dataset 2, 98.77% accuracy and 98.92 F1-score; and Dataset 3, 99.92% accuracy and 99.92 F1-score. We achieved successful performances on three brain tumor datasets, showing that the proposed framework is reliable in classification. As a result, the proposed framework outperforms existing studies, offering clinicians enhanced decision-making support through its high-accuracy classification performance.

摘要

脑肿瘤会产生致命后果,影响身体的许多功能。因此,准确且早期地检测脑肿瘤类型以便启动适当的治疗过程至关重要。尽管卷积神经网络(CNN)在从医学图像中进行疾病检测方面被广泛使用,但它们在有限的标记且多样性不足的数据集上进行训练时面临过拟合问题。现有研究使用迁移学习和集成模型来克服这些问题。在审视现有研究时,很明显缺乏将与集成技术一起使用的模型和权重比。通过本研究提出的框架,在三个脑肿瘤数据集上使用迁移学习和微调对几种具有不同架构的CNN模型进行了训练。一种基于粒子群优化的算法确定了使用集成技术组合五个最成功的CNN模型的最佳权重。三个数据集的结果如下:数据集1,准确率99.35%,F1分数99.20;数据集2,准确率98.77%,F1分数98.92;数据集3,准确率99.92%,F1分数99.92。我们在三个脑肿瘤数据集上取得了成功的表现,表明所提出的框架在分类方面是可靠的。因此,所提出的框架优于现有研究,通过其高精度的分类性能为临床医生提供了增强的决策支持。

https://cdn.ncbi.nlm.nih.gov/pmc/blobs/b0d8/10888105/6080d8614ec0/diagnostics-14-00383-g001.jpg

文献AI研究员

20分钟写一篇综述,助力文献阅读效率提升50倍

立即体验

用中文搜PubMed

大模型驱动的PubMed中文搜索引擎

马上搜索