Saved in:
Bibliographic Details
Main Authors: Pang, Teng, Wang, Bingzheng, Wu, Guoqiang, Yin, Yilong
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2503.01143
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914053234884608
author Pang, Teng
Wang, Bingzheng
Wu, Guoqiang
Yin, Yilong
author_facet Pang, Teng
Wang, Bingzheng
Wu, Guoqiang
Yin, Yilong
contents Offline preference-based reinforcement learning (PbRL) mitigates the need for reward definition, aligning with human preferences via preference-driven reward feedback without interacting with the environment. However, trajectory-wise preference labels are difficult to meet the precise learning of step-wise reward, thereby affecting the performance of downstream algorithms. To alleviate the insufficient step-wise reward caused by trajectory-wise preferences, we propose a novel preference-based reward acquisition method: Diffusion Preference-based Reward (DPR). DPR directly treats step-wise preference-based reward acquisition as a binary classification and utilizes the robustness of diffusion classifiers to infer step-wise rewards discriminatively. In addition, to further utilize trajectory-wise preference information, we propose Conditional Diffusion Preference-based Reward (C-DPR), which conditions on trajectory-wise preference labels to enhance reward inference. We apply the above methods to existing offline RL algorithms, and a series of experimental results demonstrate that the diffusion classifier-driven reward outperforms the previous reward acquisition method with the Bradley-Terry model.
format Preprint
id arxiv_https___arxiv_org_abs_2503_01143
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Diffusion Classifier-Driven Reward for Offline Preference-based Reinforcement Learning
Pang, Teng
Wang, Bingzheng
Wu, Guoqiang
Yin, Yilong
Machine Learning
Offline preference-based reinforcement learning (PbRL) mitigates the need for reward definition, aligning with human preferences via preference-driven reward feedback without interacting with the environment. However, trajectory-wise preference labels are difficult to meet the precise learning of step-wise reward, thereby affecting the performance of downstream algorithms. To alleviate the insufficient step-wise reward caused by trajectory-wise preferences, we propose a novel preference-based reward acquisition method: Diffusion Preference-based Reward (DPR). DPR directly treats step-wise preference-based reward acquisition as a binary classification and utilizes the robustness of diffusion classifiers to infer step-wise rewards discriminatively. In addition, to further utilize trajectory-wise preference information, we propose Conditional Diffusion Preference-based Reward (C-DPR), which conditions on trajectory-wise preference labels to enhance reward inference. We apply the above methods to existing offline RL algorithms, and a series of experimental results demonstrate that the diffusion classifier-driven reward outperforms the previous reward acquisition method with the Bradley-Terry model.
title Diffusion Classifier-Driven Reward for Offline Preference-based Reinforcement Learning
topic Machine Learning
url https://arxiv.org/abs/2503.01143