Top-Down Guidance for Learning Object-Centric Representations

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Zou, Junhong, Zhu, Xiangyu, Zhang, Zhaoxiang, Lei, Zhen
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866908499551715328
author Zou, Junhong
Zhu, Xiangyu
Zhang, Zhaoxiang
Lei, Zhen
author_facet Zou, Junhong
Zhu, Xiangyu
Zhang, Zhaoxiang
Lei, Zhen
contents Humans' innate ability to decompose scenes into objects allows for efficient understanding, predicting, and planning. In light of this, Object-Centric Learning (OCL) attempts to endow networks with similar capabilities, learning to represent scenes with the composition of objects. However, existing OCL models only learn through reconstructing the input images, which does not assist the model in distinguishing objects, resulting in suboptimal object-centric representations. This flaw limits current object-centric models to relatively simple downstream tasks. To address this issue, we draw on humans' top-down vision pathway and propose Top-Down Guided Network (TDGNet), which includes a top-down pathway to improve object-centric representations. During training, the top-down pathway constructs guidance with high-level object-centric representations to optimize low-level grid features output by the backbone. While during inference, it refines object-centric representations by detecting and solving conflicts between low- and high-level features. We show that TDGNet outperforms current object-centric models on multiple datasets of varying complexity. In addition, we expand the downstream task scope of object-centric representations by applying TDGNet to the field of robotics, validating its effectiveness in downstream tasks including video prediction and visual planning.
format Preprint
id arxiv_https___arxiv_org_abs_2405_10598
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Top-Down Guidance for Learning Object-Centric Representations
Zou, Junhong
Zhu, Xiangyu
Zhang, Zhaoxiang
Lei, Zhen
Computer Vision and Pattern Recognition
Humans' innate ability to decompose scenes into objects allows for efficient understanding, predicting, and planning. In light of this, Object-Centric Learning (OCL) attempts to endow networks with similar capabilities, learning to represent scenes with the composition of objects. However, existing OCL models only learn through reconstructing the input images, which does not assist the model in distinguishing objects, resulting in suboptimal object-centric representations. This flaw limits current object-centric models to relatively simple downstream tasks. To address this issue, we draw on humans' top-down vision pathway and propose Top-Down Guided Network (TDGNet), which includes a top-down pathway to improve object-centric representations. During training, the top-down pathway constructs guidance with high-level object-centric representations to optimize low-level grid features output by the backbone. While during inference, it refines object-centric representations by detecting and solving conflicts between low- and high-level features. We show that TDGNet outperforms current object-centric models on multiple datasets of varying complexity. In addition, we expand the downstream task scope of object-centric representations by applying TDGNet to the field of robotics, validating its effectiveness in downstream tasks including video prediction and visual planning.
title Top-Down Guidance for Learning Object-Centric Representations
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2405.10598