arXiv:2512.07241cs.CV2025-12

轻量模型融合手工特征,实现98.9%脑肿瘤自动分类准确率

Squeezed-Eff-Net: Edge-Computed Boost of Tomography Based Brain Tumor Classification leveraging Hybrid Neural Network Architecture

  • 用SqueezeNet与EfficientNet-B0混合架构,结合多种图像纹理特征
  • 在7023张MRI切片上达到98.93%准确率,增强后达99.08%
  • 参数少于210万,计算量低于1.2 GFLOPs,适合边缘设备部署

脑肿瘤是常见且危险的神经系统疾病,需及时准确诊断以制定治疗方案。尽管磁共振成像(MRI)广泛应用,肿瘤分割仍困难且耗时,易受观察者差异影响。为此,本文提出一种基于SqueezeNet v1(轻量模型)与EfficientNet-B0(高性能模型)的混合深度学习模型,并融合手工提取的放射组学特征,包括方向梯度直方图(HOG)、局部二值模式(LBP)、Gabor滤波器和小波变换。模型仅在公开的Nickparvar脑肿瘤MRI数据集上训练与测试,该数据集包含7,023个对比增强T1加权轴向MRI切片,分为胶质瘤、脑膜瘤、垂体瘤和无肿瘤四类。测试准确率达98.93%,使用测试时增强(TTA)后提升至99.08%,表现出优异泛化能力。所提混合网络在计算效率与诊断精度间取得平衡,仅需少于210万参数和小于1.2 GFLOPs计算量。手工特征提升了纹理敏感性,EfficientNet-B0骨干网络捕捉复杂层级特征。最终模型具备接近临床可用的自动MRI肿瘤分类能力,适用于临床决策支持系统。

原文摘要 · Abstract (English)

Brain tumors are one of the most common and dangerous neurological diseases which require a timely and correct diagnosis to provide the right treatment procedures. Even with the promotion of magnetic resonance imaging (MRI), the process of tumor delineation is difficult and time-consuming, which is prone to inter-observer error. In order to overcome these limitations, this work proposes a hybrid deep learning model based on SqueezeNet v1 which is a lightweight model, and EfficientNet-B0, which is a high-performing model, and is enhanced with handcrafted radiomic descriptors, including Histogram of Oriented Gradients (HOG), Local Binary Patterns (LBP), Gabor filters and Wavelet transforms. The framework was trained and tested only on publicly available Nickparvar Brain Tumor MRI dataset, which consisted of 7,023 contrast-enhanced T1-weighted axial MRI slices which were categorized into four groups: glioma, meningioma, pituitary tumor, and no tumor. The testing accuracy of the model was 98.93% that reached a level of 99.08% with Test Time Augmentation (TTA) showing great generalization and power. The proposed hybrid network offers a compromise between computation efficiency and diagnostic accuracy compared to current deep learning structures and only has to be trained using fewer than 2.1 million parameters and less than 1.2 GFLOPs. The handcrafted feature addition allowed greater sensitivity in texture and the EfficientNet-B0 backbone represented intricate hierarchical features. The resulting model has almost clinical reliability in automated MRI-based classification of tumors highlighting its possibility of use in clinical decision-support systems.

脑肿瘤分类轻量模型医学影像混合网络

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。