Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Robust Reinforcement Learning using Offline Data

About

The goal of robust reinforcement learning (RL) is to learn a policy that is robust against the uncertainty in model parameters. Parameter uncertainty commonly occurs in many real-world RL applications due to simulator modeling errors, changes in the real-world system dynamics over time, and adversarial disturbances. Robust RL is typically formulated as a max-min problem, where the objective is to learn the policy that maximizes the value against the worst possible models that lie in an uncertainty set. In this work, we propose a robust RL algorithm called Robust Fitted Q-Iteration (RFQI), which uses only an offline dataset to learn the optimal robust policy. Robust RL with offline data is significantly more challenging than its non-robust counterpart because of the minimization over all models present in the robust Bellman operator. This poses challenges in offline data collection, optimization over the models, and unbiased estimation. In this work, we propose a systematic approach to overcome these challenges, resulting in our RFQI algorithm. We prove that RFQI learns a near-optimal robust policy under standard assumptions and demonstrate its superior performance on standard benchmark problems.

Kishan Panaganti, Zaiyan Xu, Dileep Kalathil, Mohammad Ghavamzadeh• 2022

Related benchmarks

TaskDatasetResultRank
Mathematical ReasoningAIME24
Accuracy36.67
130
Scientific ReasoningGPQA
Accuracy3.62
50
Mathematical ReasoningMATH500
Accuracy89.4
45
Mathematical ReasoningMath MATH500, AIME24, Minerva-Math, AMC23
MATH500 Score87.4
18
Scientific ReasoningScience Domain In-Domain: SampleQA, GPQA(ALL), HLE
SampleQA Score3.14
18
Mathematical ReasoningMinerva Math
Avg@1 Accuracy31.99
18
Mathematical ReasoningAMC23
Accuracy84.17
11
Scientific Reasoning & QAHLE
Accuracy3.43
7
General Reasoning & QAAll Evaluated Datasets
Average Accuracy35.98
7
Mathematical ReasoningMath Domain
Avg Accuracy60.56
7
Showing 10 of 12 rows

Other info

Code

Follow for update