NODIS: Neural ordinary differential scene understanding

Yuren Cong, Hanno Ackermann, Wentong Liao, Michael Ying Yang*, Bodo Rosenhahn

*Corresponding author for this work

Research output: Chapter in Book/Report/Conference proceedingChapterAcademicpeer-review

Abstract

Semantic image understanding is a challenging topic in computer vision. It requires to detect all objects in an image, but also to identify all the relations between them. Detected objects, their labels and the discovered relations can be used to construct a scene graph which provides an abstract semantic interpretation of an image. In previous works, relations were identified by solving an assignment problem formulated as (Mixed-)Integer Linear Programs. In this work, we interpret that formulation as Ordinary Differential Equation (ODE). The proposed architecture performs scene graph inference by solving a neural variant of an ODE by end-to-end learning. The connection between (Mixed-)Integer Linear Program and ODEs in combination with the end-to-end training amounts to learning how to solve assignment problems with image-specific objective functions. Intuitive, visual explanations are provided for the role of the single free variable of the ODE modules which are associated with time in many natural processes. The proposed model achieves results equal to or above state-of-the-art on all three benchmark tasks: scene graph generation (SGGEN), classification (SGCLS) and visual relationship detection (PREDCLS) on Visual Genome benchmark. The strong results on scene graph classification support the claim that assignment problems can indeed be solved by neural ODEs.
Original languageEnglish
Title of host publicationLecture Notes in Computer Science
Subtitle of host publicationComputer Vision - ECCV 2020
EditorsA. Vedaldi, H. Bischof, T. Brox, J.M. Frahm
Place of PublicationCham
PublisherSpringer
Pages636-653
Number of pages17
Volume12365
ISBN (Electronic)978-3-030-58565-5
ISBN (Print)978-3-030-58564-8
DOIs
Publication statusPublished - 12 Nov 2020

Publication series

NameLecture Notes in Computer Science
PublisherSpringer
Volume12365
ISSN (Print)0302-9743
ISSN (Electronic)1611-3349

Fingerprint Dive into the research topics of 'NODIS: Neural ordinary differential scene understanding'. Together they form a unique fingerprint.

Cite this