Reinforcement Learning with Uncertainty Estimation for Tactical Decision-Making in Intersections
Paper in proceeding, 2020

This paper investigates how a Bayesian reinforcement learning method can be used to create a tactical decision-making agent for autonomous driving in an intersection scenario, where the agent can estimate the confidence of its decisions. An ensemble of neural networks, with additional randomized prior functions (RPF), are trained by using a bootstrapped experience replay memory. The coefficient of variation in the estimated Q-values of the ensemble members is used to approximate the uncertainty, and a criterion that determines if the agent is sufficiently confident to make a particular decision is introduced. The performance of the ensemble RPF method is evaluated in an intersection scenario and compared to a standard Deep Q-Network method, which does not estimate the uncertainty. It is shown that the trained ensemble RPF agent can detect cases with high uncertainty, both in situations that are far from the training distribution, and in situations that seldom occur within the training distribution. This work demonstrates one possible application of such a confidence estimate, by using this information to choose safe actions in unknown situations, which removes all collisions from within the training distribution, and most collisions outside of the distribution.

Author

Carl-Johan E Hoel

Chalmers, Mechanics and Maritime Sciences, Vehicle Engineering and Autonomous Systems

AI Sweden

Volvo Group

Tommy Tram

Zenuity AB

Chalmers, Electrical Engineering, Systems and control, Mechatronics

AI Sweden

Jonas Sjöberg

Chalmers, Electrical Engineering, Systems and control, Mechatronics

IEEE Conference on Intelligent Transportation Systems, Proceedings, ITSC

9294407

2020 IEEE 23rd International Conference on Intelligent Transportation Systems (ITSC)
Rhodes, Greece,

WASP SAS: Structuring data for continuous processing and ML systems

Wallenberg AI, Autonomous Systems and Software Program, 2018-01-01 -- 2023-01-01.

Areas of Advance

Transport

Subject Categories

Electrical Engineering, Electronic Engineering, Information Engineering

Probability Theory and Statistics

Computer Systems

DOI

10.1109/ITSC45102.2020.9294407

More information

Latest update

9/27/2021