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

基于掩码边界框重建的视觉关系检测自监督学习

Self-Supervised Learning for Visual Relationship Detection through Masked Bounding Box Reconstruction

Deeplab;National Technical University of Athens(雅典国立技术大学);SnT, University of Luxembourg(卢森堡大学SnT);ETH, Zürich(苏黎世联邦理工学院)

Zacharias Anastasakis, Dimitrios Mallis, Markos Diomataris, George Alexandridis, Stefanos Kollias, Vassilis Pitsikalis

AI总结:

该研究提出掩码边界框重建(MBBR)这一自监督方法,通过对象级掩码建模学习捕捉场景对象交互的上下文感知表征,仅用少量标注样本就在VRD的PredDet评估中超越现有最优方法。

摘要:

我们提出了一种用于表征学习的新型自监督方法,尤其适用于视觉关系检测(Visual Relationship Detection,VRD)任务。受掩码图像建模(Masked Image Modeling,MIM)有效性的启发,我们提出了掩码边界框重建(Masked Bounding Box Reconstruction,MBBR)——一种MIM变体,该方法会对场景内一定比例的实体/对象进行掩码,随后基于未掩码的对象对其进行重建。其核心思想是,通过对象级掩码建模,网络能够学习到捕捉场景内对象交互的上下文感知表征,因此对视觉对象关系具有很强的预测能力。我们在少样本设置下从定性和定量两方面对学习到的表征进行了广泛评估,证明了MBBR在学习鲁棒视觉表征方面的有效性,且该表征尤其适配VRD任务。所提方法仅使用少量标注样本,就在谓词检测(Predicate Detection,PredDet)评估设置下超越了当前最优的VRD方法。我们的代码已公开发布于https://github.com/deeplab-ai/SelfSupervisedVRD。

英文摘要:

We present a novel self-supervised approach for representation learning, particularly for the task of Visual Relationship Detection (VRD). Motivated by the effectiveness of Masked Image Modeling (MIM), we propose Masked Bounding Box Reconstruction (MBBR), a variation of MIM where a percentage of the entities/objects within a scene are masked and subsequently reconstructed based on the unmasked objects. The core idea is that, through object-level masked modeling, the network learns context-aware representations that capture the interaction of objects within a scene and thus are highly predictive of visual object relationships. We extensively evaluate learned representations, both qualitatively and quantitatively, in a few-shot setting and demonstrate the efficacy of MBBR for learning robust visual representations, particularly tailored for VRD. The proposed method is able to surpass state-of-the-art VRD methods on the Predicate Detection (PredDet) evaluation setting, using only a few annotated samples. We make our code available at https://github.com/deeplab-ai/SelfSupervisedVRD.

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

↑