Table of Contents
Fetching ...

Iterative Batch Reinforcement Learning via Safe Diversified Model-based Policy Search

Amna Najib, Stefan Depeweg, Phillip Swazinna

TL;DR

This work presents an algorithmic methodology for iterative batch reinforcement learning based on ensemble-based model-based policy search, augmented with safety and, importantly, a diversity criterion, leading to continuous improvement of learned policies while remaining within the support of collected data.

Abstract

Batch reinforcement learning enables policy learning without direct interaction with the environment during training, relying exclusively on previously collected sets of interactions. This approach is, therefore, well-suited for high-risk and cost-intensive applications, such as industrial control. Learned policies are commonly restricted to act in a similar fashion as observed in the batch. In a real-world scenario, learned policies are deployed in the industrial system, inevitably leading to the collection of new data that can subsequently be added to the existing recording. The process of learning and deployment can thus take place multiple times throughout the lifespan of a system. In this work, we propose to exploit this iterative nature of applying offline reinforcement learning to guide learned policies towards efficient and informative data collection during deployment, leading to continuous improvement of learned policies while remaining within the support of collected data. We present an algorithmic methodology for iterative batch reinforcement learning based on ensemble-based model-based policy search, augmented with safety and, importantly, a diversity criterion.

Iterative Batch Reinforcement Learning via Safe Diversified Model-based Policy Search

TL;DR

This work presents an algorithmic methodology for iterative batch reinforcement learning based on ensemble-based model-based policy search, augmented with safety and, importantly, a diversity criterion, leading to continuous improvement of learned policies while remaining within the support of collected data.

Abstract

Batch reinforcement learning enables policy learning without direct interaction with the environment during training, relying exclusively on previously collected sets of interactions. This approach is, therefore, well-suited for high-risk and cost-intensive applications, such as industrial control. Learned policies are commonly restricted to act in a similar fashion as observed in the batch. In a real-world scenario, learned policies are deployed in the industrial system, inevitably leading to the collection of new data that can subsequently be added to the existing recording. The process of learning and deployment can thus take place multiple times throughout the lifespan of a system. In this work, we propose to exploit this iterative nature of applying offline reinforcement learning to guide learned policies towards efficient and informative data collection during deployment, leading to continuous improvement of learned policies while remaining within the support of collected data. We present an algorithmic methodology for iterative batch reinforcement learning based on ensemble-based model-based policy search, augmented with safety and, importantly, a diversity criterion.

Paper Structure

This paper contains 17 sections, 14 equations, 17 figures, 1 table, 1 algorithm.

Figures (17)

  • Figure 1: Illustration of iterative model-based policy search.
  • Figure 2: 2D grid environment: The behavior policy guides agent towards the nearest behavior goal. The best reward goal represents the state with the highest reward. Reward decreases according to Gaussian distribution represented by circle lines.
  • Figure 3: Policy maps for different $\lambda$ values, colors illustrate action directions in every cell of the grid. For $\lambda = 0.0$, the policy imitates behavior policy by navigating to the closest behavior goal. With increasing $\lambda$ values, the policy moves slowly towards the reward goal.
  • Figure 5: Constrained Policy.
  • Figure 6: Soft Constrain.
  • ...and 12 more figures