融合深度特征与优化分类器,提升脑肿瘤MRI分类准确率
Hierarchical Deep Feature Fusion and Ensemble Learning for Enhanced Brain Tumor MRI Classification
- 用ViT提取深层特征,双层集成融合特征与预测结果
- 在两个公开数据集上准确率超越现有方法,达98.6%
- 适合医学影像分析、临床辅助诊断研究者参考
精准的脑肿瘤分类对医学影像诊断和治疗规划至关重要。本研究提出一种新型双重集成框架,结合预训练深度学习模型进行特征提取,与优化后的机器学习分类器协同实现鲁棒分类。框架包含全面的脑部磁共振图像(MRI)预处理与数据增强,利用预训练视觉变压器(ViT)网络通过迁移学习提取深层特征。创新点在于双层集成策略:特征级集成融合表现最佳的ViT模型输出特征,分类器级集成则聚合经过超参数优化的机器学习分类器预测结果。在两个公开的Kaggle MRI脑肿瘤数据集上的实验表明,该方法显著优于当前先进方法,验证了特征与分类器融合的重要性。研究还强调了超参数优化(HPO)和先进预处理技术在提升诊断准确率与可靠性中的关键作用,推动深度学习与机器学习在临床相关医学图像分析中的融合应用。
原文摘要 · Abstract (English)
Accurate brain tumor classification is crucial in medical imaging to ensure reliable diagnosis and effective treatment planning. This study introduces a novel double ensembling framework that synergistically combines pre-trained deep learning (DL) models for feature extraction with optimized machine learning (ML) classifiers for robust classification. The framework incorporates comprehensive preprocessing and data augmentation of brain magnetic resonance images (MRI), followed by deep feature extraction using transfer learning with pre-trained Vision Transformer (ViT) networks. The novelty lies in the dual-level ensembling strategy: feature-level ensembling, which integrates deep features from the top-performing ViT models, and classifier-level ensembling, which aggregates predictions from hyperparameter-optimized ML classifiers. Experiments on two public Kaggle MRI brain tumor datasets demonstrate that this approach significantly surpasses state-of-the-art methods, underscoring the importance of feature and classifier fusion. The proposed methodology also highlights the critical roles of hyperparameter optimization (HPO) and advanced preprocessing techniques in improving diagnostic accuracy and reliability, advancing the integration of DL and ML for clinically relevant medical image analysis.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。