Are All Vision Models Created Equal? A Study of the Open-Loop to Closed-Loop Causality Gap

Mathias Lechner,Ramin Hasani,Alexander Amini,Tsun-Hsuan Wang,Thomas A. Henzinger,Daniela Rus
DOI: https://doi.org/10.48550/arXiv.2210.04303
2022-10-10
Abstract:There is an ever-growing zoo of modern neural network models that can efficiently learn end-to-end control from visual observations. These advanced deep models, ranging from convolutional to patch-based networks, have been extensively tested on offline image classification and regression tasks. In this paper, we study these vision architectures with respect to the open-loop to closed-loop causality gap, i.e., offline training followed by an online closed-loop deployment. This causality gap typically emerges in robotics applications such as autonomous driving, where a network is trained to imitate the control commands of a human. In this setting, two situations arise: 1) Closed-loop testing in-distribution, where the test environment shares properties with those of offline training data. 2) Closed-loop testing under distribution shifts and out-of-distribution. Contrary to recently reported results, we show that under proper training guidelines, all vision models perform indistinguishably well on in-distribution deployment, resolving the causality gap. In situation 2, We observe that the causality gap disrupts performance regardless of the choice of the model architecture. Our results imply that the causality gap can be solved in situation one with our proposed training guideline with any modern network architecture, whereas achieving out-of-distribution generalization (situation two) requires further investigations, for instance, on data diversity rather than the model architecture.
Computer Vision and Pattern Recognition,Artificial Intelligence,Machine Learning,Neural and Evolutionary Computing,Robotics
What problem does this paper attempt to address?
The problem that this paper attempts to solve is the causality gap that exists when visual models are transferred from offline training to online deployment in robot applications, especially in the field of autonomous driving. Specifically, the paper explores the performance of different advanced deep - learning visual models in the following two cases: 1. **Closed - loop test with the same distribution**: In this case, the test environment has similar properties to the offline training data. 2. **Closed - loop test with distribution change or different distribution**: In this case, the test environment is significantly different from the training environment. The main contribution of the paper lies in designing a unified end - to - end training infrastructure for fairly comparing the performance of different advanced deep - learning architectures in robot imitation learning tasks, and systematically verifying the generalization ability of these models in same - distribution and different - distribution tests through experiments. The study found that in the same - distribution test, all visual models can perform well as long as appropriate training methods are adopted, thus solving the causality - gap problem. However, in the different - distribution test, the causality gap still affects the model performance, and this influence is hardly affected by the choice of model architecture. This indicates that, in addition to the model architecture itself, factors such as data diversity, augmentation strategies, and training settings are also important in improving the model generalization ability.