Action-Constrained Imitation Learning

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Yeh, Chia-Han, Nan, Tse-Sheng, Vuorio, Risto, Hung, Wei, Wu, Hung-Yen, Sun, Shao-Hua, Hsieh, Ping-Chun
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916909417496576
author Yeh, Chia-Han
Nan, Tse-Sheng
Vuorio, Risto
Hung, Wei
Wu, Hung-Yen
Sun, Shao-Hua
Hsieh, Ping-Chun
author_facet Yeh, Chia-Han
Nan, Tse-Sheng
Vuorio, Risto
Hung, Wei
Wu, Hung-Yen
Sun, Shao-Hua
Hsieh, Ping-Chun
contents Policy learning under action constraints plays a central role in ensuring safe behaviors in various robot control and resource allocation applications. In this paper, we study a new problem setting termed Action-Constrained Imitation Learning (ACIL), where an action-constrained imitator aims to learn from a demonstrative expert with larger action space. The fundamental challenge of ACIL lies in the unavoidable mismatch of occupancy measure between the expert and the imitator caused by the action constraints. We tackle this mismatch through \textit{trajectory alignment} and propose DTWIL, which replaces the original expert demonstrations with a surrogate dataset that follows similar state trajectories while adhering to the action constraints. Specifically, we recast trajectory alignment as a planning problem and solve it via Model Predictive Control, which aligns the surrogate trajectories with the expert trajectories based on the Dynamic Time Warping (DTW) distance. Through extensive experiments, we demonstrate that learning from the dataset generated by DTWIL significantly enhances performance across multiple robot control tasks and outperforms various benchmark imitation learning algorithms in terms of sample efficiency. Our code is publicly available at https://github.com/NYCU-RL-Bandits-Lab/ACRL-Baselines.
format Preprint
id arxiv_https___arxiv_org_abs_2508_14379
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Action-Constrained Imitation Learning
Yeh, Chia-Han
Nan, Tse-Sheng
Vuorio, Risto
Hung, Wei
Wu, Hung-Yen
Sun, Shao-Hua
Hsieh, Ping-Chun
Robotics
Machine Learning
Policy learning under action constraints plays a central role in ensuring safe behaviors in various robot control and resource allocation applications. In this paper, we study a new problem setting termed Action-Constrained Imitation Learning (ACIL), where an action-constrained imitator aims to learn from a demonstrative expert with larger action space. The fundamental challenge of ACIL lies in the unavoidable mismatch of occupancy measure between the expert and the imitator caused by the action constraints. We tackle this mismatch through \textit{trajectory alignment} and propose DTWIL, which replaces the original expert demonstrations with a surrogate dataset that follows similar state trajectories while adhering to the action constraints. Specifically, we recast trajectory alignment as a planning problem and solve it via Model Predictive Control, which aligns the surrogate trajectories with the expert trajectories based on the Dynamic Time Warping (DTW) distance. Through extensive experiments, we demonstrate that learning from the dataset generated by DTWIL significantly enhances performance across multiple robot control tasks and outperforms various benchmark imitation learning algorithms in terms of sample efficiency. Our code is publicly available at https://github.com/NYCU-RL-Bandits-Lab/ACRL-Baselines.
title Action-Constrained Imitation Learning
topic Robotics
Machine Learning
url https://arxiv.org/abs/2508.14379