References & Citations
Computer Science > Machine Learning
Title: Non-Markovian Reward Modelling from Trajectory Labels via Interpretable Multiple Instance Learning
(Submitted on 30 May 2022 (v1), last revised 10 Oct 2022 (this version, v2))
Abstract: We generalise the problem of reward modelling (RM) for reinforcement learning (RL) to handle non-Markovian rewards. Existing work assumes that human evaluators observe each step in a trajectory independently when providing feedback on agent behaviour. In this work, we remove this assumption, extending RM to capture temporal dependencies in human assessment of trajectories. We show how RM can be approached as a multiple instance learning (MIL) problem, where trajectories are treated as bags with return labels, and steps within the trajectories are instances with unseen reward labels. We go on to develop new MIL models that are able to capture the time dependencies in labelled trajectories. We demonstrate on a range of RL tasks that our novel MIL models can reconstruct reward functions to a high level of accuracy, and can be used to train high-performing agent policies.
Submission history
From: Joseph Early [view email][v1] Mon, 30 May 2022 18:20:22 GMT (17111kb,D)
[v2] Mon, 10 Oct 2022 17:28:57 GMT (27542kb,D)
Link back to: arXiv, form interface, contact.