arXiv:2409.05420cs.CVcs.AI2024-09被引 19

用注意力增强与引导解码,提升皮肤病变分割精度

AD-Net: Attention-based dilated convolutional residual network with guided decoder for robust skin lesion segmentation

  • 采用带注意力的空间特征增强模块和空洞残差网络
  • 在四个数据集上优于现有方法,无需数据增强
  • 参数少、收敛快,适合标注数据有限场景

在皮肤癌诊疗的计算机辅助诊断工具中,皮肤病变分割至关重要。然而,由于外观、对比度、纹理及边界模糊等固有差异,精确分割仍具挑战。本文提出一种鲁棒性方法——AD-Net,基于空洞卷积残差网络,引入注意力空间特征增强块(ASFEB)并采用引导解码策略。每个空洞残差块通过不同膨胀率的空洞卷积扩大感受野。为增强编码器的空间特征,我们在跳跃连接中加入基于平均池化与最大池化的特征融合,并通过全局平均池化与卷积运算动态加权。此外,引导解码策略使每个解码块使用独立损失函数优化,提升特征学习能力。实验表明,该方法参数更少,训练收敛更快,且在四个公开基准数据集上表现优异,即使未使用数据增强也超越当前先进方法。通过威尔科克斯符号秩检验验证了其有效性。

原文摘要 · Abstract (English)

In computer-aided diagnosis tools employed for skin cancer treatment and early diagnosis, skin lesion segmentation is important. However, achieving precise segmentation is challenging due to inherent variations in appearance, contrast, texture, and blurry lesion boundaries. This research presents a robust approach utilizing a dilated convolutional residual network, which incorporates an attention-based spatial feature enhancement block (ASFEB) and employs a guided decoder strategy. In each dilated convolutional residual block, dilated convolution is employed to broaden the receptive field with varying dilation rates. To improve the spatial feature information of the encoder, we employed an attention-based spatial feature enhancement block in the skip connections. The ASFEB in our proposed method combines feature maps obtained from average and maximum-pooling operations. These combined features are then weighted using the active outcome of global average pooling and convolution operations. Additionally, we have incorporated a guided decoder strategy, where each decoder block is optimized using an individual loss function to enhance the feature learning process in the proposed AD-Net. The proposed AD-Net presents a significant benefit by necessitating fewer model parameters compared to its peer methods. This reduction in parameters directly impacts the number of labeled data required for training, facilitating faster convergence during the training process. The effectiveness of the proposed AD-Net was evaluated using four public benchmark datasets. We conducted a Wilcoxon signed-rank test to verify the efficiency of the AD-Net. The outcomes suggest that our method surpasses other cutting-edge methods in performance, even without the implementation of data augmentation strategies.

皮肤分割注意力机制空洞卷积医学图像

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。