FORM: Learning Expressive and Transferable First-Order Logic Reward Machines

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Ardon, Leo, Furelos-Blanco, Daniel, Parac, Roko, Russo, Alessandra
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866917939885637632
author Ardon, Leo
Furelos-Blanco, Daniel
Parac, Roko
Russo, Alessandra
author_facet Ardon, Leo
Furelos-Blanco, Daniel
Parac, Roko
Russo, Alessandra
contents Reward machines (RMs) are an effective approach for addressing non-Markovian rewards in reinforcement learning (RL) through finite-state machines. Traditional RMs, which label edges with propositional logic formulae, inherit the limited expressivity of propositional logic. This limitation hinders the learnability and transferability of RMs since complex tasks will require numerous states and edges. To overcome these challenges, we propose First-Order Reward Machines ($\texttt{FORM}$s), which use first-order logic to label edges, resulting in more compact and transferable RMs. We introduce a novel method for $\textbf{learning}$ $\texttt{FORM}$s and a multi-agent formulation for $\textbf{exploiting}$ them and facilitate their transferability, where multiple agents collaboratively learn policies for a shared $\texttt{FORM}$. Our experimental results demonstrate the scalability of $\texttt{FORM}$s with respect to traditional RMs. Specifically, we show that $\texttt{FORM}$s can be effectively learnt for tasks where traditional RM learning approaches fail. We also show significant improvements in learning speed and task transferability thanks to the multi-agent learning framework and the abstraction provided by the first-order language.
format Preprint
id arxiv_https___arxiv_org_abs_2501_00364
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle FORM: Learning Expressive and Transferable First-Order Logic Reward Machines
Ardon, Leo
Furelos-Blanco, Daniel
Parac, Roko
Russo, Alessandra
Artificial Intelligence
Formal Languages and Automata Theory
Logic in Computer Science
Symbolic Computation
Reward machines (RMs) are an effective approach for addressing non-Markovian rewards in reinforcement learning (RL) through finite-state machines. Traditional RMs, which label edges with propositional logic formulae, inherit the limited expressivity of propositional logic. This limitation hinders the learnability and transferability of RMs since complex tasks will require numerous states and edges. To overcome these challenges, we propose First-Order Reward Machines ($\texttt{FORM}$s), which use first-order logic to label edges, resulting in more compact and transferable RMs. We introduce a novel method for $\textbf{learning}$ $\texttt{FORM}$s and a multi-agent formulation for $\textbf{exploiting}$ them and facilitate their transferability, where multiple agents collaboratively learn policies for a shared $\texttt{FORM}$. Our experimental results demonstrate the scalability of $\texttt{FORM}$s with respect to traditional RMs. Specifically, we show that $\texttt{FORM}$s can be effectively learnt for tasks where traditional RM learning approaches fail. We also show significant improvements in learning speed and task transferability thanks to the multi-agent learning framework and the abstraction provided by the first-order language.
title FORM: Learning Expressive and Transferable First-Order Logic Reward Machines
topic Artificial Intelligence
Formal Languages and Automata Theory
Logic in Computer Science
Symbolic Computation
url https://arxiv.org/abs/2501.00364