将皮肤癌诊断模型压缩到嵌入式设备,实现高效低耗精准检测。
Model Compression Engine for Wearable Devices Skin Cancer Diagnosis
- 用MobileNetV2加迁移学习,适配边缘设备做皮肤病变二分类。
- 压缩后模型大小减少41%,推理速度与吞吐量提升,功耗降低93%。
- 适合资源有限地区医疗筛查,可推广至其他便携式医学诊断场景。
皮肤癌是最常见且可预防的癌症之一,但早期检测仍具挑战,尤其在医疗资源匮乏地区。本研究提出一种面向嵌入式系统的AI诊断工具,采用基于MobileNetV2的迁移学习进行皮肤病变二分类('皮肤癌'与'其他')。通过TensorRT框架对模型进行压缩与优化,部署于NVIDIA Jetson Orin Nano平台,兼顾性能与能效。多基准评估显示,优化后模型保持高精度:F1分数达87.18%,准确率93.18%,召回率81.91%。模型尺寸最大缩减41%,推理速度与吞吐量显著提升,INT8精度下能耗降低93%。结果验证了高性能、低功耗诊断系统在资源受限边缘设备上的可行性。该方法亦可推广至其他医学诊断及需要高效智能的场景,助力先进AI技术普惠偏远地区医疗。
原文摘要 · Abstract (English)
Skin cancer is one of the most prevalent and preventable types of cancer, yet its early detection remains a challenge, particularly in resource-limited settings where access to specialized healthcare is scarce. This study proposes an AI-driven diagnostic tool optimized for embedded systems to address this gap. Using transfer learning with the MobileNetV2 architecture, the model was adapted for binary classification of skin lesions into "Skin Cancer" and "Other." The TensorRT framework was employed to compress and optimize the model for deployment on the NVIDIA Jetson Orin Nano, balancing performance with energy efficiency. Comprehensive evaluations were conducted across multiple benchmarks, including model size, inference speed, throughput, and power consumption. The optimized models maintained their performance, achieving an F1-Score of 87.18% with a precision of 93.18% and recall of 81.91%. Post-compression results showed reductions in model size of up to 0.41, along with improvements in inference speed and throughput, and a decrease in energy consumption of up to 0.93 in INT8 precision. These findings validate the feasibility of deploying high-performing, energy-efficient diagnostic tools on resource-constrained edge devices. Beyond skin cancer detection, the methodologies applied in this research have broader applications in other medical diagnostics and domains requiring accessible, efficient AI solutions. This study underscores the potential of optimized AI systems to revolutionize healthcare diagnostics, thereby bridging the divide between advanced technology and underserved regions.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。