Scene Graph Generation (SGG) is a challenging task of detecting objects and
predicting relationships between objects. After DETR was developed, one-stage
SGG models based on a one-stage object detector have been actively studied.
However, complex modeling is used to predict the relationship between objects,
and the inherent relationship between object queries learned in the multi-head
self-attention of the object detector has been neglected. We propose a
lightweight one-stage SGG model that extracts the relation graph from the
various relationships learned in the multi-head self-attention layers of the
DETR decoder. By fully utilizing the self-attention by-products, the relation
graph can be extracted effectively with a shallow relation extraction head.
Considering the dependency of the relation extraction task on the object
detection task, we propose a novel relation smoothing technique that adjusts
the relation label adaptively according to the quality of the detected objects.
By the relation smoothing, the model is trained according to the continuous
curriculum that focuses on object detection task at the beginning of training
and performs multi-task learning as the object detection performance gradually
improves. Furthermore, we propose a connectivity prediction task that predicts
whether a relation exists between object pairs as an auxiliary task of the
relation extraction. We demonstrate the effectiveness and efficiency of our
method for the Visual Genome and Open Image V6 datasets. Our code is publicly
available at this https URL .

提出一种轻量级的一阶场景图生成（SGG）模型，通过充分利用 DETR 解码器的多头自注意力层中学习到的各种关系，使用浅层关系抽取头有效地提取关系图，并考虑到关系抽取任务对于物体检测任务的依赖，提出了一种新颖的关系平滑技术，根据检测到的物体质量自适应地调整关系标签，同时还提出了一个辅助任务 —— 连接性预测任务来预测物体对之间是否存在关系。