计算机工程与设计2024,Vol.45Issue(9) :2859-2865.DOI:10.16208/j.issn1000-7024.2024.09.040

融合注意力与上下文信息的皮肤癌图像分割模型

Skin cancer image segmentation via combining attention and context information

支慧芳 韩建新 吴永飞
计算机工程与设计2024,Vol.45Issue(9) :2859-2865.DOI:10.16208/j.issn1000-7024.2024.09.040

融合注意力与上下文信息的皮肤癌图像分割模型

Skin cancer image segmentation via combining attention and context information

支慧芳 1韩建新 2吴永飞1
扫码查看

作者信息

  • 1. 太原理工大学 大数据学院,山西 太原 030024
  • 2. 晋中学院 物理与电子工程系,山西 晋中 030600
  • 折叠

摘要

为提高黑色素瘤分割性能,提出一种结合注意力机制和上下文信息的U-Net网络.以Resnet-34网络作为编码器,在跳跃连接中加入坐标注意力,通过捕捉精准的位置信息定位更准确的目标区域;设计上下文信息模块强化对前景特征的学习能力;加入高效通道注意力模块,重新校准权重并获得更高质量的分割图.在公共数据集ISIC 2017上验证改进模型,其结果表明,该模型召回率、F1分数达到85.29%、87.03%,与现有方法对比,在准确率、交并比、召回率、F1分数产生竞争性结果.

Abstract

To improve the performance of skin lesion segmentation,a version of U-Net network based on convolutional neural network was proposed combined with attention and context information.Using Resnet-34 network as the encoder,coordinate at-tention mechanism was added in the process of skip connections,and more accurate target area was located by capturing accurate position information.The context information module was designed to strengthen the learning ability of the foreground features and the high-efficiency channel attention module was added to recalibrate the weight and obtain a higher quality segmentation map.The designed model was verified on the public dataset ISIC 2017.Experimental results show that the recall rate and F1-score reach 85.29%and 87.03%,respectively.The proposed method outperforms the existing methods in terms of accuracy,recall rate(Recall),intersection over union(IOU),F1-score,and yields competitive results.

关键词

病变分割/多尺度融合/注意力机制/上下文信息/卷积神经网络/U-Net型网络/坐标注意力/高效通道注意力

Key words

lesion segmentation/multi-scale fusion/attention mechanism/context information/CNN/U-Net network/coordi-nate attention/efficient channel attention

引用本文复制引用

基金项目

国家自然科学基金项目(61901292)

山西省应用基础研究计划基金项目(201901D211080)

山西省应用基础研究计划基金项目(20210302124542)

出版年

2024
计算机工程与设计
中国航天科工集团二院706所

计算机工程与设计

CSTPCD北大核心
影响因子:0.617
ISSN:1000-7024
参考文献量1
段落导航相关论文