轻量级CNN提升脑瘤分类准确率,兼顾效率与泛化能力。
Efficient Brain Tumor Classification with Lightweight CNN Architecture: A Novel Approach
- 采用分离卷积与SE模块增强特征提取,减少参数量。
- 验证集准确率达99.22%,测试集98.44%,比现有模型高0.5%~1.0%。
- 适合医疗影像分析场景,尤其关注高效低资源部署的研究者。
基于MRI图像的脑肿瘤分类在医学诊断中至关重要,早期精准检测显著影响患者预后。尽管深度学习(特别是CNN)取得进展,但多数模型难以平衡精度与计算效率,且在不同数据集上鲁棒性不足。为此,我们提出一种新架构,融合分离卷积与挤压激励(SE)模块,以提升特征提取能力并保持计算高效。模型进一步引入批归一化与丢弃层防止过拟合,确保性能稳定可靠。该模型轻量化设计源于分离卷积降低参数量,并以全局平均池化替代全连接层,显著降低计算复杂度。实验表明,本模型在精度上优于其他模型0.5%至1.0%,损失降低1.5%至2.5%。验证集准确率为99.22%,测试集准确率为98.44%。结果表明其对不同类型脑瘤具有良好的泛化能力,为临床应用提供可靠工具。本工作树立了新基准,为未来优化医学图像分析中深度学习模型的精度与效率奠定基础。
原文摘要 · Abstract (English)
Brain tumor classification using MRI images is critical in medical diagnostics, where early and accurate detection significantly impacts patient outcomes. While recent advancements in deep learning (DL), particularly CNNs, have shown promise, many models struggle with balancing accuracy and computational efficiency and often lack robustness across diverse datasets. To address these challenges, we propose a novel model architecture integrating separable convolutions and squeeze and excitation (SE) blocks, designed to enhance feature extraction while maintaining computational efficiency. Our model further incorporates batch normalization and dropout to prevent overfitting, ensuring stable and reliable performance. The proposed model is lightweight because it uses separable convolutions, which reduce the number of parameters, and incorporates global average pooling instead of fully connected layers to minimize computational complexity while maintaining high accuracy. Our model does better than other models by about 0.5% to 1.0% in accuracy and 1.5% to 2.5% in loss reduction, as shown by many experiments. It has a validation accuracy of 99.22% and a test accuracy of 98.44%. These results highlight the model's ability to generalize effectively across different brain tumour types, offering a robust tools for clinical applications. Our work sets a new benchmark in the field, providing a foundation for future research in optimizing the accuracy and efficiency of DL models for medical image analysis.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。