计算机与现代化 ›› 2024, Vol. 0 ›› Issue (05): 120-126.doi: 10.3969/j.issn.1006-2475.2024.05.021
• 图像处理 • 上一篇
摘要:
摘要:传统的色素减退性皮肤病诊断依赖于皮肤科医生主观的临床经验,难以确保每位患者的皮肤病都能被及时精确诊断。因而,亟需一个快速而不依赖于经验的诊断方法。卷积神经网络(Convolutional Neural Network, CNN)具有强大的特征识别能力,为该方法的实现提供了可能。目前基于CNN的诊断方法主要集中在ResNet50等较深的模型,虽然取得了较高的准确率,但是这些模型存在参数量大、识别慢、在移动设备上可用性差的缺点。为此,本文基于MobileNetV3-Small提出一个新的轻量级CNN模型。首先,舍弃MobileNetV3-Small中计算复杂的挤压-激发(Squeeze-and-Excitation, SE)模块,引入较轻量的高效通道注意力(Efficient Channel Attention, ECA)机制;其次,使用计算方便、稳定性好的Leaky-ReLU激活函数;最后,在卷积层中引入空洞卷积,扩大感受野。经过实验测试表明,本文提出的模型相较于现有的诊断模型实现了参数量、识别时间和FLOPs的大幅减少,满足移动应用场景下的高可用性,同时其准确率和F1值仍取得领先性能。最后,基于提出的模型设计出一个移动端的色素减退性皮肤病临床诊断工具。
中图分类号: