PRL Workshop 2024 Workshop Paper
Conviction-Based Planning for Sparse Reward Reinforcement Learning Problems
- Simon Ouellette
- Eric Beaudry
- Mohamed Bouguessa
Deep reinforcement learning (RL) methods require a large amount of interactions, making them difficult to use in realworld applications. This is especially true when the rewards are sparse since random or semi-random exploration struggles to find them. Learning from Demonstrations (LfD) mitigates this by eliminating the need for random exploration. So far, most LfD solutions have been based on model-free RL approaches that struggle with tasks that require planning. We propose a new algorithm that successfully combines modelbased RL and LfD by leveraging the notion of uncertainty in the transition model during planning. We also introduce the concept of conviction, an uncertainty-to-reward ratio used to decide on optimal actions while planning. Our approach significantly outperforms the relevant baselines in the Minigrid and Sokoban environments.