arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2302.01403cs.CV

面向场景图生成的自监督关系对齐

Self-Supervised Relation Alignment for Scene Graph Generation

University of British Columbia(不列颠哥伦比亚大学);Vector Institute for AI(Vector人工智能研究所);Canada CIFAR AI Chair(加拿大CIFAR人工智能讲席)

Bicheng Xu, Renjie Liao, Leonid Sigal

AI总结:

针对场景图生成任务,提出一种可与任意现有框架结合的自监督关系对齐正则化方法,通过带掩码的辅助分支蒸馏对齐监督分支预测,在SGTR和Neural Motifs架构上均实现性能显著提升。

摘要:

场景图生成的目标是从输入图像中预测出一个图结构,其中节点对应被识别并定位的物体,边对应物体间的交互谓词。现有方法采用全监督方式训练,研究重点集中在消息传递机制、损失函数和/或偏差缓解上。本研究提出了一种简单却有效的自监督关系对齐正则化方法,旨在提升场景图生成的性能。该对齐方法具备通用性,可与任意现有场景图生成框架结合,与原模型的目标函数共同训练。对齐通过知识蒸馏实现:我们设计了一个辅助关系预测分支,该分支与监督分支结构镜像且共享参数。在辅助分支中,关系输入特征在消息传递和谓词预测前被部分掩码。随后,掩码关系的预测结果在消息传递后与监督分支的预测结果进行对齐。我们结合SGTR和Neural Motifs两种场景图生成架构验证了该自监督关系对齐方法的有效性,结果表明两种架构的性能均得到显著提升。

英文摘要:

The goal of scene graph generation is to predict a graph from an input image, where nodes correspond to identified and localized objects and edges to their corresponding interaction predicates. Existing methods are trained in a fully supervised manner and focus on message passing mechanisms, loss functions, and/or bias mitigation. In this work we introduce a simple-yet-effective self-supervised relational alignment regularization designed to improve the scene graph generation performance. The proposed alignment is general and can be combined with any existing scene graph generation framework, where it is trained alongside the original model's objective. The alignment is achieved through distillation, where an auxiliary relation prediction branch, that mirrors and shares parameters with the supervised counterpart, is designed. In the auxiliary branch, relational input features are partially masked prior to message passing and predicate prediction. The predictions for masked relations are then aligned with the supervised counterparts after the message passing. We illustrate the effectiveness of this self-supervised relational alignment in conjunction with two scene graph generation architectures, SGTR and Neural Motifs, and show that in both cases we achieve significantly improved performance.

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

↑