您好、欢迎来到现金彩票网!
当前位置:刘伯温论坛 > 图像分割 >

CNN 在基于弱监督学习的图像分割中的应用

发布时间:2019-06-12 04:24 来源:未知 编辑:admin

  最近基于深度学习的图像分割技术一般依赖于卷积神经网络 CNN 的训练,训练过程中需要非常大量的标记图像,即一般要求训练图像中都要有精确的分割结果。

  对于图像分割而言,要得到大量的完整标记过的图像非常困难,比如在 ImageNet 数据集上,有 1400 万张图有类别标记,有 50 万张图给出了 bounding box, 但是只有 4460 张图像有像素级别的分割结果。对训练图像中的每个像素做标记非常耗时,特别是对医学图像而言,完成对一个三维的 CT 或者 MRI 图像中各组织的标记过程需要数小时。

  如果学习算法能通过对一些初略标记过的数据集的学习就能完成好的分割结果,那么对训练数据的标记过程就很简单,这可以大大降低花在训练数据标记上的时间。这些初略标记可以是:

  3. 对图像中的物体区域做部分像素的标记,例如画一些线条、涂鸦等(scribbles)。

  香港中文大学的 Di Lin 提出了一个基于 Scribble 标记的弱监督学习方法。 Scribble 是一个很方便使用的标记方法,因此被用得比较广泛。如下图,只需要画五条线就能完成对一副图像的标记工作。

  ScribbleSup 分为两步,第一步将像素的类别信息从 scribbles 传播到其他未标记的像素,自动完成所有的训练图像的标记工作; 第二步使用这些标记图像训练 CNN。在第一步中,该方法先生成 super-pxels, 然后基于 graph cut 的方法对所有的 super-pixel 进行标记。

  在这个 graph 中,每个 super-pixel 是 graph 中的一个节点,相接壤的 super-pixel 之间有一条连接的边。这个能量函数中的一元项包括两种情况,一个是来自于 scribble 的,一个是来自 CNN 对该 super-pixel 预测的概率。整个最优化过程实际上是求 graph cut 能量函数和 CNN 参数联合最优值的过程:

  上式的最优化是通过交替求和的最优值来实现的。文章中发现通过三次迭代就能得到比较好的结果。

  其中是一个隐含的类别分布,是 CNN 预测的类别分布。目标函数是 KL-divergence 最小化。其中的线性限制条件来自于训练数据上的标记,例如一幅图像中前景类别像素个数期望值的上界或者下界(物体大小)、某个类别的像素个数在某图像中为 0,或者至少为 1 等。该目标函数可以转化为为一个 loss function,然后通过 SGD 进行训练。

  实验中发现单纯使用 Image tags 作为限制条件得到的分割结果还比较差,在 PASCAL VOC 2012 test 数据集上得到的 mIoU 为 35.6%,加上物体大小的限制条件后能达到 45.1%, 如果再使用 bounding box 做限制,可以达到 54%。FCN-8s 可以达到 62.2%,可见弱监督学习要取得好的结果还是比较难。

  对于 image-level 标记的数据,我们可以观测到图像的像素值和图像级别的标记, 但是不知道每个像素的标号, 因此把 当做隐变量。使用如下的概率图模式:

  使用 EM 算法估计和。E 步骤是固定求的期望值,M 步骤是固定使用 SGD 计算θ。

  对于给出 bounding box 标记的训练图像,该方法先使用 CRF 对该训练图像做自动分割,然后在分割的基础上做全监督学习。通过实验发现,单纯使用图像级别的标记得到的分割效果较差,但是使用 bounding box 的训练数据可以得到较好的结果,在 VOC2012 test 数据集上得到 mIoU 62.2%。另外如果使用少量的全标记图像和大量的弱标记图像进行结合,可以得到与全监督学习 (70.3%) 接近的分割结果 (69.0%)。

  Wisconsin-Madison 大学的 Jia Xu 提出了一个统一的框架来处理各种不同类型的弱标记:图像级别的标记、bounding box 和部分像素标记如 scribbles。该方法把所有的训练图像分成共计个 super-pixel,对每个 super-pixel 提取一个维特征向量。因为不知道每个 super-pixel 所属的类别,相当于无监督学习,因此该方法对所有的 super-pixel 做聚类,使用的是最大间隔聚类方法 (max-margin clustering, MMC), 该过程的最优化目标函数是:

  其中是一个特征矩阵,每列代表了对于的类别的聚类特征。是将第个 super-pixel 划分到第类的代价。在这个目标函数的基础上,根据不同的弱标记方式,可以给出不同的限制条件,因此该方法就是在相应的限制条件下求最大间隔聚类。

  小结:在弱标记的数据集上训练图像分割算法可以减少对大量全标记数据的依赖,在大多数应用中会更加贴合实际情况。弱标记可以是图像级别的标记、边框和部分像素的标记等。训练的方法一般看做是限制条件下的最优化方法。另外 EM 算法可以用于 CNN 参数和像素类别的联合求优。

http://airgomusic.com/tuxiangfenge/239.html
锟斤拷锟斤拷锟斤拷QQ微锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷锟斤拷微锟斤拷
关于我们|联系我们|版权声明|网站地图|
Copyright © 2002-2019 现金彩票 版权所有