ACL-QL: Adaptive Conservative Level in Q-Learning for Offline Reinforcement Learning

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Wu, Kun, Zhao, Yinuo, Xu, Zhiyuan, Che, Zhengping, Yin, Chengxiang, Liu, Chi Harold, Feng, Feiferi, Tang, Jian
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913737393307648
author Wu, Kun
Zhao, Yinuo
Xu, Zhiyuan
Che, Zhengping
Yin, Chengxiang
Liu, Chi Harold
Feng, Feiferi
Tang, Jian
author_facet Wu, Kun
Zhao, Yinuo
Xu, Zhiyuan
Che, Zhengping
Yin, Chengxiang
Liu, Chi Harold
Feng, Feiferi
Tang, Jian
contents Offline Reinforcement Learning (RL), which operates solely on static datasets without further interactions with the environment, provides an appealing alternative to learning a safe and promising control policy. The prevailing methods typically learn a conservative policy to mitigate the problem of Q-value overestimation, but it is prone to overdo it, leading to an overly conservative policy. Moreover, they optimize all samples equally with fixed constraints, lacking the nuanced ability to control conservative levels in a fine-grained manner. Consequently, this limitation results in a performance decline. To address the above two challenges in a united way, we propose a framework, Adaptive Conservative Level in Q-Learning (ACL-QL), which limits the Q-values in a mild range and enables adaptive control on the conservative level over each state-action pair, i.e., lifting the Q-values more for good transitions and less for bad transitions. We theoretically analyze the conditions under which the conservative level of the learned Q-function can be limited in a mild range and how to optimize each transition adaptively. Motivated by the theoretical analysis, we propose a novel algorithm, ACL-QL, which uses two learnable adaptive weight functions to control the conservative level over each transition. Subsequently, we design a monotonicity loss and surrogate losses to train the adaptive weight functions, Q-function, and policy network alternatively. We evaluate ACL-QL on the commonly used D4RL benchmark and conduct extensive ablation studies to illustrate the effectiveness and state-of-the-art performance compared to existing offline DRL baselines.
format Preprint
id arxiv_https___arxiv_org_abs_2412_16848
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle ACL-QL: Adaptive Conservative Level in Q-Learning for Offline Reinforcement Learning
Wu, Kun
Zhao, Yinuo
Xu, Zhiyuan
Che, Zhengping
Yin, Chengxiang
Liu, Chi Harold
Feng, Feiferi
Tang, Jian
Machine Learning
Artificial Intelligence
Robotics
Offline Reinforcement Learning (RL), which operates solely on static datasets without further interactions with the environment, provides an appealing alternative to learning a safe and promising control policy. The prevailing methods typically learn a conservative policy to mitigate the problem of Q-value overestimation, but it is prone to overdo it, leading to an overly conservative policy. Moreover, they optimize all samples equally with fixed constraints, lacking the nuanced ability to control conservative levels in a fine-grained manner. Consequently, this limitation results in a performance decline. To address the above two challenges in a united way, we propose a framework, Adaptive Conservative Level in Q-Learning (ACL-QL), which limits the Q-values in a mild range and enables adaptive control on the conservative level over each state-action pair, i.e., lifting the Q-values more for good transitions and less for bad transitions. We theoretically analyze the conditions under which the conservative level of the learned Q-function can be limited in a mild range and how to optimize each transition adaptively. Motivated by the theoretical analysis, we propose a novel algorithm, ACL-QL, which uses two learnable adaptive weight functions to control the conservative level over each transition. Subsequently, we design a monotonicity loss and surrogate losses to train the adaptive weight functions, Q-function, and policy network alternatively. We evaluate ACL-QL on the commonly used D4RL benchmark and conduct extensive ablation studies to illustrate the effectiveness and state-of-the-art performance compared to existing offline DRL baselines.
title ACL-QL: Adaptive Conservative Level in Q-Learning for Offline Reinforcement Learning
topic Machine Learning
Artificial Intelligence
Robotics
url https://arxiv.org/abs/2412.16848