基于改进AlexNet的可变形卷积皮肤病变识别算法

Improved AlexNet Based Recognition Algorithm for Deformable Convolution Skin Lesions

  • 摘要: 为了有效解决类间相似度高、类内差异化大、数据类别不平衡的皮肤病变识别,提出了基于改进AlexNet的可变形卷积网络皮肤病变识别算法. 构建改进的AlexNet可变形卷积网络模型,增加采样偏移量,使不同位置的卷积核采样点可根据图像内容自适应变化,自动调整不同尺度或感受野,提取比标准卷积更精细的特征. 使用交叉熵损失函数和焦点损失函数的加权损失函数,削弱易分类样本在训练中所占的权重,使模型专注于相似度高、易错分的样本,解决样本比例不平衡的问题,优化模型的识别率. 在HAM10000数据集上进行仿真实验,主客观的实验结果表明, 提出的方法在7种皮肤病变上的识别优于现有方法,具有更高的准确性、特异性和鲁棒性.

     

    Abstract: In order to solve the problem of poor adaptability of standard convolution to unknown changes and high similarity between classes of skin lesions, large differences within a class, and serious class imbalance in data, a skin lesion recognition algorithm of deformable convolutional network was proposed based on improved AlexNet. First, an improved AlexNet deformable convolutional network model was established, adding sampling offset, making the position of the convolution kernel sampling point change adaptively according to the image content. And it was arranged to automatically adjust different scales or acceptable information field, and to extract finer features than standard convolution. And then, a weighted loss function of the cross-entropy loss function and the focus loss function was used to reduce the weight of samples classified easily in training, making the model focus on difficult and error-prone samples to solve the problem of sample proportion imbalance and to optimize the recognition rate of the model. Finally, a simulation experiment was carried out with the HAM10000 data set. The subjective and objective experimental results show that the proposed method is superior to the existing methods in identifying seven skin lesions, and has higher accuracy, specificity and robustness.

     

/

返回文章
返回
Baidu
map