arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2409.05024cs.CV
收藏

基于噪声标签的医学图像分割深度自清洗方法

Deep Self-Cleansing for Medical Image Segmentation with Noisy Labels

College of Computer Science and Technology, Zhejiang University(浙江大学计算机科学与技术学院);School of Biomedical Engineering, Division of Life Sciences and Medicine, University of Science and Technology of China(中国科学技术大学生命科学与医学部生物医学工程学院);Suzhou Institute for Advanced Research, University of Science and Technology of China(中国科学技术大学苏州研究院);Army Medical University (Third Military Medical University)(中国人民解放军陆军军医大学(第三军医大学));The First Affiliated Hospital, College of Medicine, Zhejiang University(浙江大学医学院附属第一医院);College of Information Science and Engineering, Ritsumeikan University(立命馆大学信息科学与工程学院);University of Science and Technology of China(中国科学技术大学)

Jiahua Dong, Yue Zhang, Qiuli Wang, Ruofeng Tong, Shihong Ying, Shaolin Gong, Xuanpu Zhang, Lanfen Lin, Yen-Wei Chen, S. Kevin Zhou

AI总结:

提出一种深度自清洗分割框架,通过高斯混合模型过滤噪声标签并生成伪标签,有效抑制噪声干扰,提升医学图像分割性能。

摘要:

医学图像分割在医学成像领域至关重要,有助于疾病诊断和手术规划。大多数现有的分割方法依赖于有监督的深度学习,其中干净且精确的标签对于监督至关重要,并显著影响模型的性能。然而,人工描绘的标签通常包含噪声,例如标签缺失和不准确的边界描绘,这会阻碍网络正确建模目标特征。在本文中,我们提出了一种深度自清洗分割框架,能够在训练阶段保留干净标签的同时清洗噪声标签。为此,我们设计了一个基于高斯混合模型的标签过滤模块,用于区分噪声标签和干净标签。此外,我们开发了一个标签清洗模块,为识别出的噪声样本生成伪低噪声标签。随后,保留的干净标签和伪标签被联合用于监督网络。在临床肝肿瘤数据集和公开心脏诊断数据集上验证表明,我们的方法能有效抑制噪声标签的干扰,并取得显著的分割性能。

英文摘要:

Medical image segmentation is crucial in the field of medical imaging, aiding in disease diagnosis and surgical planning. Most established segmentation methods rely on supervised deep learning, in which clean and precise labels are essential for supervision and significantly impact the performance of models. However, manually delineated labels often contain noise, such as missing labels and inaccurate boundary delineation, which can hinder networks from correctly modeling target characteristics. In this paper, we propose a deep self-cleansing segmentation framework that can preserve clean labels while cleansing noisy ones in the training phase. To achieve this, we devise a gaussian mixture model-based label filtering module that distinguishes noisy labels from clean labels. Additionally, we develop a label cleansing module to generate pseudo low-noise labels for identified noisy samples. The preserved clean labels and pseudo-labels are then used jointly to supervise the network. Validated on a clinical liver tumor dataset and a public cardiac diagnosis dataset, our method can effectively suppress the interference from noisy labels and achieve prominent segmentation performance.

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

↑