Test-time Adaptation with Slot-Centric Models

July 2023

Test-time Adaptation with Slot-Centric Models

Authors:

Mihir Prabhudesai, Anirudh Goyal, Sujoy Paul, Sjoerd Van Steenkiste, Mehdi S. M. Sajjadi, Gaurav Aggarwal, Thomas Kipf, Deepak Pathak, and Katerina Fragkiadaki
Conference Paper
Proceedings of (ICML) International Conference on Machine Learning

Abstract:

Current visual detectors, though impressive within their training distribution, often fail to parse out-of-distribution scenes into their constituent entities. Recent test-time adaptation methods use auxiliary self-supervised losses to adapt the network parameters to each test example independently and have shown promising results towards generalization outside the training distribution for the task of image classification. In our work, we find evidence that these losses are insufficient for the task of scene decomposition, without also considering architectural inductive biases. Recent slot-centric generative models attempt to decompose scenes into entities in a self-supervised manner by reconstructing pixels. Drawing upon these two lines of work, we propose Slot-TTA, a semi-supervised slot-centric scene decomposition model that at test time is adapted per scene through gradient descent on reconstruction or cross-view synthesis objectives. We evaluate Slot-TTA across multiple input modalities, images or 3D point clouds, and show substantial out-of-distribution performance improvements against state-of-the-art supervised feed-forward detectors, and alternative test-time adaptation methods.
@conference{Prabhudesai-2023-136799,
author = {Mihir Prabhudesai And Anirudh Goyal And Sujoy Paul And Sjoerd Van Steenkiste And Mehdi S. M. Sajjadi And Gaurav Aggarwal And Thomas Kipf And Deepak Pathak And Katerina Fragkiadaki},
title = {Test-time Adaptation with Slot-Centric Models},
booktitle = {Proceedings of (ICML) International Conference on Machine Learning},
year = {2023},
month = {July},
keywords = {object-centric learning, test-time adaptation, entity-centric models, scene decomposition},
}
Copyright notice: This material is presented to ensure timely dissemination of scholarly and technical work. Copyright and all rights therein are retained by authors or by other copyright holders. All persons copying this information are expected to adhere to the terms and constraints invoked by each author's copyright. These works may not be reposted without the explicit permission of the copyright holder.