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

使用元数据进行线性条件化图像分割的优势

Benefits of Linear Conditioning with Metadata for Image Segmentation

Polytechnique Montreal(蒙特利尔理工学院);Mila, Quebec AI Institute(魁北克人工智能研究所米拉);Beijing Tiantan Hospital, Capital Medical University(首都医科大学附属北京天坛医院);Stanford University Center for Artificial Intelligence in Medicine & Imaging(斯坦福大学医学与影像人工智能中心);University of Montreal(蒙特利尔大学)

Andreanne Lemay, Charley Gros, Olivier Vincent, Yaou Liu, Joseph Paul Cohen, Julien Cohen-Adad

摘要:

医学图像通常伴有描述图像(供应商、采集参数)和患者(疾病类型或严重程度、人口统计学、基因组学)的元数据。这种元数据通常被图像分割方法忽略。在这项工作中,我们适应了一个线性条件的方法,称为电影(智能线性调制)的图像分割任务。此FiLM适配支持将元数据集成到分段模型中,以获得更好的性能。当将肿瘤类型与FiLM结合时,我们观察到脊髓肿瘤分割的平均Dice评分增加了5.1%。元数据通过应用于特征图的低成本仿射变换来调节分割过程,特征图可以包含在任何神经网络的架构中。此外,我们还评估了分割薄膜层的相关性,以应对医学成像中的常见挑战:缺少分割的多类训练,模型适应多个任务,以及有限或不平衡数量的注释数据的训练。我们的研究结果证明了FiLM在分割方面的以下优势:与单任务U-Net相比,FiLMed U-Net对缺失标签具有鲁棒性,并且在标签较少的情况下达到了更高的Dice分数(高达16.7%)。该代码是开源的,可在www.ivadomed.org上获取。

英文摘要:

Medical images are often accompanied by metadata describing the image (vendor, acquisition parameters) and the patient (disease type or severity, demographics, genomics). This metadata is usually disregarded by image segmentation methods. In this work, we adapt a linear conditioning method called FiLM (Feature-wise Linear Modulation) for image segmentation tasks. This FiLM adaptation enables integrating metadata into segmentation models for better performance. We observed an average Dice score increase of 5.1% on spinal cord tumor segmentation when incorporating the tumor type with FiLM. The metadata modulates the segmentation process through low-cost affine transformations applied on feature maps which can be included in any neural network's architecture. Additionally, we assess the relevance of segmentation FiLM layers for tackling common challenges in medical imaging: multi-class training with missing segmentations, model adaptation to multiple tasks, and training with a limited or unbalanced number of annotated data. Our results demonstrated the following benefits of FiLM for segmentation: FiLMed U-Net was robust to missing labels and reached higher Dice scores with few labels (up to 16.7%) compared to single-task U-Net. The code is open-source and available at www.ivadomed.org.

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

↑