Jun, 2024

利用知识蒸馏实现轻量级皮肤癌分类:平衡准确性与计算效率

TL;DR我们提出基于知识蒸馏的方法来创建一个轻量级且高性能的皮肤癌分类器,通过融合ResNet152V2、ConvNeXtBase和ViT Base等三个模型来创建一个有效的教师模型,进而进行指导以得到一个大小为2.03 MB的轻量级学生模型,通过16位量化将其进一步压缩到469.77 KB,以实现边缘设备的平滑整合,在六个阶段的图像预处理、数据增强以及严谨的消融研究的基础上,在HAM10000数据集上取得了98.75%的准确率,在Kaggle数据集上取得了98.94%的准确率,在资源受限的环境中,我们的模型以其高准确率和紧凑的大小成为一个潜在的准确的皮肤癌分类选择。