arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2306.11490cs.CV

UM-CAM:用于弱监督胎儿脑部分割的不确定性加权多分辨率类激活图

UM-CAM: Uncertainty-weighted Multi-resolution Class Activation Maps for Weakly-supervised Fetal Brain Segmentation

University of Electronic Science and Technology of China(电子科技大学);Shanghai Artificial Intelligence Laboratory(上海人工智能实验室)

Jia Fu, Tao Lu, Shaoting Zhang, Guotai Wang

AI总结:

本文提出UM-CAM方法,利用不确定性加权多分辨率类激活图和测地距离种子扩展,解决弱监督胎儿脑部分割中激活区域不完整和边界模糊问题,实验证明其优于现有方法。

摘要:

从磁共振图像(MRI)中准确分割胎儿大脑对于产前胎儿发育评估至关重要。尽管深度学习已展现出完成此任务的潜力,但它需要大量难以收集的精细标注数据集。为解决此问题,基于图像级标签的弱监督分割方法受到关注,这些方法通常基于由图像标签训练的分类网络生成的类激活图。然而,由于低分辨率定位缺乏详细边界线索,这些方法大多存在激活区域不完整的问题。为此,我们提出了一种基于语义特征和上下文信息探索的、使用图像级标签的新型弱监督方法。我们首先提出了一种不确定性加权多分辨率类激活图(UM-CAM)来生成高质量的像素级监督。然后,我们设计了一种基于测地距离的种子扩展(GSE)方法,以提供上下文信息来修正UM-CAM的模糊边界。在胎儿大脑数据集上进行的大量实验表明,与现有CAM变体相比,我们的UM-CAM能提供更准确的激活区域且假阳性区域更少,并且我们提出的方法优于使用图像级标签的最先进的弱监督方法。

英文摘要:

Accurate segmentation of the fetal brain from Magnetic Resonance Image (MRI) is important for prenatal assessment of fetal development. Although deep learning has shown the potential to achieve this task, it requires a large fine annotated dataset that is difficult to collect. To address this issue, weakly-supervised segmentation methods with image-level labels have gained attention, which are commonly based on class activation maps from a classification network trained with image tags. However, most of these methods suffer from incomplete activation regions, due to the low-resolution localization without detailed boundary cues. To this end, we propose a novel weakly-supervised method with image-level labels based on semantic features and context information exploration. We first propose an Uncertainty-weighted Multi-resolution Class Activation Map (UM-CAM) to generate high-quality pixel-level supervision. Then, we design a Geodesic distance-based Seed Expansion (GSE) method to provide context information for rectifying the ambiguous boundaries of UM-CAM. Extensive experiments on a fetal brain dataset show that our UM-CAM can provide more accurate activation regions with fewer false positive regions than existing CAM variants, and our proposed method outperforms state-of-the-art weakly-supervised methods with image-level labels.

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

↑