BoxSup: Exploiting Bounding Boxes to Supervise Convolutional Networks for Semantic Segmentation读书笔记

  在人工智能领域有一句很流行的话:“有多少人工,就有多少智能”。这确实是一个很尴尬的问题,计算机的理解能力是建立在大量的人工标注数据之上的。我们知道,数据标注需要很高的成本,特别是像素级的图像,而且在复杂场景之下,也非常容易出错。
  俗话说,人民的智慧是无穷的。为了改善或者解决这个问题,弱监督和无监督学习得到了用武之地。
  好了,说到这篇论文,本文讨论的是弱监督学习的问题,主要内容是利用bounding box和少量的mask来进行图像的语义分割。
  看一下论文的具体内容:

3. Baseline

  BoxSup方法可以应用在所有的基于cnn的mask监督的语义分割网络中。作者采用的是FCN网络,然后使用条件随机场对结果进行优化。(ps:其实,就是FCN那一套,这篇论文也是在FCN的基础上做了一些改进)
  而且,在FCN中,目标函数是这样的:



  这篇论文既然是在FCN上进行的改进,那么目标函数肯定是要变一变的,具体怎么变,后面具体说。

4. Approach
4.1. Unsupervised Segmentation for Supervised Training

  首先使用无监督方法产生一些candidate segments,然后利用这些candidate segments去更新网络参数。接下来,更新之后的网络又会产生一些更好的candidate。这个过程一直迭代进行下去,下面来张图来描述这个过程:



  是不是很形象?这里有一点要注意的是,产生region proposal的步骤只出现在训练的过程中,在测试的时候,直接把网络的前向传播跑一遍就完事了。

4.2. Formulation

  这部分只有两个公式,就是把FCN的目标函数按照这个思路稍微改造一下就可以了。


  稍微解释一下这个公式,首先给产生的所有candidate segments一个类别标签,其中lB表示:bouding box的类别标签,lS表示:semantic label,如果两者相等,那么
\delta
为1,否则为0。IoU(B, S)表示:candidate segments和bouding box的IoU。N表示:candidate segments的数量。
  那么,这个公式的含义就很好理解了,如果candidate segments和bouding box的重合度越高,那么,\varepsilon0的值就越小。
  下面来看一下第二个公式:

  是不是和FCN的目标函数很像?其实就是把标注的mask换成了我们自己生成的candidate segments。
  最终的目标函数就变成了这样滴:

  其中,
\lambda
的值为:3。其实,从公式也可以看出作者的思路,相加的这两个部分,分别代表了我们一直在迭代的两个过程,稍后,我们会详细说明。

4.3. Training Algorithm

  每一次迭代的过程分为两个阶段,第一个阶段:固定candidate segments,然后更新网络参数;第二个阶段:固定网络参数,然后产生新的candidate segments。注意,两个阶段都要把所有的图片过一遍。也就是说,每迭代一次,训练集的图片要使用两遍。
  在第二个阶段还有一个问题需要解决,网络每次可能都会产生很多的candidate segments,怎么给它们打标签呢。作者给出了一个贪婪算法。对于每一个bounding box,只选择一个和它的IoU最大的那个segments作为对应的前景,其他没被选中的全部标记为背景。
  但是,这种方法很可能导致,每一代选择的segments都是同一个,有可能会陷入局部最优。所以,这里改进的方法是:每次找出前5个IoU最大的segments,然后从里面随机选择一个。(作者在后面的验证中,也证明了这种方法比每次只选一个IoU最大的效果要好,其实,这也算一种数据增强方法,可以增加样本的差异性。)
  到目前为止,上面提到的方法是只使用bouding box来训练语义分割,如果你有mask的分割数据的话,也可以进行混合训练,方法很简单,直接把公式中的candidate segments全部改成ground-truth masks就可以了。

Comparisons of Supervision Strategies

  在这一部分,作者比较了只使用bounding box信息、只使用mask信息、同时使用bounding box和mask信息,三种方法的效果,得出如下结论:



  从结果中可以看出,只使用bounding box的话,平均IoU和其他两种方法稍微逊色。只使用mask信息和semi方法(同时使用bounding box和mask信息)得到的结果几乎差不多。但是,semi方法的优势在于bounding box数据和mask数据的比例为1:9。也就是说:这里只使用了少量的mask数据,就达到了和全部都使用mask数据差不多的效果,节约了很多成本。

Error Analysis

  从作者的分析中,得到了一个很有意思的结论:将bounding box的信息应用在网络中,对于物体边缘的识别效果提升有限,而对于物体内部信息的识别效果可以得到很大程度的提高。
  也就是说,应用bounding box的信息之后,网络在物体内部(特别是物体中心)的特征提取能力得到了很大的提高,最后反而会使得物体的分类精度得到提高。当然,这也会提高网络segmatic segmentation的表现。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 214,128评论 6 493
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 91,316评论 3 388
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 159,737评论 0 349
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 57,283评论 1 287
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 66,384评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,458评论 1 292
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,467评论 3 412
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,251评论 0 269
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,688评论 1 306
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,980评论 2 328
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,155评论 1 342
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,818评论 4 337
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,492评论 3 322
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,142评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,382评论 1 267
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,020评论 2 365
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,044评论 2 352