3 个月前

弱监督语义分割中的信息瓶颈缩减

弱监督语义分割中的信息瓶颈缩减

摘要

弱监督语义分割能够从类别标签中实现像素级定位,然而,基于此类标签训练的分类器往往仅关注目标物体中的局部判别性区域。我们基于信息瓶颈原理对这一现象进行解释:深度神经网络的最终层在使用Sigmoid或Softmax激活函数激活后,会产生信息瓶颈效应,导致仅有部分与任务相关的信息被传递至输出层。我们首先通过一个模拟的简化实验验证了该观点,随后提出一种方法,通过移除最后一层的激活函数来缓解信息瓶颈问题。此外,我们引入一种新型池化策略,进一步促进非判别性区域信息向分类任务的传递。实验结果表明,这一简单改进显著提升了在PASCAL VOC 2012和MS COCO 2014数据集上的定位图质量,实现了弱监督语义分割的新最先进性能。代码已开源,地址为:https://github.com/jbeomlee93/RIB。

代码仓库

jbeomlee93/rib
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
weakly-supervised-semantic-segmentation-onRIB (DeepLabV2-ResNet101)
Mean IoU: 68.3
weakly-supervised-semantic-segmentation-onRIB+Sal (DeepLabV2-ResNet101)
Mean IoU: 70.2
weakly-supervised-semantic-segmentation-on-1RIB+Sal (DeepLabV2-ResNet101)
Mean IoU: 70.0
weakly-supervised-semantic-segmentation-on-1RIB (DeepLabV2-ResNet101)
Mean IoU: 68.6
weakly-supervised-semantic-segmentation-on-4RIB (DeepLabV2-ResNet101, No Saliency)
mIoU: 43.8

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
弱监督语义分割中的信息瓶颈缩减 | 论文 | HyperAI超神经