Memory-Based Network for Scene Graph with Unbalanced Relations

Publisher:
ASSOC COMPUTING MACHINERY
Publication Type:
Conference Proceeding
Citation:
MM 2020 - Proceedings of the 28th ACM International Conference on Multimedia, 2020, pp. 2400-2408
Issue Date:
2020-10-12
Filename Description Size
3394171.3413507.pdfAccepted version3.84 MB
Adobe PDF
Full metadata record
The scene graph which can be represented by a set of visual triples is composed of objects and the relations between object pairs. It is vital for image captioning, visual question answering, and many other applications. However, there is a long tail distribution on the scene graph dataset, and the tail relation cannot be accurately identified due to the lack of training samples. The problem of the nonstandard label and feature overlap on the scene graph affects the extraction of discriminative features and exacerbates the effect of data imbalance on the model. For these reasons, we propose a novel scene graph generation model that can effectively improve the detection of low-frequency relations. We use the method of memory features to realize the transfer of high-frequency relation features to low-frequency relation features. Extensive experiments on scene graph datasets show that our model significantly improved the performance of two evaluation metrics R@K and mR@K compared with state-of-the-art baselines.
Please use this identifier to cite or link to this item: