LLM2D
UltraLightSqueezeNet:一种针对资源受限设备的疟疾分类深度学习架构,参数量最多减少54倍
UltraLightSqueezeNet: A Deep Learning Architecture for Malaria Classification with up to 54x fewer trainable parameters for resource constrained devices
作者: Suresh Babu Nettur, Shanthi Karpurapu, Unnati Nettur, Likhit Sagar Gajja, Sravanthy Myneni, Akhil Dusi, Lalithya Posham
发布日期: 2/4/2025
arXiv ID: oai:arXiv.org:2501.14172v2

摘要

arXiv:2501.14172v2 通知类型: 交叉替换 摘要:轻量级深度学习方法在疟疾检测中的应用引起了人们对其实施数字化诊断在资源有限环境中增强诊断能力的兴趣。在本研究中,我们选择了SqueezeNet1.1,因为它是最受欢迎的轻量级架构之一。SqueezeNet1.1是SqueezeNet1.0的后续版本,相比原始模型在计算效率上提高了2.4倍。我们提出了并对SqueezeNet1.1架构进行了三个超轻量级架构变体的实现,分别为变体1(一个fire模块)、变体2(两个fire模块)和变体3(四个fire模块),这三个变体比SqueezeNetV1.1(八个fire模块)更为紧凑。这些模型被实现用于评估能在不牺牲准确性的情况下,在疟疾血液细胞分类中实现最佳计算效率的变体。这些模型使用NIH疟疾数据集进行了训练和评估。我们根据准确性、召回率、精确率、F1分数和曲线下面积(AUC)等指标评估了每个模型的性能。结果显示,SqueezeNet1.1模型在所有指标上均表现出最佳性能,分类准确率为97.12%。变体3(四个fire模块)提供了具有竞争力的替代方案,其准确率为96.55%,计算开销减少了6倍,几乎与SqueezeNet1.1的表现相同。变体2和变体1的性能略低于变体3,变体2(两个fire模块)将计算开销减少了28倍,而变体1(一个fire模块)将可训练参数减少了54倍,相比SqueezeNet1.1。这些发现表明,我们的SqueezeNet1.1架构变体提供了一种灵活的疟疾检测方法,能够根据资源限制和性能选择合适的变体。