基于群体智能混合框架的脑肿瘤MRI精准分类:DenseWolf-K模型实现99.64%准确率
《Scientific Reports》:A swarm intelligence-driven hybrid framework for brain tumor classification with enhanced deep features
【字体:
大
中
小
】
时间:2025年10月29日
来源:Scientific Reports 3.9
编辑推荐:
本研究针对脑肿瘤MRI图像人工诊断效率低、主观性强的问题,提出了一种集成CNN深度特征提取、LMNN度量学习和群体智能优化的混合框架。研究通过系统评估五种预训练CNN模型,结合PSO/GWO特征选择算法,在7023张MRI图像上实现了四类分类(胶质瘤、脑膜瘤、垂体瘤、健康)。最优模型DenseNet201-LMNN-GWO-KNN(DenseWolf-K)准确率达99.64%,并通过外部数据集验证和可解释性分析(特征排名+遮挡敏感度图),为临床提供高精度、可解释的AI辅助诊断方案。
脑肿瘤作为最致命的神经系统疾病之一,其早期准确诊断对治疗计划和患者预后至关重要。磁共振成像(MRI)因其高分辨率和优异的软组织对比度,成为脑部结构可视化的首选无创检查手段。然而,医生对脑部MRI图像的人工判读不仅耗时耗力,还存在观察者间差异大、高度依赖放射科医生经验等问题。随着人工智能技术的快速发展,深度学习特别是卷积神经网络(CNN)在医学图像分析中展现出巨大潜力,但其提取的高维深度特征往往存在冗余,可能导致计算负担加重和过拟合风险。
为解决这些挑战,土耳其塞尔丘克大学的Aynur Yonar在《Scientific Reports》上发表了一项创新研究,提出了一种融合CNN深度特征、大间隔最近邻(LMNN)度量学习和群体智能特征选择的混合框架,用于脑肿瘤MRI图像的四分类任务。该研究通过对7,023张T1加权轴位MRI图像(包含胶质瘤、脑膜瘤、垂体瘤和健康类别)的系统分析,发现DenseNet201模型提供了92.66%的最高基线准确率。研究创新性地在LMNN度量学习转换前后分别应用粒子群优化(PSO)和灰狼优化器(GWO)算法进行特征选择,并通过k最近邻(KNN)、支持向量机(SVM)、人工神经网络(ANN)和随机森林(RF)分类器进行评估。
关键技术方法方面,作者首先对7023张MRI图像进行预处理(调整尺寸至224×224×3、像素值归一化)和数据增强,随后使用五种预训练CNN(DenseNet201、MobileNetV2、ResNet50、ResNet101、InceptionV3)提取深度特征。接着应用LMNN算法提升特征可分性,并分别采用PSO和GWO在原始特征空间和LMNN转换空间进行特征选择。最终使用KNN、SVM、ANN、RF四种分类器进行性能评估,通过五折交叉验证和多种指标(准确率、精确度、召回率、F1分数等)全面衡量模型效果。
研究首先评估了五种CNN模型的直接分类性能。DenseNet201表现最佳,准确率达91.60%,同时具有高特异性(97.21%)和F1分数(91.94%)。MobileNetV2以90.67%的准确率紧随其后,且训练时间最短。所有模型均未出现过拟合,为后续特征提取奠定了可靠基础。
在DenseNet201特征基础上,LMNN-GWO-KNN组合(命名为DenseWolf-K)实现了99.64%的最高准确率,仅出现4例误分类。GWO相比PSO能选择更紧凑的特征子集(945个特征),且计算效率更高。LMNN的贡献具有架构依赖性,对DenseNet201和ResNet50提升显著,而对MobileNetV2和InceptionV3效果有限。
DenseWolf-K在准确率(99.64%)上超越了近期文献报道的多种方法(如EfficientNetB0-SVM的98.10%,LBP+CNN的98.90%),证明了所提框架的优越性。在外部验证集BRISC 2025上,该模型仍保持93.3%的准确率,显著高于基线CNN模型(78.9-86.7%),体现了良好的泛化能力。
研究通过特征级和图像级分析增强模型透明度。特征重要性排名显示f1203、f1384等特征在分类中起关键作用;遮挡敏感度图成功定位了各类肿瘤的解剖学相关区域(如胶质瘤的脑实质内区域、脑膜瘤的硬脑膜附着处),为临床决策提供了直观依据。
该研究提出的DenseWolf-K框架通过集成深度特征提取、度量学习和群体智能优化,实现了脑肿瘤MRI分类的近乎完美性能(99.64%准确率)。其双阶段特征选择设计(LMNN转换前后分别优化)属方法论创新,首次系统评估了度量学习对特征选择的影响。GWO相比PSO在特征紧凑性和计算效率方面更具优势。外部验证和可解释性分析进一步证明了模型的鲁棒性和临床适用性。未来研究方向包括扩展到多中心多模态数据、开发轻量级实时版本,以及应用于其他医学影像分类任务,推动AI辅助诊断系统的实际临床应用。
生物通微信公众号
生物通新浪微博
今日动态 |
人才市场 |
新技术专栏 |
中国科学人 |
云展台 |
BioHot |
云讲堂直播 |
会展中心 |
特价专栏 |
技术快讯 |
免费试用
版权所有 生物通
Copyright© eBiotrade.com, All Rights Reserved
联系信箱:
粤ICP备09063491号