Distributionally Robust Deep Q-Learning

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Lu, Chung I, Sester, Julian, Zhang, Aijia
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916757204107264
author Lu, Chung I
Sester, Julian
Zhang, Aijia
author_facet Lu, Chung I
Sester, Julian
Zhang, Aijia
contents We propose a novel distributionally robust $Q$-learning algorithm for the non-tabular case accounting for continuous state spaces where the state transition of the underlying Markov decision process is subject to model uncertainty. The uncertainty is taken into account by considering the worst-case transition from a ball around a reference probability measure. To determine the optimal policy under the worst-case state transition, we solve the associated non-linear Bellman equation by dualising and regularising the Bellman operator with the Sinkhorn distance, which is then parameterized with deep neural networks. This approach allows us to modify the Deep Q-Network algorithm to optimise for the worst case state transition. We illustrate the tractability and effectiveness of our approach through several applications, including a portfolio optimisation task based on S\&{P}~500 data.
format Preprint
id arxiv_https___arxiv_org_abs_2505_19058
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Distributionally Robust Deep Q-Learning
Lu, Chung I
Sester, Julian
Zhang, Aijia
Machine Learning
Optimization and Control
Portfolio Management
We propose a novel distributionally robust $Q$-learning algorithm for the non-tabular case accounting for continuous state spaces where the state transition of the underlying Markov decision process is subject to model uncertainty. The uncertainty is taken into account by considering the worst-case transition from a ball around a reference probability measure. To determine the optimal policy under the worst-case state transition, we solve the associated non-linear Bellman equation by dualising and regularising the Bellman operator with the Sinkhorn distance, which is then parameterized with deep neural networks. This approach allows us to modify the Deep Q-Network algorithm to optimise for the worst case state transition. We illustrate the tractability and effectiveness of our approach through several applications, including a portfolio optimisation task based on S\&{P}~500 data.
title Distributionally Robust Deep Q-Learning
topic Machine Learning
Optimization and Control
Portfolio Management
url https://arxiv.org/abs/2505.19058