《Scientific Reports》:Multi-class classification of brain tumors using optimized CNN and transfer learning techniques
编辑推荐:
本研究针对脑肿瘤早期诊断难题,提出一种基于四卷积块优化的CNN模型,实现对神经胶质瘤、脑膜瘤、垂体瘤及无肿瘤四类MRI图像的自动分类。模型采用Adam优化器最小化交叉熵损失,以98.5%的分类准确率超越ResNet50等迁移学习模型,其中无肿瘤类别识别准确率最高达0.98。该技术有望提升放射科医生的临床诊断效率。
在大脑这个人体最精密的指挥中心里,异常细胞的增殖就像潜伏的破坏者,不仅扰乱正常的神经功能,更可能危及生命。脑肿瘤作为中枢神经系统最常见的病变之一,其早期发现和精准分类直接关系到治疗方案的制定与患者预后。目前临床上主要依赖磁共振成像(MRI)技术进行检测,但不同病理类型的肿瘤在影像学表现上存在重叠特征,即使对经验丰富的放射科医生也是不小的挑战。
近日发表在《Scientific Reports》的一项研究提出了一种基于优化卷积神经网络(CNN)的自动分类系统,为脑肿瘤诊断提供了新的解决方案。研究人员针对四类常见脑部状况(神经胶质瘤、脑膜瘤、垂体瘤和无肿瘤)的MRI图像,设计了一个包含四层卷积块的深度学习架构。通过Adam优化器对模型进行训练,并采用交叉熵损失函数来提升分类性能。
关键技术方法包括:构建四卷积块的CNN主干网络,使用Adam优化算法进行模型参数优化,采用交叉熵损失函数作为训练目标,在包含四类脑部状况(神经胶质瘤、脑膜瘤、垂体瘤、无肿瘤)的MRI数据集上进行模型训练与验证,并与ResNet50、VGG19、DenseNet121三种迁移学习模型进行对比实验。
模型性能比较
本研究提出的CNN模型在四分类任务中达到98.5%的整体准确率,显著优于ResNet50(93.2%)、VGG19和DenseNet121等迁移学习模型。这表明专门设计的轻量级网络结构在特定医学影像任务上可以超越复杂的预训练模型。
类别特异性分析
按肿瘤类型细分的评估显示:无肿瘤(Category 3)识别准确率最高(0.98),其次为垂体瘤(Category 2,0.96)、脑膜瘤(Category 1,0.89)和神经胶质瘤(Category 0,0.86)。这种差异可能与不同肿瘤在MRI影像中的形态特征明显程度有关。
结论与展望
该研究证实了优化CNN模型在脑肿瘤多分类任务中的有效性,特别是对无肿瘤病例的高识别率有助于减少不必要的进一步检查。虽然神经胶质瘤的分类精度相对较低,但整体性能已展现临床应用潜力。未来通过扩大数据集和融入多模态影像数据,有望进一步提升模型鲁棒性,最终形成辅助放射科医生诊断的可靠工具。