一种基于卷积神经网络和贝叶斯优化的脑肿瘤分类新型磁共振成像诊断方法。

A Novel MRI Diagnosis Method for Brain Tumor Classification Based on CNN and Bayesian Optimization.

作者信息

Ait Amou Mohamed, Xia Kewen, Kamhi Souha, Mouhafid Mohamed

机构信息

School of Electronics and Information Engineering, Hebei University of Technology, Tianjin 300401, China.

State Key Laboratory of Reliability and Intelligence of Electrical Equipment, Hebei University of Technology, Tianjin 300130, China.

出版信息

Healthcare (Basel). 2022 Mar 8;10(3):494. doi: 10.3390/healthcare10030494.

Abstract

Brain tumor is one of the most aggressive diseases nowadays, resulting in a very short life span if it is diagnosed at an advanced stage. The treatment planning phase is thus essential for enhancing the quality of life for patients. The use of Magnetic Resonance Imaging (MRI) in the diagnosis of brain tumors is extremely widespread, but the manual interpretation of large amounts of images requires considerable effort and is prone to human errors. Hence, an automated method is necessary to identify the most common brain tumors. Convolutional Neural Network (CNN) architectures are successful in image classification due to their high layer count, which enables them to conceive the features effectively on their own. The tuning of CNN hyperparameters is critical in every dataset since it has a significant impact on the efficiency of the training model. Given the high dimensionality and complexity of the data, manual hyperparameter tuning would take an inordinate amount of time, with the possibility of failing to identify the optimal hyperparameters. In this paper, we proposed a Bayesian Optimization-based efficient hyperparameter optimization technique for CNN. This method was evaluated by classifying 3064 T-1-weighted CE-MRI images into three types of brain tumors (Glioma, Meningioma, and Pituitary). Based on Transfer Learning, the performance of five well-recognized deep pre-trained models is compared with that of the optimized CNN. After using Bayesian Optimization, our CNN was able to attain 98.70% validation accuracy at best without data augmentation or cropping lesion techniques, while VGG16, VGG19, ResNet50, InceptionV3, and DenseNet201 achieved 97.08%, 96.43%, 89.29%, 92.86%, and 94.81% validation accuracy, respectively. Moreover, the proposed model outperforms state-of-the-art methods on the CE-MRI dataset, demonstrating the feasibility of automating hyperparameter optimization.

摘要

脑肿瘤是当今最具侵袭性的疾病之一,如果在晚期被诊断出来,会导致患者寿命极短。因此,治疗规划阶段对于提高患者的生活质量至关重要。磁共振成像(MRI)在脑肿瘤诊断中的应用极为广泛,但对大量图像进行人工解读需要耗费大量精力,且容易出现人为错误。因此,需要一种自动化方法来识别最常见的脑肿瘤。卷积神经网络(CNN)架构因其层数众多,能够有效地自行构思特征,在图像分类方面取得了成功。CNN超参数的调整在每个数据集中都至关重要,因为它对训练模型的效率有重大影响。鉴于数据的高维度和复杂性,手动调整超参数将花费大量时间,并且有可能无法识别最优超参数。在本文中,我们提出了一种基于贝叶斯优化的CNN高效超参数优化技术。该方法通过将3064张T1加权对比增强MRI图像分类为三种类型的脑肿瘤(胶质瘤、脑膜瘤和垂体瘤)进行评估。基于迁移学习,将五个公认的深度预训练模型的性能与优化后的CNN的性能进行了比较。在使用贝叶斯优化后,我们的CNN在不使用数据增强或裁剪病变技术的情况下,最高能够达到98.70%的验证准确率,而VGG16、VGG19、ResNet50、InceptionV3和DenseNet201的验证准确率分别为97.08%、96.43%、89.29%、92.86%和94.81%。此外,所提出的模型在对比增强MRI数据集上优于现有方法,证明了超参数优化自动化的可行性。

https://cdn.ncbi.nlm.nih.gov/pmc/blobs/5a9a/8949584/8a34ecd88dec/healthcare-10-00494-g001.jpg

文献AI研究员

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

立即体验

用中文搜PubMed

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

马上搜索