Yikang Li, Wanli Ouyang, Bolei Zhou, Kun Wang, Xiaogang Wang
Object detection, scene graph generation and region captioning, which are three scene understanding tasks at different semantic levels, are tied together: scene graphs are generated on top of objects detected in an image with their pairwise relationship predicted, while region captioning gives a language description of the objects, their attributes, relations, and other context information. In this work, to leverage the mutual connections across semantic levels, we propose a novel neural network model, termed as Multi-level Scene Description Network (denoted as MSDN), to solve the three vision tasks jointly in an end-to-end manner. Objects, phrases, and caption regions are first aligned with a dynamic graph based on their spatial and semantic connections. Then a feature refining structure is used to pass messages across the three levels of semantic tasks through the graph. We benchmark the learned model on three tasks, and show the joint learning across three tasks with our proposed method can bring mutual improvements over previous models. Particularly, on the scene graph generation task, our proposed method outperforms the state-of-art method with more than 3% margin.
| Task | Dataset | Metric | Value | Model |
|---|---|---|---|---|
| Scene Parsing | Visual Genome | Recall@50 | 10.72 | MSDN |
| Object Detection | Visual Genome | MAP | 7.43 | MSDN |
| 3D | Visual Genome | MAP | 7.43 | MSDN |
| 2D Semantic Segmentation | Visual Genome | Recall@50 | 10.72 | MSDN |
| 2D Classification | Visual Genome | MAP | 7.43 | MSDN |
| Scene Graph Generation | Visual Genome | Recall@50 | 10.72 | MSDN |
| 2D Object Detection | Visual Genome | MAP | 7.43 | MSDN |
| 16k | Visual Genome | MAP | 7.43 | MSDN |