arXiv:2505.21597eess.IVcs.CV2025-05被引 7

轻量CNN在皮肤癌分类中大幅降低计算开销,几乎不损失精度。

Optimizing Deep Learning for Skin Cancer Classification: A Computationally Efficient CNN with Minimal Accuracy Trade-Off

  • 设计轻量级CNN,参数减少96.7%至69.2万
  • 在HAM10000数据集上准确率仅降0.022%
  • 推理耗时和能耗显著下降,适合移动端部署

深度学习在医学图像分析中的快速发展极大提升了皮肤癌分类的准确性。然而,当前基于迁移学习的先进模型(如ResNet50)存在显著计算开销,难以在资源受限环境中部署。本研究提出一种定制化CNN模型,在将参数从ResNet50的2390万降至69.2万(减少96.7%)的同时,保持分类准确率偏差小于0.022%。对HAM10000数据集的实证分析显示,尽管迁移学习模型仅带来约0.022%的准确率提升,却导致浮点运算量(FLOPs)激增13,216.76%。相比之下,本模型仅需3004万FLOPs(远低于ResNet50的40亿),显著降低能耗、内存占用和推理延迟。结果揭示了模型复杂度与实际可行性的权衡,证明该优化模型是移动与边缘皮肤癌诊断的理想方案。

原文摘要 · Abstract (English)

The rapid advancement of deep learning in medical image analysis has greatly enhanced the accuracy of skin cancer classification. However, current state-of-the-art models, especially those based on transfer learning like ResNet50, come with significant computational overhead, rendering them impractical for deployment in resource-constrained environments. This study proposes a custom CNN model that achieves a 96.7\% reduction in parameters (from 23.9 million in ResNet50 to 692,000) while maintaining a classification accuracy deviation of less than 0.022\%. Our empirical analysis of the HAM10000 dataset reveals that although transfer learning models provide a marginal accuracy improvement of approximately 0.022\%, they result in a staggering 13,216.76\% increase in FLOPs, considerably raising computational costs and inference latency. In contrast, our lightweight CNN architecture, which encompasses only 30.04 million FLOPs compared to ResNet50's 4.00 billion, significantly reduces energy consumption, memory footprint, and inference time. These findings underscore the trade-off between the complexity of deep models and their real-world feasibility, positioning our optimized CNN as a practical solution for mobile and edge-based skin cancer diagnostics.

皮肤癌分类轻量模型计算效率

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