Unbiased Scene Graph Generation From Biased Training

By Kaihua Tang, Yulei Niu, Jianqiang Huang, Jiaxin Shi, Hanwang Zhang. 2020

853 citations recorded by the source corpus.

Today’s scene graph generation (SGG) task is still far from practical, mainly due to the severe training bias, e.g., collapsing diverse "human walk on / sit on / lay on beach" into "human on beach". Given such SGG, the down-stream tasks such as VQA can hardly infer better scene structures than merely a bag of objects. However, debiasing in SGG is not trivial because traditional debiasing methods cannot distinguish between the good and bad bias, e.g., good context prior (e.g., "person read book" rather than "eat") and bad long-tailed bias (e.g., "near" dominating "behind / in front of"). In this paper, we present a novel SGG framework based on causal inference but not the conventional likelihood. We first build a causal graph for SGG, and perform traditional biased training with the graph. Then, we propose to draw the counterfactual causality from the trained graph to infer the effect from the bad bias, which should be removed. In particular, we use Total Direct Effect (TDE) as the proposed final predicate score for unbiased SGG. Note that our framework is agnostic to any SGG model and thus can be widely applied in the community who seeks unbiased predictions. By using the proposed Scene Graph Diagnosis toolkit on the SGG benchmark Visual Genome and several prevailing models, we observed significant improvements over the previous state-of-the-art methods.

Subfields: Artificial Intelligence, Computer Vision and Pattern Recognition, Graph Learning

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. Mohit Bansal
  3. Tat-seng Chua
  4. Jianfeng Gao
  5. Xiangnan He
  6. Yi Yang
  7. Trevor Darrell
  8. Yoshua Bengio

In Computer Vision and Pattern Recognition

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

In Graph Learning

  1. Hanwang Zhang
  2. Bo Wu
  3. Jun Liu
  4. Yueting Zhuang
  5. Graham W. Taylor
  6. Ji-rong Wen
  7. Chunyan Miao
  8. Jianfei Cai

Citations in the corpus