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

利用从大型基础模型蒸馏的知识提升显著目标检测

Boosting Salient Object Detection with Knowledge Distillated from Large Foundation Models

Miaoyang He, Shuyong Gao, Tsui Qin Mok, Weifeng Ge, Wengqiang Zhang

AI总结:

本文利用大型基础模型通过文本提示和弱监督微调生成高质量伪标签,构建更大规模的 BDS-TR 数据集,并提出动态上采样边缘解码器,显著提升 SOD 性能。

摘要:

显著目标检测(Salient Object Detection,SOD)旨在识别并分割场景中的显著区域。传统模型依赖具有精确像素级准确性的人工标注伪标签,这一过程十分耗时。为应对这些挑战,我们开发了一种利用大型基础模型的低成本、高精度标注方法。具体而言,我们采用弱监督方法,通过文本提示引导大型模型生成伪标签。由于大型模型无法有效关注图像中的显著区域,我们人工标注了一部分文本以微调该模型。基于这种能够精确、快速生成伪标签的方法,我们引入了一个新数据集 BDS-TR。与此前的 DUTS-TR 数据集相比,BDS-TR 在规模上更为突出,并涵盖更广泛的类别和场景。这一扩展将增强我们的模型在更多场景中的适用性,并为未来的 SOD 研究提供更全面的基础数据集。此外,我们提出了一种基于动态上采样的边缘解码器,它在逐步恢复图像特征分辨率的同时关注目标边缘。在五个基准数据集上的综合实验表明,我们的方法显著优于最先进方法,并且超过了若干现有的全监督 SOD 方法。代码和结果将会公开。

英文摘要:

Salient Object Detection (SOD) aims to identify and segment prominent regions within a scene. Traditional models rely on manually annotated pseudo labels with precise pixel-level accuracy, which is time-consuming. We developed a low-cost, high-precision annotation method by leveraging large foundation models to address the challenges. Specifically, we use a weakly supervised approach to guide large models in generating pseudo-labels through textual prompts. Since large models do not effectively focus on the salient regions of images, we manually annotate a subset of text to fine-tune the model. Based on this approach, which enables precise and rapid generation of pseudo-labels, we introduce a new dataset, BDS-TR. Compared to the previous DUTS-TR dataset, BDS-TR is more prominent in scale and encompasses a wider variety of categories and scenes. This expansion will enhance our model's applicability across a broader range of scenarios and provide a more comprehensive foundational dataset for future SOD research. Additionally, we present an edge decoder based on dynamic upsampling, which focuses on object edges while gradually recovering image feature resolution. Comprehensive experiments on five benchmark datasets demonstrate that our method significantly outperforms state-of-the-art approaches and also surpasses several existing fully-supervised SOD methods. The code and results will be made available.

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

↑