Can we learn policies in reinforcement learning without rewards? Can we learn a policy just by trying to reach a goal state? We answer these questions positively by proposing a multi-step procedure that first learns a world model that goes backward in time, secondly generates goal-reaching backward trajectories, thirdly improves those sequences using shortest path finding algorithms, and finally trains a neural network policy by imitation learning. We evaluate our method on a deterministic maze environment where the observations are 64×64 pixel bird's eye images and can show that it consistently reaches several goals.
@article{arxiv.2312.05044,
title = {Backward Learning for Goal-Conditioned Policies},
author = {Marc Höftmann and Jan Robine and Stefan Harmeling},
journal= {arXiv preprint arXiv:2312.05044},
year = {2024}
}
Comments
World Models, Goal-conditioned, Reward-free, Workshop on Goal-Conditioned Reinforcement Learning - NeurIPS 2023