4 个月前

一种基于混合标签的弱监督学习框架用于显著目标检测

一种基于混合标签的弱监督学习框架用于显著目标检测

摘要

全监督显著物体检测(SOD)方法已经取得了很大进展,但这些方法通常依赖于大量的像素级注释,这既耗时又费力。在本文中,我们专注于一种新的混合标签下的弱监督SOD任务,其中监督标签包括大量由传统无监督方法生成的粗略标签和少量真实标签。为了解决该任务中的标签噪声和数量不平衡问题,我们设计了一个包含三种复杂训练策略的新管道框架。在模型框架方面,我们将任务分解为标签细化子任务和显著物体检测子任务,这两个子任务相互协作并交替训练。具体而言,R-Net被设计为一个双流编码器-解码器模型,并配备了引导和聚合机制(Blender with Guidance and Aggregation Mechanisms, BGA),旨在修正粗略标签以生成更可靠的伪标签;而S-Net则是一个可替换的SOD网络,由当前R-Net生成的伪标签进行监督。需要注意的是,在测试阶段我们只需使用训练好的S-Net。此外,为了保证网络训练的有效性和效率,我们设计了三种训练策略,包括交替迭代机制、分组增量机制和可信度验证机制。在五个SOD基准数据集上的实验表明,我们的方法在定性和定量方面均达到了与弱监督/无监督方法相当的性能。

基准测试

基准方法指标
salient-object-detection-on-duts-teHybridSOD
MAE: 0.05
S-Measure: 0.837
salient-object-detection-on-ecssdHybridSOD
F-Score: 0.899
MAE: 0.051
S-Measure: 0.886
salient-object-detection-on-hku-isHybridSOD
F-Score: 0.892
MAE: 0.038
S-Measure: 0.887
salient-object-detection-on-pascal-sHybridSOD
F-Score: 0.827
MAE: 0.076
S-Measure: 0.828

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
一种基于混合标签的弱监督学习框架用于显著目标检测 | 论文 | HyperAI超神经