arXiv:2605.19207cs.CVcs.AI2026-05

轻量化量化模型在低资源医疗影像中实现高精度脑肿瘤分类。

Quantized Machine Learning Models for Medical Imaging in Low-Resource Healthcare Settings

  • 结合知识蒸馏与量化感知训练,压缩模型体积。
  • 量化后模型仅5.76MB,准确率仍达82.37%,压缩6.14倍。
  • 适合部署在计算资源受限的基层医疗机构。

深度学习在医学影像分析中表现优异,但在低资源临床环境中部署仍面临算力、内存和功耗限制。本文提出一种多策略压缩框架,用于磁共振成像(MRI)脑肿瘤分类,包括量化感知训练、从DenseNet-101教师模型向紧凑DenseNet-32学生模型的知识蒸馏,以及对轻量级MobileNetV2骨干网络的低比特后训练量化和Float16量化。基于包含胶质瘤、脑膜瘤、垂体瘤及健康对照的多类脑肿瘤MRI数据集,对MobileNetV2管道进行了完整实验验证,通过三阶段迁移学习训练分类器,并使用TensorFlow Lite实现Float16量化。DenseNet-based蒸馏与量化感知训练被描述为互补压缩方法,其完整评估留待未来工作。实验结果表明,移动网络基线量化模型在验证集上达到82.37%准确率,相比全精度基线82.20%无明显损失,模型大小由35.34MB降至5.76MB,压缩比达6.14倍,且各类肿瘤诊断性能保持均匀稳定。结果表明,轻量化量化模型可在资源受限医疗场景中实现可临床应用的脑肿瘤筛查。

原文摘要 · Abstract (English)

Deep learning models have shown strong performance in medical image analysis, but deploying them in low-resource clinical environments remains difficult due to computational, memory, and power constraints. This paper presents a multi-strategy compression framework for brain tumor classification from MRI, encompassing quantization-aware training, knowledge distillation from a DenseNet-101 teacher to a compact DenseNet-32 student with low-bit post-training quantization, and Float16 post-training quantization on a lightweight MobileNetV2 backbone. Using a multi-class brain tumor MRI dataset containing glioma, meningioma, pituitary tumors, and healthy controls, we provide full experimental validation of the MobileNetV2-based pipeline, training the classifier through a three-stage transfer learning process and applying Float16 quantization via TensorFlow Lite. The DenseNet-based distillation and quantization-aware training strategies are described as complementary compression approaches within the framework, with their complete empirical evaluation reserved for future work. Experimental results on the MobileNetV2 pipeline show that the quantized model achieves 82.37 percent validation accuracy compared to the 82.20 percent full-precision baseline, reducing model size from 35.34 MB to 5.76 MB, a 6.14x compression ratio with no meaningful accuracy loss. Per-class evaluation confirms that quantization preserves diagnostic performance uniformly across all four tumor categories. These findings demonstrate that lightweight quantized models can deliver clinically viable brain tumor screening in resource-constrained healthcare settings.

医学影像模型压缩量化轻量化

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