QHD: A brain-inspired hyperdimensional reinforcement learning algorithm

Yang Ni,Yeseong Kim,Danny Abraham,Pietro Mercati,M. Imani,Mariam Issa
DOI: https://doi.org/10.48550/arXiv.2205.06978
Abstract:Reinforcement Learning (RL) has opened up new opportunities to solve a wide range of complex decision-making tasks. However, modern RL algorithms, e.g., Deep Q-Learning, are based on deep neural networks, putting high computational costs when running on edge devices. In this paper, we propose QHD , a Hyper-dimensional Reinforcement Learning, that mimics brain properties towards robust and real-time learning. QHD relies on a light-weight brain-inspired model to learn an optimal policy in an unknown environment. We first develop a novel mathematical foundation and encoding module that maps state-action space into high-dimensional space. We accordingly develop a hyperdimen-sional regression model to approximate the Q-value function. QHD - powered agent makes decisions by comparing Q-values of each possible action. We evaluate the effect of the different RL training batch sizes and local memory capacity on the QHD quality of learning. Our QHD is also capable of online learning with tiny local memory capacity, which can be as small as the training batch size. QHD provides real-time learning by further decreasing the memory capacity and the batch size. This makes QHD suitable for highly-efficient reinforcement learning in the edge environment, where it is crucial to support online and real-time learning. Our solution also supports small experience replay batch size that provides 12.3 × speedup compared to DQN while ensuring minimal quality loss. Our evaluation shows QHD capability for real-time learning, providing 34.6 × speedup and significantly better quality of learning than state-of-the-art deep RL algorithms.
Mathematics,Computer Science
What problem does this paper attempt to address?