arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2402.11273cs.CVcs.AI
收藏

基于利用强弱数据增强策略的交叉伪标签的半监督医学图像分割方法

Semi-supervised Medical Image Segmentation Method Based on Cross-pseudo Labeling Leveraging Strong and Weak Data Augmentation Strategies

Hangzhou Dianzi University(杭州电子科技大学);Zhejiang University of Finance and Economics(浙江财经大学);Shenzhen Institute of Advanced Technology, Chinese Academy of Sciences(中国科学院深圳先进技术研究院);Shenzhen Children's Hospital(深圳市儿童医院);Shenzhen Research Institute of Big Data(深圳市大数据研究院)

Yifei Chen, Chenyan Zhang, Yifan Ke, Yiyu Huang, Xuezhou Dai, Feiwei Qin, Yongquan Zhang, Xiaodong Zhang, Changmiao Wang

AI总结:

本文提出半监督模型DFCPS,结合Fixmatch概念与强弱数据增强策略,引入交叉伪监督机制,在Kvasir-SEG数据集上显著提升了医学图像分割的性能与泛化能力。

摘要:

由于生物医学图像采集过程困难、标注成本高、信噪比低且特征复杂,传统的监督学习方法在医学图像分割中一直面临诸多限制。本文提出了一种半监督模型DFCPS,创新性地引入了Fixmatch概念。通过对未标记数据采用不同的策略进行数据增强处理,显著提升了模型的性能和泛化能力。同时,该模型设计对伪标签的生成、过滤和细化过程给予了适当重视。引入了交叉伪监督的新概念,将一致性学习与自训练相结合。这使得模型能够从多个角度充分利用伪标签,从而增强训练的多样性。使用公开的Kvasir-SEG数据集将DFCPS模型与基线模型和先进模型进行了比较。在包含不同比例未标记数据的全部四个子集中,我们的模型始终表现出优越的性能。源代码可在https://github.com/JustlfC03/DFCPS获取。

英文摘要:

Traditional supervised learning methods have historically encountered certain constraints in medical image segmentation due to the challenging collection process, high labeling cost, low signal-to-noise ratio, and complex features characterizing biomedical images. This paper proposes a semi-supervised model, DFCPS, which innovatively incorporates the Fixmatch concept. This significantly enhances the model's performance and generalizability through data augmentation processing, employing varied strategies for unlabeled data. Concurrently, the model design gives appropriate emphasis to the generation, filtration, and refinement processes of pseudo-labels. The novel concept of cross-pseudo-supervision is introduced, integrating consistency learning with self-training. This enables the model to fully leverage pseudo-labels from multiple perspectives, thereby enhancing training diversity. The DFCPS model is compared with both baseline and advanced models using the publicly accessible Kvasir-SEG dataset. Across all four subdivisions containing different proportions of unlabeled data, our model consistently exhibits superior performance. Our source code is available at https://github.com/JustlfC03/DFCPS.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑