Divergence-Augmented Policy Optimization

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Wang, Qing, Li, Yingru, Xiong, Jiechao, Zhang, Tong
Format: Preprint
Veröffentlicht: 2025
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866913665321533440
author Wang, Qing
Li, Yingru
Xiong, Jiechao
Zhang, Tong
author_facet Wang, Qing
Li, Yingru
Xiong, Jiechao
Zhang, Tong
contents In deep reinforcement learning, policy optimization methods need to deal with issues such as function approximation and the reuse of off-policy data. Standard policy gradient methods do not handle off-policy data well, leading to premature convergence and instability. This paper introduces a method to stabilize policy optimization when off-policy data are reused. The idea is to include a Bregman divergence between the behavior policy that generates the data and the current policy to ensure small and safe policy updates with off-policy data. The Bregman divergence is calculated between the state distributions of two policies, instead of only on the action probabilities, leading to a divergence augmentation formulation. Empirical experiments on Atari games show that in the data-scarce scenario where the reuse of off-policy data becomes necessary, our method can achieve better performance than other state-of-the-art deep reinforcement learning algorithms.
format Preprint
id arxiv_https___arxiv_org_abs_2501_15034
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Divergence-Augmented Policy Optimization
Wang, Qing
Li, Yingru
Xiong, Jiechao
Zhang, Tong
Machine Learning
Artificial Intelligence
In deep reinforcement learning, policy optimization methods need to deal with issues such as function approximation and the reuse of off-policy data. Standard policy gradient methods do not handle off-policy data well, leading to premature convergence and instability. This paper introduces a method to stabilize policy optimization when off-policy data are reused. The idea is to include a Bregman divergence between the behavior policy that generates the data and the current policy to ensure small and safe policy updates with off-policy data. The Bregman divergence is calculated between the state distributions of two policies, instead of only on the action probabilities, leading to a divergence augmentation formulation. Empirical experiments on Atari games show that in the data-scarce scenario where the reuse of off-policy data becomes necessary, our method can achieve better performance than other state-of-the-art deep reinforcement learning algorithms.
title Divergence-Augmented Policy Optimization
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2501.15034