arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2406.01451cs.CVcs.MM
收藏

SAM作为向导:掌握半监督指代表达分割中的伪标签精炼

SAM as the Guide: Mastering Pseudo-Label Refinement in Semi-Supervised Referring Expression Segmentation

School of Informatics, Xiamen University(厦门大学信息学院);Youtu Lab, Tencent(腾讯腾讯云实验室)

Danni Yang, Jiayi Ji, Yiwei Ma, Tianyu Guo, Haowei Wang, Xiaoshuai Sun, Rongrong Ji

AI总结:

提出半监督框架SemiRES,利用SAM精炼边界伪标签,结合IOM、CPI和PWA策略提升学生模型训练精度,在1%标注数据下较监督基线提升18.64%。

摘要:

本文介绍了SemiRES,这是一个半监督框架,能够有效利用标注数据和未标注数据的组合来执行RES。将半监督技术应用于RES的一个重大障碍是存在大量噪声伪标签,尤其是在物体边界处。SemiRES集成了以精确边界划分而闻名的Segment Anything Model (SAM),以提高这些伪标签的准确性。在SemiRES中,我们提供了两种可选的匹配策略:基于IoU的最优匹配(IOM)和复合部件整合(CPI)。这些策略旨在从SAM的输出中提取最准确的掩码,从而以更高的精度指导学生模型的训练。在无法从可用候选中匹配到精确掩码的情况下,我们开发了像素级调整(PWA)策略,直接通过伪标签指导学生模型的训练。在三个RES基准——RefCOCO、RefCOCO+和G-Ref上的大量实验表明,与全监督方法相比,其性能更优。值得注意的是,仅使用1%的标注数据,我们的SemiRES就大幅优于监督基线,例如在RefCOCO验证集上获得了+18.64%的增益。项目代码可在https://github.com/nini0919/SemiRES获取。

英文摘要:

In this paper, we introduce SemiRES, a semi-supervised framework that effectively leverages a combination of labeled and unlabeled data to perform RES. A significant hurdle in applying semi-supervised techniques to RES is the prevalence of noisy pseudo-labels, particularly at the boundaries of objects. SemiRES incorporates the Segment Anything Model (SAM), renowned for its precise boundary demarcation, to improve the accuracy of these pseudo-labels. Within SemiRES, we offer two alternative matching strategies: IoU-based Optimal Matching (IOM) and Composite Parts Integration (CPI). These strategies are designed to extract the most accurate masks from SAM's output, thus guiding the training of the student model with enhanced precision. In instances where a precise mask cannot be matched from the available candidates, we develop the Pixel-Wise Adjustment (PWA) strategy, guiding the student model's training directly by the pseudo-labels. Extensive experiments on three RES benchmarks--RefCOCO, RefCOCO+, and G-Ref reveal its superior performance compared to fully supervised methods. Remarkably, with only 1% labeled data, our SemiRES outperforms the supervised baseline by a large margin, e.g. +18.64% gains on RefCOCO val set. The project code is available at \url{https://github.com/nini0919/SemiRES}.

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

↑