arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1707.09423cs.CV
收藏

基于内外语言知识蒸馏的视觉关系检测

Visual Relationship Detection with Internal and External Linguistic Knowledge Distillation

University of Maryland, College Park(马里兰大学帕克分校)

Ruichi Yu, Ang Li, Vlad I. Morariu, Larry S. Davis

摘要:

理解视觉关系涉及识别主题、对象以及与它们相关的谓词。我们利用谓词和(subj,obj)对(语义和空间)之间的强相关性来预测以主语和宾语为条件的谓词。对这三个实体联合建模更准确地反映了它们的关系,但是由于视觉关系的语义空间巨大并且训练数据有限,特别是对于具有很少实例的长尾关系,因此使学习复杂化。为了克服这个问题,我们使用语言统计学的知识来规范视觉模型学习。我们通过从训练注释(内部知识)和公开文本中挖掘来获得语言知识,例如,Wikipedia(外部知识),计算给定(subj,obj)对的谓词的条件概率分布。然后,我们将知识提取到深层模型中,以实现更好的泛化。我们在视觉关系检测(VRD)和视觉基因组数据集上的实验结果表明,通过这种语言知识蒸馏,我们的模型显著优于最先进的方法,特别是在预测看不见的关系(例如,在VRD zero-shot测试集上,召回率从8.45%提高到19.17%)。

英文摘要:

Understanding visual relationships involves identifying the subject, the object, and a predicate relating them. We leverage the strong correlations between the predicate and the (subj,obj) pair (both semantically and spatially) to predict the predicates conditioned on the subjects and the objects. Modeling the three entities jointly more accurately reflects their relationships, but complicates learning since the semantic space of visual relationships is huge and the training data is limited, especially for the long-tail relationships that have few instances. To overcome this, we use knowledge of linguistic statistics to regularize visual model learning. We obtain linguistic knowledge by mining from both training annotations (internal knowledge) and publicly available text, e.g., Wikipedia (external knowledge), computing the conditional probability distribution of a predicate given a (subj,obj) pair. Then, we distill the knowledge into a deep model to achieve better generalization. Our experimental results on the Visual Relationship Detection (VRD) and Visual Genome datasets suggest that with this linguistic knowledge distillation, our model outperforms the state-of-the-art methods significantly, especially when predicting unseen relationships (e.g., recall improved from 8.45% to 19.17% on VRD zero-shot testing set).

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

↑