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

Relation Transformer Network

Ludwig Maximilian University of Munich(慕尼黑大学);Technical University of Munich(慕尼黑工业大学);Siemens AG(西门子股份公司)

Rajat Koner, Suprosanna Shit, Volker Tresp

英文摘要:

The extraction of a scene graph with objects as nodes and mutual relationships as edges is the basis for a deep understanding of image content. Despite recent advances, such as message passing and joint classification, the detection of visual relationships remains a challenging task due to sub-optimal exploration of the mutual interaction among the visual objects. In this work, we propose a novel transformer formulation for scene graph generation and relation prediction. We leverage the encoder-decoder architecture of the transformer for rich feature embedding of nodes and edges. Specifically, we model the node-to-node interaction with the self-attention of the transformer encoder and the edge-to-node interaction with the cross-attention of the transformer decoder. Further, we introduce a novel positional embedding suitable to handle edges in the decoder. Finally, our relation prediction module classifies the directed relation from the learned node and edge embedding. We name this architecture as Relation Transformer Network (RTN). On the Visual Genome and GQA dataset, we have achieved an overall mean of 4.85% and 3.1% point improvement in comparison with state-of-the-art methods. Our experiments show that Relation Transformer can efficiently model context across various datasets with small, medium, and large-scale relation classification.

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

↑