arXiv:2410.03161cs.AI2024-10被引 7

通过自适应掩码提升视觉定位在少样本场景下的表现。

Adaptive Masking Enhances Visual Grounding

  • 基于视觉主干特征图的显著区域自适应掩码。
  • 在COCO和ODinW上实现零样本与少样本任务性能提升。
  • 无需扩大数据集即可增强模型泛化能力,适合低资源场景。

近年来,零样本与少样本视觉定位受到广泛关注,主要得益于大规模视觉语言预训练在LAION-5B和DataComp-1B等海量数据集上的成功。然而,这些数据集的持续扩张带来了数据可用性与计算开销的挑战,制约了低样本学习能力的发展。本文提出IMAGE( Interpretative MAsking with Gaussian radiation modEling),旨在不增加数据集规模的前提下,提升词汇定位在低样本学习中的表现。受认知科学及掩码自编码器(MAE)启发,该方法对视觉主干生成的特征图中显著区域进行自适应掩码,使模型通过重建被遮蔽信息学习鲁棒且泛化的表示,从而有效关注局部与全局特征。我们在COCO和ODinW等基准数据集上评估了该方法,结果表明其在零样本与少样本任务中均优于基线模型,显著提升了泛化能力与低样本表现。这说明通过注意力机制与高斯建模进行自适应特征操作,可作为替代不断扩增数据集的有前景方案。代码已公开于https://github.com/git-lenny/IMAGE。

原文摘要 · Abstract (English)

In recent years, zero-shot and few-shot learning in visual grounding have garnered considerable attention, largely due to the success of large-scale vision-language pre-training on expansive datasets such as LAION-5B and DataComp-1B. However, the continuous expansion of these datasets presents significant challenges, particularly with respect to data availability and computational overhead, thus creating a bottleneck in the advancement of low-shot learning capabilities. In this paper, we propose IMAGE, Interpretative MAsking with Gaussian radiation modEling, aimed at enhancing vocabulary grounding in low-shot learning scenarios without necessitating an increase in dataset size. Drawing inspiration from cognitive science and the recent success of masked autoencoders (MAE), our method leverages adaptive masking on salient regions of the feature maps generated by the vision backbone. This enables the model to learn robust, generalized representations through the reconstruction of occluded information, thereby facilitating effective attention to both local and global features. We evaluate the efficacy of our approach on benchmark datasets, including COCO and ODinW, demonstrating its superior performance in zero-shot and few-shot tasks. Experimental results consistently show that IMAGE outperforms baseline models, achieving enhanced generalization and improved performance in low-shot scenarios. These findings highlight the potential of adaptive feature manipulation through attention mechanisms and Gaussian modeling as a promising alternative to approaches that rely on the continual scaling of dataset sizes for the advancement of zero-shot and few-shot learning. Our code is publicly available at https://github.com/git-lenny/IMAGE.

视觉定位少样本学习自适应掩码扩散模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。