Aug, 2023
基于注意力机制的轻量级模型用于色素性皮损检测
Attention-Driven Lightweight Model for Pigmented Skin Lesion Detection
TL;DR该研究提出了一种轻量级的皮肤病变检测流程,应对类别分布不平衡和一些病变的微妙或非典型表现的挑战。该流程基于一种轻量级模型,利用幽灵特征和DFC注意力机制减少计算复杂性同时保持高性能。通过合成少数类过采样技术和各种图像增强技术来解决数据集中的类别不平衡问题。模型还采用基于知识的损失加权技术,在类别级别和实例级别对损失函数赋予不同的权重,以帮助模型关注少数类别和具有挑战性的样本。该模型在检测和分类不同的皮肤病变方面取得了92.4%的准确率,84.2%的精确度,86.9%的召回率和85.4%的F1得分,尤其在识别良性角化病样病变(BKL)和痣(NV)方面表现出色。尽管具有卓越的性能,但该模型的计算成本远低于一些准确度较低的模型,使其成为实际应用中准确度和效率都至关重要的最佳解决方案。