Learning to Compose Dynamic Tree Structures for Visual Contexts

By Kaihua Tang, Hanwang Zhang, Baoyuan Wu, Wenhan Luo, Wei David Liu. Computer Vision and Pattern Recognition, 2018

612 citations recorded by the source corpus.

We propose to compose dynamic tree structures that place the objects in an image into a visual context, helping visual reasoning tasks such as scene graph generation and visual Q&A. Our visual context tree model, dubbed VCTree, has two key advantages over existing structured object representations including chains and fully-connected graphs: 1) The efficient and expressive binary tree encodes the inherent parallel/hierarchical relationships among objects, e.g., ``clothes'' and ``pants'' are usually co-occur and belong to ``person''; 2) the dynamic structure varies from image to image and task to task, allowing more content-/task-specific message passing among objects. To construct a VCTree, we design a score function that calculates the task-dependent validity between each object pair, and the tree is the binary version of the maximum spanning tree from the score matrix. Then, visual contexts are encoded by bidirectional TreeLSTM and decoded by task-specific models. We develop a hybrid learning procedure which integrates end-task supervised learning and the tree structure reinforcement learning, where the former's evaluation result serves as a self-critic for the latter's structure exploration. Experimental results on two benchmarks, which require reasoning over contexts: Visual Genome for scene graph generation and VQA2.0 for visual Q&A, show that VCTree outperforms state-of-the-art results while discovering interpretable visual context structures.

Subfields: Artificial Intelligence, Computer Vision and Pattern Recognition

View Paper: https://www.semanticscholar.org/paper/88ec56eee6a787ccda0cf7fbfbec48c6b4ad68fe

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Hanwang Zhang
  2. Tat-seng Chua
  3. Dacheng Tao
  4. Zhiyuan Liu
  5. Mohit Bansal
  6. Yi Yang
  7. Xiangnan He
  8. Maosong Sun

In Computer Vision and Pattern Recognition

  1. Dacheng Tao
  2. Yi Yang
  3. Luc Van Gool
  4. Xiaogang Wang
  5. Hanwang Zhang
  6. Trevor Darrell
  7. Ming-Hsuan Yang
  8. Cordelia Schmid

Citations in the corpus