SHIRE: Enhancing Sample Efficiency using Human Intuition in REinforcement Learning

arXiv:2409.09990 · cs.LG, cs.NE, cs.RO · Submitted 2024-09-16 · Read on arXiv

cs.LG, cs.NE, cs.RO

Submitted: 2024-09-16

Updated: 2025-04-26

Comments: Accepted for publication at the IEEE International Conference on Robotics & Automation (ICRA) 2025

Journal ref: 2025 IEEE International Conference on Robotics and Automation (ICRA), Atlanta, GA, USA, 2025, pp. 13399-13405

DOI: 10.1109/ICRA55743.2025.11128459

License: http://creativecommons.org/licenses/by-nc-nd/4.0/

The gist: The ability of neural networks to perform robotic perception and control tasks such as depth and optical flow estimation, simultaneous localization and mapping (SLAM), and automatic control has led

Terminology

Abstract

The ability of neural networks to perform robotic perception and control tasks such as depth and optical flow estimation, simultaneous localization and mapping (SLAM), and automatic control has led to their widespread adoption in recent years. Deep Reinforcement Learning has been used extensively in these settings, as it does not have the unsustainable training costs associated with supervised learning. However, DeepRL suffers from poor sample efficiency, i.e., it requires a large number of environmental interactions to converge to an acceptable solution. Modern RL algorithms such as Deep Q Learning and Soft Actor-Critic attempt to remedy this shortcoming but can not provide the explainability required in applications such as autonomous robotics. Humans intuitively understand the long-time-horizon sequential tasks common in robotics. Properly using such intuition can make RL policies more explainable while enhancing their sample efficiency. In this work, we propose SHIRE, a novel framework for encoding human intuition using Probabilistic Graphical Models (PGMs) and using it in the Deep RL training pipeline to enhance sample efficiency. Our framework achieves 25-78% sample efficiency gains across the environments we evaluate at negligible overhead cost. Additionally, by teaching RL agents the encoded elementary behavior, SHIRE enhances policy explainability. A real-world demonstration further highlights the efficacy of policies trained using our framework.

Sources

Related papers