arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2002.11949cs.CVcs.LG

从有偏训练中学习无偏场景图生成

Unbiased Scene Graph Generation from Biased Training

Nanyang Technological University(南洋理工大学);Damo Academy, Alibaba Group(阿里达摩院);Renmin University of China(中国人民大学);Tsinghua University(清华大学)

Kaihua Tang, Yulei Niu, Jianqiang Huang, Jiaxin Shi, Hanwang Zhang

AI总结:

针对场景图生成中的训练偏差问题,提出基于因果推断和总直接效应的去偏框架,在Visual Genome上显著提升无偏性能。

摘要:

当今的场景图生成(SGG)任务仍远未达到实用水平,主要原因是严重的训练偏差,例如,将多样的“人在沙滩上行走/坐在沙滩上/躺在沙滩上”压缩为“人在沙滩上”。鉴于这种SGG,诸如视觉问答(VQA)之类的下游任务很难推断出比单纯的对象集合更好的场景结构。然而,在SGG中进行去偏并非易事,因为传统的去偏方法无法区分好的偏差和坏的偏差,例如,好的上下文先验(如“人读书”而非“吃”)和坏的长尾偏差(如“附近”主导“后面/前面”)。在本文中,我们提出了一种基于因果推断而非传统似然的新型SGG框架。我们首先为SGG构建一个因果图,并使用该图进行传统的有偏训练。然后,我们提出从训练好的图中推断反事实因果关系,以提取应被移除的坏偏差效应。具体而言,我们使用总直接效应(TDE)作为最终的无偏SGG谓词分数。值得注意的是,我们的框架与任何SGG模型无关,因此可以广泛应用于社区中寻求无偏预测的场景。通过在SGG基准Visual Genome上使用所提出的场景图诊断工具包,并对几种主流模型进行评估,我们观察到相较于之前的最先进方法有显著改进。

英文摘要:

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.

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

↑