Offline Reinforcement Learning of High-Quality Behaviors Under Robust Style Alignment

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Petitbois, Mathieu, Portelas, Rémy, Lamprier, Sylvain
Format: Preprint
Published: 2026
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914294969401344
author Petitbois, Mathieu
Portelas, Rémy
Lamprier, Sylvain
author_facet Petitbois, Mathieu
Portelas, Rémy
Lamprier, Sylvain
contents We study offline reinforcement learning of style-conditioned policies using explicit style supervision via subtrajectory labeling functions. In this setting, aligning style with high task performance is particularly challenging due to distribution shift and inherent conflicts between style and reward. Existing methods, despite introducing numerous definitions of style, often fail to reconcile these objectives effectively. To address these challenges, we propose a unified definition of behavior style and instantiate it into a practical framework. Building on this, we introduce Style-Conditioned Implicit Q-Learning (SCIQL), which leverages offline goal-conditioned RL techniques, such as hindsight relabeling and value learning, and combine it with a new Gated Advantage Weighted Regression mechanism to efficiently optimize task performance while preserving style alignment. Experiments demonstrate that SCIQL achieves superior performance on both objectives compared to prior offline methods. Code, datasets and visuals are available in: https://sciql-iclr-2026.github.io/.
format Preprint
id arxiv_https___arxiv_org_abs_2601_22823
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Offline Reinforcement Learning of High-Quality Behaviors Under Robust Style Alignment
Petitbois, Mathieu
Portelas, Rémy
Lamprier, Sylvain
Machine Learning
Artificial Intelligence
Robotics
We study offline reinforcement learning of style-conditioned policies using explicit style supervision via subtrajectory labeling functions. In this setting, aligning style with high task performance is particularly challenging due to distribution shift and inherent conflicts between style and reward. Existing methods, despite introducing numerous definitions of style, often fail to reconcile these objectives effectively. To address these challenges, we propose a unified definition of behavior style and instantiate it into a practical framework. Building on this, we introduce Style-Conditioned Implicit Q-Learning (SCIQL), which leverages offline goal-conditioned RL techniques, such as hindsight relabeling and value learning, and combine it with a new Gated Advantage Weighted Regression mechanism to efficiently optimize task performance while preserving style alignment. Experiments demonstrate that SCIQL achieves superior performance on both objectives compared to prior offline methods. Code, datasets and visuals are available in: https://sciql-iclr-2026.github.io/.
title Offline Reinforcement Learning of High-Quality Behaviors Under Robust Style Alignment
topic Machine Learning
Artificial Intelligence
Robotics
url https://arxiv.org/abs/2601.22823