A goal-conditioned policy search method with multi-timescale value function tuning

Zhihong Jiang, Jiachen Hu, Yan Zhao, Xiao Huang*, Hui Li

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

Abstract

Purpose: Current reinforcement learning (RL) algorithms are facing issues such as low learning efficiency and poor generalization performance, which significantly limit their practical application in real robots. This paper aims to adopt a hybrid model-based and model-free policy search method with multi-timescale value function tuning, aiming to allow robots to learn complex motion planning skills in multi-goal and multi-constraint environments with a few interactions. Design/methodology/approach: A goal-conditioned model-based and model-free search method with multi-timescale value function tuning is proposed in this paper. First, the authors construct a multi-goal, multi-constrained policy optimization approach that fuses model-based policy optimization with goal-conditioned, model-free learning. Soft constraints on states and controls are applied to ensure fast and stable policy iteration. Second, an uncertainty-aware multi-timescale value function learning method is proposed, which constructs a multi-timescale value function network and adaptively chooses the value function planning timescales according to the value prediction uncertainty. It implicitly reduces the value representation complexity and improves the generalization performance of the policy. Findings: The algorithm enables physical robots to learn generalized skills in real-world environments through a handful of trials. The simulation and experimental results show that the algorithm outperforms other relevant model-based and model-free RL algorithms. Originality/value: This paper combines goal-conditioned RL and the model predictive path integral method into a unified model-based policy search framework, which improves the learning efficiency and policy optimality of motor skill learning in multi-goal and multi-constrained environments. An uncertainty-aware multi-timescale value function learning and selection method is proposed to overcome long horizon problems, improve optimal policy resolution and therefore enhance the generalization ability of goal-conditioned RL.

Original languageEnglish
Pages (from-to)549-559
Number of pages11
JournalRobotic Intelligence and Automation
Volume44
Issue number4
DOIs
Publication statusPublished - 18 Jul 2024

Keywords

  • Bioinspired learning
  • Goal-conditioned reinforcement learning
  • Model-based policy search
  • Multi-timescale value function

Fingerprint

Dive into the research topics of 'A goal-conditioned policy search method with multi-timescale value function tuning'. Together they form a unique fingerprint.

Cite this