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

基于极值点的置信度图用于深度神经网络的类无关分割

Extreme Points Derived Confidence Map as a Cue For Class-Agnostic Segmentation Using Deep Neural Network

Inception Institute of Artificial Intelligence(Inception人工智能研究院)

Shadab Khan, Ahmed H. Shahin, Javier Villafruela, Jianbing Shen, Ling Shao

摘要:

为了自动分割感兴趣的解剖结构的过程,我们可以从先前注释的数据中学习模型。基于学习的方法使用注释来训练模型,该模型试图在新数据集上模仿专家标记。虽然使用这些方法已经取得了巨大的进步,但医学图像的标记仍然是一项耗时且昂贵的任务。在本文中,我们评估的效用极值点在学习分割。具体来说,我们提出了一种新的方法来计算一个置信图的极值点,定量编码的先验来自极值点。我们使用置信图作为线索来训练一个基于ResNet-101和PSP模块的深度神经网络,以开发一个与类别无关的分割模型,该模型优于采用极值点作为线索的最先进方法。此外,我们通过使用我们的模型为监督学习(U-Net)生成训练数据来评估一个现实的用例,并观察到U-Net在使用生成的数据或地面真实数据进行训练时会执行训练。这些发现表明,使用线索训练的模型可用于生成可靠的训练数据。

英文摘要:

To automate the process of segmenting an anatomy of interest, we can learn a model from previously annotated data. The learning-based approach uses annotations to train a model that tries to emulate the expert labeling on a new data set. While tremendous progress has been made using such approaches, labeling of medical images remains a time-consuming and expensive task. In this paper, we evaluate the utility of extreme points in learning to segment. Specifically, we propose a novel approach to compute a confidence map from extreme points that quantitatively encodes the priors derived from extreme points. We use the confidence map as a cue to train a deep neural network based on ResNet-101 and PSP module to develop a class-agnostic segmentation model that outperforms state-of-the-art method that employs extreme points as a cue. Further, we evaluate a realistic use-case by using our model to generate training data for supervised learning (U-Net) and observed that U-Net performs comparably when trained with either the generated data or the ground truth data. These findings suggest that models trained using cues can be used to generate reliable training data.

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

↑