A novel update mechanism for Q-Networks based on extreme learning machines

Wilson, Callum and Riccardi, Annalisa and Minisci, Edmondo; (2020) A novel update mechanism for Q-Networks based on extreme learning machines. In: 2020 International Joint Conference on Neural Networks (IJCNN). IEEE, GBR. ISBN 9781728169262 (https://doi.org/10.1109/IJCNN48605.2020.9207098)

[thumbnail of Wilson-etal-WCCI2020-A-novel-update-mechanism-for-Q-Networks]
Preview
Text. Filename: Wilson_etal_WCCI2020_A_novel_update_mechanism_for_Q_Networks.pdf
Accepted Author Manuscript

Download (319kB)| Preview

Abstract

Reinforcement learning is a popular machine learning paradigm which can find near optimal solutions to complex problems. Most often, these procedures involve function approximation using neural networks with gradient based updates to optimise weights for the problem being considered. While this common approach generally works well, there are other update mechanisms which are largely unexplored in reinforcement learning. One such mechanism is Extreme Learning Machines. These were initially proposed to drastically improve the training speed of neural networks and have since seen many applications. Here we attempt to apply extreme learning machines to a reinforcement learning problem in the same manner as gradient based updates. This new algorithm is called Extreme Q-Learning Machine (EQLM). We compare its performance to a typical Q-Network on the cart-pole task - a benchmark reinforcement learning problem - and show EQLM has similar long-term learning performance to a Q-Network.