3 个月前

循环引导的弱监督联合检测与分割

循环引导的弱监督联合检测与分割

摘要

弱监督学习因其在需要图像内标注的任务(如目标检测与语义分割)中显著降低标注成本,近年来受到越来越多研究关注。现有弱监督目标检测与语义分割方法通常采用迭代式标签挖掘与模型训练的流程。然而,这种自我强化的训练机制容易使两类任务陷入局部最优解。本文首次提出将弱监督目标检测与语义分割任务通过多任务学习框架进行联合建模,利用两者各自在学习过程中的失败模式相互补充,实现跨任务的协同增强,从而帮助两类任务有效跳出各自的局部最优。为此,我们提出一种高效且有效的框架——弱监督联合检测与分割(Weakly Supervised Joint Detection and Segmentation, WS-JDS)。该框架包含两个并行分支,分别对应目标检测与语义分割任务,共享同一骨干网络。在训练阶段,WS-JDS采用统一的循环训练范式,并引入特定设计的损失函数,使两个分支在训练过程中相互促进、协同优化。在广泛使用的Pascal VOC与COCO基准数据集上进行了大量实验,结果表明,所提出的模型在性能上达到了与当前最先进方法相媲美的水平。

基准测试

基准方法指标
image-level-supervised-instance-segmentation-2WS-JDS
AP: 6.1
AP@50: 11.7
AP@75: 5.5
weakly-supervised-object-detection-on-pascalWS-JDS FRCNN
MAP: 46.1
weakly-supervised-object-detection-on-pascal-1WS-JDS FRCNN
MAP: 52.5

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
循环引导的弱监督联合检测与分割 | 论文 | HyperAI超神经