Hostname: page-component-74656f9884-skzmx Total loading time: 0 Render date: 2026-11-05T15:48:30.078Z Has data issue: false hasContentIssue false

Integrating task objectives for improved performance of preference-based reinforcement learning in different tasks

Published online by Cambridge University Press:  30 March 2026

Jiabin Hu
Affiliation:
College of Intelligent Robotics and Advanced Manufacturing, Fudan University, China
Peng Zhai*
Affiliation:
College of Intelligent Robotics and Advanced Manufacturing, Fudan University, China
Xiaoyi Wei
Affiliation:
College of Intelligent Robotics and Advanced Manufacturing, Fudan University, China
Cunwang Liu
Affiliation:
College of Intelligent Robotics and Advanced Manufacturing, Fudan University, China
Lihua Zhang*
Affiliation:
College of Intelligent Robotics and Advanced Manufacturing, Fudan University, China Jilin Provincial Key Laboratory of Intelligence Science and Engineering, Changchun, Jilin, China
*
Corresponding authors: Peng Zhai; Email: pzhai@fudan.edu.cn; Lihua Zhang; Email: lihuazhang@fudan.edu.cn
Corresponding authors: Peng Zhai; Email: pzhai@fudan.edu.cn; Lihua Zhang; Email: lihuazhang@fudan.edu.cn
Rights & Permissions [Opens in a new window]

Abstract

Preference-based reinforcement learning (PbRL) significantly simplifies the design of reward functions in reinforcement learning (RL) tasks. However, because of the tasks’ complexity, intransitive preferences, and sensitivity to preference errors, PbRL requires substantial feedback to achieve the desired performance. This extensive reliance on expert input notably increases the burden on participants. We have developed a novel framework: Self-teacher-learning preference-based reinforcement learning (STL-PbRL). In the teacher-led (TL) module, agents learn more reliable reward prediction models (RM) through TL PbRL. In the self-learning (SL) module, agents utilizing the preference comparison approach for trajectory segments integrate sparse but critical, easily designed task-oriented information into the feedback process. The STL-PbRL framework incorporates the SL module to refine the RM initially generated by the TL module. We have demonstrated that this integration significantly enhances RM by enabling RM to converge toward an optimal reward model that effectively supports achieving a training policy that meets task objectives. This streamlined and efficient STL-PbRL framework enables a more accurate and efficient training process. Our experimental results confirm that the SL module seamlessly integrates with existing PbRL algorithms, significantly reducing the need for feedback and alleviating the impact of errors in preference indications. This efficiency and effectiveness highlight that STL-PbRL innovates, simplifies, and enhances the RL training process across various applications.

Information

Type
Research Article
Copyright
© The Author(s), 2026. Published by Cambridge University Press
Figure 0

Figure 1. Framework of STL-PbRL. In the original PbRL, the training of the RM (module 3) is driven by expert (human) feedback (module 1), which captures expert preferences for agent trajectories by directly influencing the reward system with preferred trajectory segments. STL-PbRL enhances this methodology by integrating a self-learning mechanism (Module 2), which autonomously analyzes the merits of trajectories using a fitness function. By comparing trajectories with high and low fitness values, it optimizes the Reward Model (RM), thereby improving the overall learning process.

Figure 1

Algorithm 1 Trajectory Classification and Storage

Figure 2

Table I. The specific details of parts of environments. || denotes the L2 norm and 1[] denotes the indicator function.

Figure 3

Table II. Results on Meta-world tasks (mean $\pm$ std).

Figure 4

Figure 2. Training curves on six continuous control tasks from Meta-world. The solid line and shaded regions denote the success rate’s mean and standard deviation across five runs.

Figure 5

Figure 3. Training curves on three continuous control tasks from DMControl. The solid line and shaded regions denote the episode target value mean and standard deviation across five runs.

Figure 6

Table III. Results on DMControl tasks (mean $\pm$ std).

Figure 7

Table IV. STL-PbRL is applied in three different PbRL algorithms to verify its versatility.

Figure 8

Figure 4. STL-PbRL is applied in three different PbRL algorithms in the Button Press task to verify its versatility. The solid line and shaded regions denote the success rate’s mean and standard deviation across five runs.

Figure 9

Figure 5. The impact of varying numbers of preference feedback (Feedback=100, 400, 700 & 1000) on Door Open task performance. The solid line and shaded regions denote the success rate’s mean and standard deviation across five runs.

Figure 10

Table V. Comparison of the impact of the TL module on the SL module. It is noted that MRN includes TL module but SAC does not.

Figure 11

Figure 6. The training curve for the impact of preference errors on STL-MRN, MRN, and RIME performance across different tasks.

Figure 12

Table VI. MRN with fitness function directly.

Figure 13

Table VII. Comparison between BC-MRN and STL-MRN on six Meta-world tasks.

Figure 14

Figure 7. Real-world deployment of STL-PbRL on the Unitree Go1 quadruped. STL-PbRL successfully maintains the target base height and orientation (30$^{\circ }$inclination) through self-learning refinement. In contrast, baseline PbRL trained purely from pairwise preferences exhibits unstable posture and fails to achieve target alignment. (a) STL-PbRL (height=0.2 m, real), (b) PbRL (height=0.2 m, simulation), (c) STL-PbRL (slope=30$^{\circ }$, real), (d) PbRL (slope=30$^{\circ }$, simulation).

Figure 15

Table VIII. Average wall-clock training time (minutes:seconds) on Meta-world door-open-v2 and relative overhead of STL variants.

Figure 16

Algorithm 2 STL-PbRL(PEBBLE) Algorithm

Figure 17

Table IX. Fitness function of DMControl.

Figure 18

Figure 8. DMControl environments.

Figure 19

Figure 9. Meta-world environments. Each subfigure illustrates a distinct manipulation task.

Figure 20

Table X. Fitness function of Meta-world.

Figure 21

Table XI. Hyperparameters of SAC.

Figure 22

Table XII. Hyperparameters of PEBBLE.

Figure 23

Table XIII. Hyperparameters of SURF.

Figure 24

Table XIV. Hyperparameters of MRN.

Figure 25

Table XV. Hyperparameters of RIME.

Figure 26

Table XVI. Hyperparameters of QPA.

Figure 27

Table XVII. Hyperparameters of STL-PbRL.

Figure 28

Table XVIII. Reward terms used in Go1 STL-PbRL training.

Figure 29

Table XIX. Domain randomization parameters.

Supplementary material: File

Hu et al. supplementary material

Hu et al. supplementary material
Download Hu et al. supplementary material(File)
File 12.7 MB