arXivDaily arXiv每日学术速递 周一至周五更新
​通知:联邦节假日调休:哥伦布日 / 原住民日(Columbus Day / Indigenous Peoples' Day),今日(13日)无更新。
arXiv 2310.18142cs.CV

半监督全景叙事定位

Semi-Supervised Panoptic Narrative Grounding

Key Laboratory of Multimedia Trusted Perception and Efficient Computing, Ministry of Education of China, Xiamen University(厦门大学多媒体可信感知与高效计算教育部重点实验室)

Danni Yang, Jiayi Ji, Xiaoshuai Sun, Haowei Wang, Yinan Li, Yiwei Ma, Rongrong Ji

AI总结:

本文提出半监督全景叙事定位方案 SS-PNG-NW+,通过 Burn-In、数据增强和基于质量驱动的损失调整(QLA)利用少量标注数据,在仅 30% 和 50% 监督数据下超越全监督模型。

摘要:

尽管取得了相当大的进展,全景叙事定位(Panoptic Narrative Grounding,PNG)的发展仍然受到高昂标注成本的制约。在本文中,我们提出了一种新颖的半监督全景叙事定位(Semi-Supervised Panoptic Narrative Grounding,SS-PNG)学习方案,利用较少的已标注图像-文本对和较多的未标注图像-文本对来实现具有竞争力的性能。与视觉分割任务不同,PNG 涉及一个像素属于多个开放式名词。因此,现有的基于多类别的半监督分割框架无法直接应用于该任务。为了解决这一挑战,我们首先开发了一种专门针对 SS-PNG 设置的新型 SS-PNG 网络(SS-PNG-NW)。我们深入研究了 Burn-In 和数据增强等策略,以确定 SS-PNG-NW 的最佳通用配置。此外,为了解决伪标签质量不平衡的问题,我们提出了一种基于质量驱动的损失调整(Quality-Based Loss Adjustment,QLA)方法来调整半监督目标,从而得到增强的 SS-PNG-NW+。采用我们提出的 QLA,我们分别在像素级和掩码级改进了 BCE Loss 和 Dice loss。我们在 PNG 数据集上进行了大量实验,我们的 SS-PNG-NW+ 在所有数据比例下都表现出与全监督模型相当的有希望的结果。值得注意的是,我们的 SS-PNG-NW+ 在仅使用 30% 和 50% 监督数据的情况下优于全监督模型,分别超过其性能 0.8% 和 1.1%。这凸显了我们提出的 SS-PNG-NW+ 在克服有限标注带来的挑战和增强 PNG 任务适用性方面的有效性。源代码可在 https://github.com/nini0919/SSPNG 获取。

英文摘要:

Despite considerable progress, the advancement of Panoptic Narrative Grounding (PNG) remains hindered by costly annotations. In this paper, we introduce a novel Semi-Supervised Panoptic Narrative Grounding (SS-PNG) learning scheme, capitalizing on a smaller set of labeled image-text pairs and a larger set of unlabeled pairs to achieve competitive performance. Unlike visual segmentation tasks, PNG involves one pixel belonging to multiple open-ended nouns. As a result, existing multi-class based semi-supervised segmentation frameworks cannot be directly applied to this task. To address this challenge, we first develop a novel SS-PNG Network (SS-PNG-NW) tailored to the SS-PNG setting. We thoroughly investigate strategies such as Burn-In and data augmentation to determine the optimal generic configuration for the SS-PNG-NW. Additionally, to tackle the issue of imbalanced pseudo-label quality, we propose a Quality-Based Loss Adjustment (QLA) approach to adjust the semi-supervised objective, resulting in an enhanced SS-PNG-NW+. Employing our proposed QLA, we improve BCE Loss and Dice loss at pixel and mask levels, respectively. We conduct extensive experiments on PNG datasets, with our SS-PNG-NW+ demonstrating promising results comparable to fully-supervised models across all data ratios. Remarkably, our SS-PNG-NW+ outperforms fully-supervised models with only 30% and 50% supervision data, exceeding their performance by 0.8% and 1.1% respectively. This highlights the effectiveness of our proposed SS-PNG-NW+ in overcoming the challenges posed by limited annotations and enhancing the applicability of PNG tasks. The source code is available at https://github.com/nini0919/SSPNG.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑