Variational Autoencoder for End-to-End Control of Autonomous Driving with Novelty Detection and Training De-biasing
Author(s)Amini, Alexander; Araki, Brandon; Rus, Daniela; Schwarting, Wilko; Rosman, Guy; Karaman, Sertac; Rus, Daniela L; ... Show more Show less
MetadataShow full item record
This paper introduces a new method for end-to-end training of deep neural networks (DNNs) and evaluates it in the context of autonomous driving. DNN training has been shown to result in high accuracy for perception to action learning given sufficient training data. However, the trained models may fail without warning in situations with insufficient or biased training data. In this paper, we propose and evaluate a novel architecture for self-supervised learning of latent variables to detect the insufficiently trained situations. Our method also addresses training data imbalance, by learning a set of underlying latent variables that characterize the training data and evaluate potential biases. We show how these latent distributions can be leveraged to adapt and accelerate the training pipeline by training on only a fraction of the total dataset. We evaluate our approach on a challenging dataset for driving. The data is collected from a full-scale autonomous vehicle. Our method provides qualitative explanation for the latent variables learned in the model. Finally, we show how our model can be additionally trained as an end-to-end controller, directly outputting a steering control command for an autonomous vehicle.
DepartmentMassachusetts Institute of Technology. Computer Science and Artificial Intelligence Laboratory; Massachusetts Institute of Technology. Department of Aeronautics and Astronautics; Massachusetts Institute of Technology. Department of Electrical Engineering and Computer Science; Massachusetts Institute of Technology. Department of Mechanical Engineering
2018 IEEE/RSJ International Conference on Intelligent Robots and Systems (IROS)
Amini, Alexander, Wilko Schwarting, Guy Rosman, Brandon Araki. Sertac Karaman and Daniela Rus. "Variational Autoencoder for End-to-End Control of Autonomous Driving with Novelty Detection and Training De-biasing." 2018 IEEE/RSJ International Conference on Intelligent Robots and Systems (IROS) (Palacio Municipal de Congresos, Madrid, Spain, Oct.1-5 2018)
Author's final manuscript