arXiv:2501.14172cs.LGcs.AI2025-01被引 11

提出轻量版SqueezeNet,54倍减少参数,适合低算力设备疟疾分类

UltraLightSqueezeNet: A Deep Learning Architecture for Malaria Classification with up to 54x fewer trainable parameters for resource constrained devices

  • 在SqueezeNet1.1基础上设计三类极简变体,仅用1~4个fire模块
  • 最简模型参数少54倍,准确率达96.55%,接近原模型97.12%水平
  • 适合移动端或嵌入式设备部署,兼顾精度与计算效率

针对资源受限环境下的疟疾检测需求,本文以SqueezeNet1.1为基础,提出三种超轻量变体:Variant 1(1个fire模块)、Variant 2(2个fire模块)和Variant 3(4个fire模块),均比原版的8个fire模块更紧凑。所有模型在NIH Malaria数据集上训练评估,结果显示SqueezeNet1.1准确率最高,达97.12%。Variant 3性能接近,准确率为96.55%,计算开销降低6倍;Variant 2计算开销降低28倍;Variant 1可实现54倍参数量缩减,且保持良好分类性能。实验表明,该系列变体为疟疾分类提供了灵活的轻量化解决方案。

原文摘要 · Abstract (English)

Lightweight deep learning approaches for malaria detection have gained attention for their potential to enhance diagnostics in resource constrained environments. For our study, we selected SqueezeNet1.1 as it is one of the most popular lightweight architectures. SqueezeNet1.1 is a later version of SqueezeNet1.0 and is 2.4 times more computationally efficient than the original model. We proposed and implemented three ultra-lightweight architecture variants to SqueezeNet1.1 architecture, namely Variant 1 (one fire module), Variant 2 (two fire modules), and Variant 3 (four fire modules), which are even more compact than SqueezeNetV1.1 (eight fire modules). These models were implemented to evaluate the best performing variant that achieves superior computational efficiency without sacrificing accuracy in malaria blood cell classification. The models were trained and evaluated using the NIH Malaria dataset. We assessed each model's performance based on metrics including accuracy, recall, precision, F1-score, and Area Under the Curve (AUC). The results show that the SqueezeNet1.1 model achieves the highest performance across all metrics, with a classification accuracy of 97.12%. Variant 3 (four fire modules) offers a competitive alternative, delivering almost identical results (accuracy 96.55%) with a 6x reduction in computational overhead compared to SqueezeNet1.1. Variant 2 and Variant 1 perform slightly lower than Variant 3, with Variant 2 (two fire modules) reducing computational overhead by 28x, and Variant 1 (one fire module) achieving a 54x reduction in trainable parameters compared to SqueezeNet1.1. These findings demonstrate that our SqueezeNet1.1 architecture variants provide a flexible approach to malaria detection, enabling the selection of a variant that balances resource constraints and performance.

轻量模型疟疾检测嵌入式部署SqueezeNet

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