From Action Labels to Sets: Rethinking Action Supervision for Imitation Learning from Corrective Feedback

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Li, Zhaoting, Pérez-Dattari, Rodrigo, Babuska, Robert, Della Santina, Cosimo, Kober, Jens
Format: Preprint
Veröffentlicht: 2025
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866917448897265664
author Li, Zhaoting
Pérez-Dattari, Rodrigo
Babuska, Robert
Della Santina, Cosimo
Kober, Jens
author_facet Li, Zhaoting
Pérez-Dattari, Rodrigo
Babuska, Robert
Della Santina, Cosimo
Kober, Jens
contents Behavior cloning (BC) optimizes policies by treating human demonstrations as pointwise action labels. While effective with accurate action labels, this formulation is brittle in practice: when human-provided actions are imperfect, treating each label as an exact target can steer the policy away from the underlying desired behavior, particularly when expressive models are used (e.g., energy-based models). As a result, we propose a human-in-the-loop alternative that replaces pointwise supervision with set-valued action targets. We introduce Contrastive policy Learning from Interactive Corrections (CLIC). CLIC leverages human corrections to construct and refine sets of desired actions, and optimizes a policy to place probability mass over these sets rather than over a single action target. This formulation naturally accommodates both absolute and relative corrections and can represent complex multi-modal behaviors. Extensive simulation and real-robot experiments show that the proposed approach leads to effective policy learning across diverse settings: CLIC remains competitive with the state of the art under accurate data while being substantially more robust under noisy, relative, and partial feedback. Our implementation is publicly available at https://clic-webpage.github.io/.
format Preprint
id arxiv_https___arxiv_org_abs_2502_07645
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle From Action Labels to Sets: Rethinking Action Supervision for Imitation Learning from Corrective Feedback
Li, Zhaoting
Pérez-Dattari, Rodrigo
Babuska, Robert
Della Santina, Cosimo
Kober, Jens
Robotics
Behavior cloning (BC) optimizes policies by treating human demonstrations as pointwise action labels. While effective with accurate action labels, this formulation is brittle in practice: when human-provided actions are imperfect, treating each label as an exact target can steer the policy away from the underlying desired behavior, particularly when expressive models are used (e.g., energy-based models). As a result, we propose a human-in-the-loop alternative that replaces pointwise supervision with set-valued action targets. We introduce Contrastive policy Learning from Interactive Corrections (CLIC). CLIC leverages human corrections to construct and refine sets of desired actions, and optimizes a policy to place probability mass over these sets rather than over a single action target. This formulation naturally accommodates both absolute and relative corrections and can represent complex multi-modal behaviors. Extensive simulation and real-robot experiments show that the proposed approach leads to effective policy learning across diverse settings: CLIC remains competitive with the state of the art under accurate data while being substantially more robust under noisy, relative, and partial feedback. Our implementation is publicly available at https://clic-webpage.github.io/.
title From Action Labels to Sets: Rethinking Action Supervision for Imitation Learning from Corrective Feedback
topic Robotics
url https://arxiv.org/abs/2502.07645