Hierarchies of reward machines
File(s)ICML2023.pdf (2.11 MB)
Published version
Author(s)
Furelos-Blanco, D
Law, M
Jonsson, A
Broda, K
Russo, A
Type
Conference Paper
Abstract
Reward machines (RMs) are a recent formalism for representing the reward function of a reinforcement learning task through a finite-state machine whose edges encode subgoals of the task using high-level events. The structure of RMs enables the decomposition of a task into simpler and independently solvable subtasks that help tackle long-horizon and/or sparse reward tasks. We propose a formalism for further abstracting the subtask structure by endowing an RM with the ability to call other RMs, thus composing a hierarchy of RMs (HRM). We exploit HRMs by treating each call to an RM as an independently solvable subtask using the options framework, and describe a curriculum-based method to learn HRMs from traces observed by the agent. Our experiments reveal that exploiting a handcrafted HRM leads to faster convergence than with a flat HRM, and that learning an HRM is feasible in cases where its equivalent flat representation is not.
Date Issued
2023-07-23
Date Acceptance
2023-04-24
Citation
Proceedings of Machine Learning Research, 2023, 202
ISSN
2640-3498
Publisher
PMLR
Journal / Book Title
Proceedings of Machine Learning Research
Volume
202
Copyright Statement
Copyright 2023 by the author(s).
License URL
Identifier
https://proceedings.mlr.press/v202/
Source
International Conference on Machine Learning
Publication Status
Published
Start Date
2023-07-23
Finish Date
2023-07-29
Coverage Spatial
Honolulu, Hawaii, USA
Date Publish Online
2023-07-23