Neural Network-Based Learning from Demonstration of an Autonomous Ground Robot

• Main Authors: Yiwei Fu, Devesh K. Jha, Zeyu Zhang, Zhenyuan Yuan, Asok Ray
• Format: Article
• Language: English
• Published: MDPI AG 2019-04-01
• Series: Machines
• Subjects: autonomous robots; neural networks; imitation learning
• Online Access: https://www.mdpi.com/2075-1702/7/2/24

This paper presents and experimentally validates a concept of end-to-end imitation learning for autonomous systems by using a composite architecture of convolutional neural network (ConvNet) and Long Short Term Memory (LSTM) neural network. In particular, a spatio-temporal deep neural network is developed, which learns to imitate the policy used by a human supervisor to drive a car-like robot in a maze environment. The spatial and temporal components of the imitation model are learned by using deep convolutional network and recurrent neural network architectures, respectively. The imitation model learns the policy of a human supervisor as a function of laser light detection and ranging (LIDAR) data, which is then used in real time to drive a robot in an autonomous fashion in a laboratory setting. The performance of the proposed model for imitation learning is compared with that of several other state-of-the-art methods, reported in the machine learning literature, for spatial and temporal modeling. The learned policy is implemented on a robot using a Nvidia Jetson TX2 board which, in turn, is validated on test tracks. The proposed spatio-temporal model outperforms several other off-the-shelf machine learning techniques to learn the policy.