arXiv:2505.05913cs.CV2025-05被引 2

通过双层特征均衡提升医学图像分割精度,解决边界像素误分问题。

DFEN: Dual Feature Equalization Network for Medical Image Segmentation

  • 设计图像级与类别级双重特征均衡模块,增强像素表示
  • 在BUSI、ISIC2017等4个数据集上达到最优性能
  • 适合关注医学图像细粒度分割的研究者与临床应用开发者

现有医学图像分割方法多从整体图像视角提取上下文特征信息,虽表现良好,却未考虑边界区域及少数类像素更依赖其他类上下文信息的问题,导致因上下文特征不均引发误分类。本文提出基于Swin Transformer与卷积神经网络混合架构的双特征均衡网络(DFEN),通过图像级和类别级特征均衡机制增强像素特征表示:首先利用图像级均衡模块统一图像内各像素的上下文信息;其次通过类别级均衡模块聚合同类别区域,使同类像素特征一致;最后学习两种均衡信息的加权融合。采用Swin Transformer作为编码器与解码器,强化长程依赖与空间关联捕捉能力。在乳腺超声(BUSI)、皮肤影像(ISIC2017)、心脏诊断挑战(ACDC)及PH²数据集上进行大量实验,结果表明该方法性能达到当前最优水平。代码已开源于https://github.com/JianJianYin/DFEN。

原文摘要 · Abstract (English)

Current methods for medical image segmentation primarily focus on extracting contextual feature information from the perspective of the whole image. While these methods have shown effective performance, none of them take into account the fact that pixels at the boundary and regions with a low number of class pixels capture more contextual feature information from other classes, leading to misclassification of pixels by unequal contextual feature information. In this paper, we propose a dual feature equalization network based on the hybrid architecture of Swin Transformer and Convolutional Neural Network, aiming to augment the pixel feature representations by image-level equalization feature information and class-level equalization feature information. Firstly, the image-level feature equalization module is designed to equalize the contextual information of pixels within the image. Secondly, we aggregate regions of the same class to equalize the pixel feature representations of the corresponding class by class-level feature equalization module. Finally, the pixel feature representations are enhanced by learning weights for image-level equalization feature information and class-level equalization feature information. In addition, Swin Transformer is utilized as both the encoder and decoder, thereby bolstering the ability of the model to capture long-range dependencies and spatial correlations. We conducted extensive experiments on Breast Ultrasound Images (BUSI), International Skin Imaging Collaboration (ISIC2017), Automated Cardiac Diagnosis Challenge (ACDC) and PH$^2$ datasets. The experimental results demonstrate that our method have achieved state-of-the-art performance. Our code is publicly available at https://github.com/JianJianYin/DFEN.

医学图像分割特征均衡Swin Transformer

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