UNIQ: Offline Inverse Q-learning for Avoiding Undesirable Demonstrations

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Hoang, Huy, Mai, Tien, Varakantham, Pradeep
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866912068200824832
author Hoang, Huy
Mai, Tien
Varakantham, Pradeep
author_facet Hoang, Huy
Mai, Tien
Varakantham, Pradeep
contents We address the problem of offline learning a policy that avoids undesirable demonstrations. Unlike conventional offline imitation learning approaches that aim to imitate expert or near-optimal demonstrations, our setting involves avoiding undesirable behavior (specified using undesirable demonstrations). To tackle this problem, unlike standard imitation learning where the aim is to minimize the distance between learning policy and expert demonstrations, we formulate the learning task as maximizing a statistical distance, in the space of state-action stationary distributions, between the learning policy and the undesirable policy. This significantly different approach results in a novel training objective that necessitates a new algorithm to address it. Our algorithm, UNIQ, tackles these challenges by building on the inverse Q-learning framework, framing the learning problem as a cooperative (non-adversarial) task. We then demonstrate how to efficiently leverage unlabeled data for practical training. Our method is evaluated on standard benchmark environments, where it consistently outperforms state-of-the-art baselines. The code implementation can be accessed at: https://github.com/hmhuy0/UNIQ.
format Preprint
id arxiv_https___arxiv_org_abs_2410_08307
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle UNIQ: Offline Inverse Q-learning for Avoiding Undesirable Demonstrations
Hoang, Huy
Mai, Tien
Varakantham, Pradeep
Machine Learning
Artificial Intelligence
We address the problem of offline learning a policy that avoids undesirable demonstrations. Unlike conventional offline imitation learning approaches that aim to imitate expert or near-optimal demonstrations, our setting involves avoiding undesirable behavior (specified using undesirable demonstrations). To tackle this problem, unlike standard imitation learning where the aim is to minimize the distance between learning policy and expert demonstrations, we formulate the learning task as maximizing a statistical distance, in the space of state-action stationary distributions, between the learning policy and the undesirable policy. This significantly different approach results in a novel training objective that necessitates a new algorithm to address it. Our algorithm, UNIQ, tackles these challenges by building on the inverse Q-learning framework, framing the learning problem as a cooperative (non-adversarial) task. We then demonstrate how to efficiently leverage unlabeled data for practical training. Our method is evaluated on standard benchmark environments, where it consistently outperforms state-of-the-art baselines. The code implementation can be accessed at: https://github.com/hmhuy0/UNIQ.
title UNIQ: Offline Inverse Q-learning for Avoiding Undesirable Demonstrations
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2410.08307