A Meta-Reinforcement Learning Approach to Process Control

Meta-learning is a branch of machine learning which aims to quickly adapt models, such as neural networks, to perform new tasks by learning an underlying structure across related tasks. In essence, models are being trained to learn new tasks effectively rather than master a single task. Meta-learning is appealing for process control applications because the perturbations to a process required to train an AI controller can be costly and unsafe. Additionally, the dynamics and control objectives are similar across many different processes, so it is feasible to create a generalizable controller through meta-learning capable of quickly adapting to different systems. In this work, we construct a deep reinforcement learning (DRL) based controller and meta-train the controller using a latent context variable through a separate embedding neural network. We test our meta-algorithm on its ability to adapt to new process dynamics as well as different control objectives on the same process. In both cases, our meta-learning algorithm adapts very quickly to new tasks, outperforming a regular DRL controller trained from scratch. Meta-learning appears to be a promising approach for constructing more intelligent and sample-efficient controllers.



There are no comments yet.


page 1

page 2

page 3

page 4


Unsupervised Meta-Learning for Reinforcement Learning

Meta-learning is a powerful tool that builds on multi-task learning to l...

Behaviour-conditioned policies for cooperative reinforcement learning tasks

The cooperation among AI systems, and between AI systems and humans is b...

Model Primitive Hierarchical Lifelong Reinforcement Learning

Learning interpretable and transferable subpolicies and performing task ...

Fast Slow Learning: Incorporating Synthetic Gradients in Neural Memory Controllers

Neural Memory Networks (NMNs) have received increased attention in recen...

Adaptive-Control-Oriented Meta-Learning for Nonlinear Systems

Real-time adaptation is imperative to the control of robots operating in...

Meta-Learning of Compositional Task Distributions in Humans and Machines

Modern machine learning systems struggle with sample efficiency and are ...

Learning and Fast Adaptation for Grid Emergency Control via Deep Meta Reinforcement Learning

As power systems are undergoing a significant transformation with more u...
This week in AI

Get the week's most popular data science and artificial intelligence research sent straight to your inbox every Saturday.

1 Introduction

Reinforcement learning (RL) is a branch of machine learning in which the objective is to learn an optimal policy through interactions with a stochastic environment (Sutton and Barto, 2018). Some notable examples of the potential of RL are by Silver et al. (2016) and Berner et al. (2019) due to their algorithmic and computational advances. Despite all the success RL has seen in recent years, there are still many practical challenges preventing it from being a viable and ubiquitous control framework.

The review paper of Badgwell et al. (2018) discusses such challenges, which can roughly be categorized into several categories: algorithmic (for example, convergence, modularity, hierarchical design), practical and technological (for example, state constraints and integrating MPC), robustness (for example, learning under uncertainty or exploiting system knowledge). Overshadowing all of these challenges is the problem of sample efficiency; that is, the number of (online) interactions with an environment the agent needs in order to achieve high performance.

Perhaps the most intuitive approach to increasing sample efficiency is model-based RL. A model can improve sample efficiency because it can augment otherwise model-free algorithms with simulated experiences (Janner et al., 2019). However, the underlying algorithm still operates online, meaning the model is continually updated based on recent experience. Alternatively, constructing and training in a simulated environment is also possible (Petsagkourakis et al., 2020). Despite significant improvements in sample efficiency, these approaches aim to learn a control law for a particular system. In contrast, we are interested in more general algorithms specifically designed to utilize past experience for rapid adaptation to new environments.

Meta-learning, or “learning to learn”, is an active area of research in machine learning in which the objective is to learn an underlying structure governing a distribution of possible tasks (Finn et al., 2017). In process control applications, meta-learning is appealing because many systems have similar dynamics or a known structure, which suggests training over a distribution could improve the sample efficiency when learning any single task. Moreover, extensive online learning is impractical for training over a large number of systems; by focusing on learning a latent structure for the tasks, we can more readily adapt to a new system.

In this work, we propose using meta-reinforcement learning (meta-RL) for process control applications. We create a deep deterministic policy gradient-based controller which contains an embedding neural network. This embedding network uses process data, referred to as “context”, to learn about the system dynamics and encode this information in a low-dimensional vector fed to the “actor-critic” part of the controller responsible for creating a control policy. This framework extends model-based RL to problems where no model is available. The controller is trained using a distribution of different processes and control objectives, referred to as “tasks”:

where is a process and set of control objectives while is a distribution of all possible process dynamics and control objectives. We aim to use this framework to develop a “universal controller” which can quickly adapt to effectively control any process by learning a control policy which covers a distribution of all possible tasks rather than a single task.

This paper is organized as follows: In Section 2 we summarize key concepts from RL and meta-RL; Section 3 describes our algorithm for meta-RL; we then demonstrate our approach through numerical examples in Section 4, and conclude in Section 5.

2 Background

In this section, we give a brief overview of DRL and highlight some popular meta-RL methods. We refer the reader to Nian et al. (2020); Spielberg et al. (2019) for a tutorial overview of DRL with applications to process control. We use the standard RL terminology that can be found in Sutton and Barto (2018).

The RL framework consists of an agent and and environment. One can imagine a controller and tuning algorithm (agent) operating in a continuously stirred tank reactor (environment). For each state the agent encounters, it takes some action , leading to a new state

. The action is chosen according to a conditional probability distribution

called a policy; we denote this relationship by

. Although the system dynamics are not necessarily known, we assume they can be described as a Markov decision process (MDP) with initial distribution

and transition probability

. A state-space model in control is a special case of a MDP. At each time step, a bounded scalar reward (or negative cost, rather) is evaluated. The reward function describes the desirability of a state-action pair: defining it is a key part of the design process. The overall objective, however, is the expected long-term reward. In terms of a user-specified discount factor , the optimization problem of interest becomes

maximize (1)
over all

In this problem, refers to a typical trajectory generated by the policy with sub-sequential states distributed according to . Within the space of all possible policies, we optimize over a parameterized subset whose members are denoted . For example, could denote the weights in a deep neural network or the coefficients in a proportional-integral-derivative (PID) controller. Throughout this paper, we use as a generic term for neural network weights, sometimes differentiating between them with .

Common approaches to solving (1) involve -learning (value-based methods) and the policy gradient theorem (policy-based methods) (Sutton and Barto, 2018). These methods form the basis for DRL algorithms, that is, a class of algorithms for solving RL tasks with the aid of deep neural networks. Deep neural networks are a flexible form of function approximators, well-suited for learning complex control laws. Moreover, function approximation methods make RL problems tractable in continuous state and action spaces (Lillicrap et al., 2015; Silver et al., 2014; Sutton et al., 2000)

. Without them, discretization of the state and action spaces is necessary, accentuating the “curse of dimensionality”.

A standard approach to solving (1) uses gradient ascent:


where is a step-size parameter. Analytic expressions for such a gradient exist for both stochastic and deterministic policies (Sutton and Barto, 2018; Silver et al., 2014). Crucially, these formulas rely on the state-action value function, or -function:



is not known precisely, as it depends both on the dynamics and the policy, it is estimated with a deep neural network, which we denote by

(Mnih et al., 2015). In particular, is trained to minimize the temporal difference error across samples of observations indexed by (or variations of this, as given in the forthcoming references):


where . represents the next state in the trajectory following policy . With an up-to-date -network network, we then define the following approximation for our true objective :


These ideas are the basis of popular DRL algorithms such as DDPG, TD3, SAC (Lillicrap et al., 2015; Fujimoto et al., 2018; Haarnoja et al., 2018). More generally, they fall into the class of actor-critic methods (Konda and Tsitsiklis, 2000), as they learn both a parameterized policy and value function . The term “actor” is interchangeable with “policy” when it is trained in this setting.

While the algorithms mentioned above can achieve impressive results in a wide range of domains, they are designed to be applied to a single MDP. In contrast, meta-RL aims to generalize agents to a distribution of related tasks. A task is the collection of state and action spaces, dynamics, and rewards as described in the introduction of this section (Finn et al., 2017). Crucially, meta-RL does not aim to find a single controller that performs well across different plants; instead, meta-RL agents aims to simultaneously learn the underlying structure characterizing different plants and the corresponding optimal control strategy under its reward function. The practical benefit is that this enables RL agents to quickly adapt to novel environments. In this paper, the terms “environment” and “task” can be be used interchangeably.

There are two components to meta-learning algorithms: the models (e.g., actor-critic networks) that solve a given task, and a set of meta-parameters that learn how to update the model (Bengio et al., 1992; Andrychowicz et al., 2016). The popular algorithm Model-agnostic meta-learning (MAML) (Finn et al., 2017), and methods thereafter such as Proximal meta-policy search (Rothfuss et al., 2018), combine these two steps by optimizing the model parameters for fast adaptation (that is, with few gradient descent update steps), rather than for performance on individual tasks. Unfortunately, in RL, this algorithm only works with on-policy data, meaning it does not make use of past samples. This results in agents that can indeed adapt to new tasks quickly, but require an unrealistic amount of experience to get to this point (Mendonca et al., 2019).

Due to the shared structure among tasks in process control applications, we are interested in context-based meta-RL methods (Rakelly et al., 2019; Duan et al., 2016; Wang et al., 2016). These approaches learn a latent representation of each task, enabling the agent to simultaneously learn the context and policy for a given task. In particular, we adopt the method proposed by Rakelly et al. (2019) because of its modular structure, meaning it can be ‘layered’ on top of a DRL algorithm of choice, and improve performance over previous approaches. The next section provides more details and covers our modifications.

3 Off-policy Meta-Learning

In our work, we start with Deep Deterministic Policy Gradient (DDPG) algorithm as our reinforcement learning algorithm. DDPG is appealing because it is model-free, off-policy, and compatible with continuous action spaces (Lillicrap et al., 2015). Off-policy refers to the fact that DDPG is able to learn from previous interactions it has had with its environment. This means it can learn an optimal control law, in part, with past data. Many deep RL algorithms are on-policy and can only learn from their most recent experiences with the environment which are produced using the controller’s current policy. Storing and utilizing past experiences make off-policy algorithms much more sample efficient, a useful property in the context of creating a controller which can adapt to new tasks with as few interactions with its environment as possible.

To make the DDPG algorithm a meta-RL algorithm, we use the approach developed by Rakelly et al. (2019). A batch of prior task-specific experience is fed to an embedding network which produces a low-dimensional latent variable . The actor and critic networks are trained using as an augmented component in the state vector. The latent variable aims to represent the process dynamics and understand the control objective for the agent’s current task. This disentangles the problems of understanding the process dynamics and control objectives from figuring out a policy to achieve the control objectives on the given process dynamics. The embedding network is tasked with solving for the process dynamics given raw process data while the actor-critic networks are tasked with developing an optimal control strategy given the embedded latent variable . If the controller is trained across a large distribution of tasks, we hypothesize it should be able to adapt to controlling a new process with similar dynamics with no task-specific training by exploiting the shared structure across the tasks.

Figure 1 shows the structure of the meta-RL controller used in this paper, building on the work of Rakelly et al. (2019). Interactions between the controller and an environment (task) generate experience tuples of states, actions, rewards, and next states which are stored in a replay buffer. Small batches of these experiences are sampled as context () to the embedding network, , which computes the latent context variable . During training, individual state-action pairs are fed to the actor-critic network along with the latent context variable. The actor uses and to select an action. The critic is used to create a value function and judges how desirable the actions taken by the actor are.

Figure 1: Diagram of the meta-RL network during simulation and training. is the embedding network, is the critic network, and is the actor network. We train over a distribution of tasks; for example, a sample could be the transfer function . , , are used to highlight the 3 neural networks have unique parameters.

Past experience is sampled differently for training the embedding network versus the actor-critic networks. Rakelly et al. (2019) showed training is more efficient when context that is recent, hence closer to on-policy, is used to create the embeddings. The actor-critic instead samples uniformly from the replay buffer. The embedding network context sampler is denoted as while the actor-critic experience replay sampler is denoted as .

We experimented with both deterministic embeddings (DE), probabilistic embeddings (PE), and no embeddings at all (also called multi-task learning — a regular DRL controller is trained across a distribution of tasks). Rakelly et al. (2019) suggest using PEs and treating

as a random variable. The embedding network

calculates the posterior . In contrast, DEs treat as a deterministic variable and calculate . Rakelly et al. (2019) demonstrate that PEs have better performance in sparse reward or partially observable environments. However, the use of DEs may be justified in many industrial control problems as the reward signal is present at every time-step (the set-point tracking error is commonly used) and the environment dynamics are fully observable if the batch of experience used to construct the latent variable is large (i.e., the embedding network produces through looking at many different state transitions), and contains informative data like setpoint changes. Algorithms 1 and 2 outline the meta-training and meta-testing procedures for our controller, respectively.

Require: Batch of training tasks from ,  learning rates

1:Initialize replay buffer for each task
2:for each training episode do
3:     for each  do
4:         Initialize process simulation
5:         Sample context from replay buffer
7:         while not done do
8:              Gather data from and add to
9:         end while
10:     end for
11:     for each training step do
12:         for each  do
13:              Sample context batch
14:              Sample transitions batch
18:         end for
19:         )
22:     end for
23:end for
Algorithm 1 Meta-RL Controller Training
Adapted from Rakelly et al. (2019)

Require: Testing task from learning rates

1:Initialize replay buffer
2:for each episode do
3:     Initialize process simulation
4:     Sample context from replay buffer
6:     while not done do
7:         Gather data from and add to
8:     end while
9:     for each training step do
10:         Sample context
11:         Sample transitions batch
17:     end for
18:end for
Algorithm 2 Meta-RL Controller Adaptability Testing

4 Experimental Results

We perform two experiments to assess the efficacy of our meta-RL algorithm for industrial process control applications. In each example, we examine how context embeddings affect the agent’s ability to simultaneously control multiple tasks (generalizability) and also the agent’s sample efficiency when presented with a novel task (adaptability). We compare the relative performance of an agent using DE, PE, and no embeddings. In Section 4.1, we look at an example where an agent is trained on multiple systems with different dynamics then tested on a new system with novel dynamics. In Section 4.2, we look at an example of an agent being trained across multiple control objectives while the system dynamics are held constant; the model is then evaluated based on its adaptability to a new control objective.

4.1 Learning New Dynamics

4.1.1 Preliminary Binary Gain Example

In this preliminary experiment, the performance of a DRL controller with no embeddings and a DRL controller with DEs are compared on the simple transfer functions and . The state vector is

where is the setpoint tracking error and is the integral of the setpoint tracking error over the current training episode; the same as would be found in a PID controller. Note that is used throughout this paper to represent the state of the system, while s in transfer functions represents the Laplace variable. While in ideal circumstances only would need to be included in the state to completely describe the first order systems used in this example, we include additional -values in the state to allow the controller to better respond to the Gaussian measurement noise in the system.

The reward function is

i.e. the negative absolute setpoint tracking error. While in many process control contexts, the controller optimization problem is based on minimizing a quadratic function such as (the squared error), the absolute value places more emphasis on attenuating small tracking errors.

A sample trajectory of each controller is shown in Figure 2. In each case, the different controllers are tasked with tracking the same set point and given the same initial condition. The sampling time used by the controllers in this example and all following examples is 0.5 seconds.

The meta-RL controller is able to master this toy problem while the controller with no embeddings fails. This makes sense when considering the composition of . No past actions are included in the state, so it is impossible for the controller to determine the causal effects of its actions to understand the environment’s dynamics. Because the controllers are being trained across a distribution of process dynamics, the Markov property only holds if the controllers are given additional information to identify which process (MDP) they are controlling.This information is implicitly given to the DE meta-RL controller through the latent context variable.

While this problem is very simple, it highlights one strength of meta-learning for model-free process control. Meta-learning disentangles the problem of understanding the process dynamics from the problem of developing an optimal control policy. Using a well-trained embedding network, the controller can be directly trained on a low-dimensional representation of the process dynamics. This makes training more efficient and enables simpler state representations which do not have to include all information necessary to understand the process dynamics. The process dynamics do not have to be rediscovered every time step; the latent context variable can be calculated once in a new environment and held constant.

Figure 2: Sample trajectory of DRL controllers on two transfer functions with opposite gains.

4.1.2 First Order Dynamics Example: Generalizability

In this experiment, our controllers are trained across 15 different first order transfer functions (listed in Figure 5). The agent’s performance is then evaluated on the new transfer function . These systems were selected as a simple illustration of the latent context variable embedding system dynamics. The test system is a novel composition of dynamics the agent has already seen; the same gain, time constant, and order, so process dynamics embeddings developed during training are likely to be useful in adapting to the test system.

For this example, the controller with no embeddings has a modified state: . Including previous actions in the state gives this controller enough information to understand the process dynamics and fairly compete with the meta-RL controllers (whose states do not include previous actions so they are forced to use the latent context variable to encode this information). The effect of using a DE versus a PE in the meta-RL controller is also examined. Controller performance across three sample transfer functions they are trained on is shown in Figure 3.

The DE meta-RL controller outperforms both the PE controller and the controller with no embeddings and avoids overshoot when controlling processes with slower dynamics such as .

When comparing the control actions taken in response to the step-changes at the 10 and 20-second marks, it is clear the DE meta-RL controller can distinguish between the and processes, whereas the controller with no embeddings and the PE meta-RL controller’s response to both systems is nearly identical, resulting in sub-optimal performance on the slower dynamics of .

The deterministic context embedding likely has better performance than the probabilistic context embedding because the problem has relatively little stochasticity. The process dynamics are fully observable from the context and the only random feature of the problem is the small amount of Gaussian noise added to the measurements during training. This environment enables the context embedding network to reliably encode the process dynamics accurately, meaning sampling the latent variable from a distribution is unnecessary as the variance would naturally be low. Learning to encode a probability distribution is inherently less sample efficient and harder to train than encoding a deterministic variable. The controller with no embeddings likely performed worse due to the increased difficulty of simultaneously solving for the process dynamics and optimal control policy in the same neural network, making it slower to train or causing it to converge to a sub-optimal control law solution.

Figure 3: Performance of DRL controllers across different process dynamics after training.

4.1.3 First Order Dynamics Example: Adaptability

Next, the adaptability of the controllers to the transfer function is tested. The adaptive performance of the controllers, as well as a DRL controller with no prior training, is shown in Figure 4. The large shaded interquartile regions are mostly due to the variable nature of the environment rather than the variable performance of the controllers. During every episode, each controller is tested on 10 random setpoint changes. A controller tasked with managing a setpoint change from 0.1 to 0.11 is likely to experience a smaller cumulative offset penalty than the exact same controller tasked with managing a setpoint change from 0.1 to 1.0, for example. The 10 random setpoint changes are consistent across every controller for a fair comparison. The DE meta-RL controller was chosen to represent the meta-RL controllers for this experiment due to its superior performance over the PE meta-RL controller in the previous generalizability experiment.

The DE meta-RL controller had the best initial performance of the three controllers before any additional training on the new system. This is desirable for industrial applications as we want effective process control as soon as the controller is installed. Perturbations to a system during adaptive tuning can be costly and, in some cases, even unsafe. Additionally, the DE meta-RL controller is more robust than the controller trained without embeddings as can be seen from the latter’s significant performance dip during adaptive training. All controllers attain a similar asymptotic performance.

Figure 4: Moving 20-episode average of adaptive performance of controllers to new process. The shaded region represents the interquartile range calculated from the controller performance distribution across 10 different tests.

4.1.4 First Order Dynamics Example: Embeddings

The DE meta-RL controller’s latent context variable () is shown in Figure 5. We chose , noting that needs to be kept low-dimensional to create an information bottleneck between the embedding network and the actor-critic network to ensure the problems of understanding a task and developing an optimal control strategy are disentangled. If this bottleneck did not exist, the controller would be functionally the same as a regular DRL controller trained across a distribution of tasks. While only two

dimensions are necessary to give the embeddings the degrees of freedom necessary for communicating the system dynamics in the first order processes examined in this paper (i.e., process gain and time constant), three dimensions are used so that the same models can be applied to more complex processes in future work.

Figure 5 helps describe which aspects of the process dynamics the embedding network is good at identifying and which features the network has trouble differentiating based on the relative distances between different processes. Processes with the same gain are coded with the same color. Processes with the same time constant are coded with the same shade. The most noticeable trend in Figure 5 is that embeddings are most similar between processes with the same gain. The left plot also shows clear separation based on the gain magnitude: gains of are clustered together and gains of have a separate cluster. Within some of the clusters of processes with the same gain, there are slight trends in terms of time constants. Processes with closer time constants tend to be positioned slightly closer together, however this differentiation is much weaker than the differentiation based on process gain.

The embeddings for the transfer function from the adaptability test in Section 4.1.3 are also plotted in Figure 5. Its embedding visualization helps explain why the meta-RL controller was able to adapt to the new process so quickly. The latent context variable passed to actor-critic DRL controller identified the new process as having a gain of based on its clustering on the right-side plot. Additionally, the new process’ embeddings have overlap with the transfer function in the left-side plot. This makes sense as this transfer function has the same time constant and gain magnitude, but this is also interesting in that it breaks from trends established among the other embeddings wherein processes with the same gain are positioned nearest to each other. Based on these embeddings, the actor-critic controller can readily detect that the new process it is controlling has a gain of and a time constant of , and it has already learned how to control processes with these parameters, allowing for quick adaptation to this new parameter combination.

Figure 5: Visualization of the latent context variables from Experiment 4.1.2 and 4.1.3. The asterisk besides indicates the model was not trained on this transfer function and it was used during the adaptability test in Experiment 5.1.3. for each transfer function.

4.2 Learning New Control Objectives

In this experiment, our controllers are trained on the transfer function . The controllers are trained across different control objectives by manipulating the parameters in the RL reward function shown below:


In addition to penalizing setpoint error, the term penalizes jerky control motion to encourage smooth action. The term penalizes large control actions, useful for applications where input to a process may be costly. The term penalizes overshoot, defined as where there is a sign change in setpoint error relative to a reference time-step, , which was chosen as the initial state of the system after a setpoint change (e.g., if the system starts below the setpoint, overshoot is defined as ending up above the setpoint). This is a rather strict definition of overshoot which aims to make the control action critically dampen the system. In future work, the term could be modified so the controller does not incur a penalty as long as the setpoint is not overshot by some buffer which could allow for training underdamped control policies while still penalizing overshoot. Selecting well-suited values for and can be used to develop a control policy optimized for any specific application’s objectives. For this experiment, for the controller with no embeddings and the original state definition from Section 4.1.1 is still used for the meta-RL controller. Previous rewards are added to the state for the controller with no embeddings to have the information necessary to discriminate different tasks (control objectives) from each other.

4.2.1 Different Control Objectives Example: Generalizability

Controllers with DE, PE, and no embeddings are trained across four different control objectives by changing the reward function parameters. The first environment only aims to minimize setpoint tracking error, one has an additional penalty for the change in action, another has an additional penalty on the action magnitude, and the last environment has an additional penalty for overshoot. The adaptive performance of these trained controllers is tested in an environment with penalties for both changes in action and action magnitude. Unlike Example 4.1.2, where the controller’s environment is fully observable from the context, this problem is not fully observable from context; the overshoot penalty cannot be known by the controller until it overshoots the setpoint. For this reason, probabilistic context embeddings are a reasonable choice.

Figure 6 shows the performance of the controllers across the training environments. Consistent with 4.1.2, the multi-task controller tends to learn a single generalized policy for all environments whereas the meta-RL controllers tailor their policy to the specific environment. For example, when not penalized for changes to control action or action magnitude, the meta-RL controllers take large oscillating actions whereas they avoid this behaviour when in an environment penalizing such action. All of the controllers have offset from the setpoint: in future work this offset could be avoided by adding an integral error penalty to the reward function.

This example highlights the importance of incorporating a penalty for changes to control input into the reward function, just as it is if often incorporated into the objective function used in model predictive control. We see the meta-RL controller produces oscillating and erratic control action when not penalized for such action. In preliminary experiments, the same problem was observed with the RL controller with no embeddings as well. In this example, the RL controller with no embeddings does not have this problem because it is penalized for changes to control input in one task and is unable to distinguish between tasks (learns one general policy) so it avoids this action at all times.

The probabilistic meta-RL controller develops a significant offset from the setpoint; this behaviour can be explained by the reward function formulation. In the overshoot environment, the controller learns it is best to keep a distance away from the setpoint related to the variance in the Gaussian measurement noise added to the experiment during training because this noise could result in accidental overshoot. To avoid constantly being penalized for passing the setpoint, it is safer to keep a small distance away from it. The probabilistic meta-RL controller does not learn to distinguish the overshoot environment from the others and applies this buffer between the output and setpoint to every environment. This problem with the reward function formulation could be solved in future work by adding a buffer to the overshoot penalty as previously mentioned.

Figure 6: Performance of DRL controllers with no embeddings, DE, and PE trained across different control objectives on the transfer function .

4.2.2 Different Control Objectives Example: Adaptability

Figure 7 shows the adaptive performance to the testing environment where there is a small penalty for changes in action and action magnitude simultaneously. The PE meta-RL controller was chosen to represent the meta-RL controllers as it had better generalization performance across the different control tasks in terms of its cumulative reward being higher than the DE’s controller.

The PE meta-RL controller and the controller with no embeddings have nearly identical performance and adapt to the task faster than a DRL controller trained from scratch. In future work, the controllers can be trained across a larger distribution of control objectives to see if this gives the controllers with embeddings an edge over conventional DRL controllers. A larger distribution of training data would likely lead to better embeddings which could improve performance.

Figure 7: Moving 20-episode average of adaptive performance of controllers to a new control objective. The shaded region represents the interquartile range calculated from the controller performance distribution across 10 different tests.

5 Conclusion

Meta-RL is a promising idea for adaptive control which could be integrated into existing control structures (PID and MPC tuning) or be used to construct new, entirely neural network-based controllers and allows for controllers to better adapt to new processes with less process-specific data. This work has highlighted two interesting use cases of meta-learning for process control: embedding process dynamics and embedding control objectives into low dimensional variable representations inferred directly from process data. The next steps in making meta-RL practical for process control will be performing larger scale tests across a much greater number and variety of processes to see if more generalizable embeddings can be created. Additionally, future work could explore training the embedding network using a supervised or unsupervised learning approach rather than using the gradient of a DRL controller. This would enable the embeddings to more easily be used for tuning PID or MPC controllers rather than being used as part of a DRL controller.

We also acknowledge this work has introduced additional hyperparameters to RL control. Namely, the number of previous time steps included in the state vector and the number of dimensions in the latent context variable. These hyperparameters have not been rigorously tuned in this paper, and further research into their optimal values is needed.

We gratefully acknowledge the financial support from Natural Sciences and Engineering Research Council of Canada (NSERC) and Honeywell Connected Plant.


  • M. Andrychowicz, M. Denil, S. Gomez, M. W. Hoffman, D. Pfau, T. Schaul, B. Shillingford, and N. De Freitas (2016) Learning to learn by gradient descent by gradient descent. In Advances in neural information processing systems, pp. 3981–3989. Cited by: §2.
  • T. A. Badgwell, J. H. Lee, and K. Liu (2018) Reinforcement learning–overview of recent progress and implications for process control. In Computer Aided Chemical Engineering, Vol. 44, pp. 71–85. Cited by: §1.
  • S. Bengio, Y. Bengio, J. Cloutier, and J. Gecsei (1992) On the optimization of a synaptic learning rule. In Preprints Conf. Optimality in Artificial and Biological Neural Networks, Vol. 2. Cited by: §2.
  • C. Berner, G. Brockman, B. Chan, V. Cheung, P. Dębiak, C. Dennison, D. Farhi, Q. Fischer, S. Hashme, C. Hesse, et al. (2019) Dota 2 with large scale deep reinforcement learning. arXiv preprint arXiv:1912.06680. Cited by: §1.
  • Y. Duan, J. Schulman, X. Chen, P. L. Bartlett, I. Sutskever, and P. Abbeel (2016) : Fast reinforcement learning via slow reinforcement learning. arXiv preprint arXiv:1611.02779. Cited by: §2.
  • C. Finn, P. Abbeel, and S. Levine (2017) Model-agnostic meta-learning for fast adaptation of deep networks. arXiv preprint arXiv:1703.03400. Cited by: §1, §2, §2.
  • S. Fujimoto, H. Van Hoof, and D. Meger (2018) Addressing function approximation error in actor-critic methods. arXiv preprint arXiv:1802.09477. Cited by: §2.
  • T. Haarnoja, A. Zhou, P. Abbeel, and S. Levine (2018) Soft actor-critic: off-policy maximum entropy deep reinforcement learning with a stochastic actor. arXiv preprint arXiv:1801.01290. Cited by: §2.
  • M. Janner, J. Fu, M. Zhang, and S. Levine (2019) When to trust your model: model-based policy optimization. In Advances in Neural Information Processing Systems, pp. 12519–12530. Cited by: §1.
  • V. R. Konda and J. N. Tsitsiklis (2000) Actor-critic algorithms. In Proceedings of the Advances in Neural Information Processing Systems, Denver, USA, pp. 1008–1014. Cited by: §2.
  • T. P. Lillicrap, J. J. Hunt, A. Pritzel, N. Heess, T. Erez, Y. Tassa, D. Silver, and D. Wierstra (2015) Continuous control with deep reinforcement learning. arXiv Preprint, arXiv:1509.02971. Cited by: §2, §2.
  • R. Mendonca, A. Gupta, R. Kralev, P. Abbeel, S. Levine, and C. Finn (2019) Guided meta-policy search. In Advances in Neural Information Processing Systems, pp. 9656–9667. Cited by: §2.
  • V. Mnih, K. Kavukcuoglu, D. Silver, A. A. Rusu, J. Veness, M. G. Bellemare, A. Graves, M. Riedmiller, A. K. Fidjeland, G. Ostrovski, et al. (2015) Human-level control through deep reinforcement learning. Nature 518, pp. 529–533. Cited by: §2.
  • R. Nian, J. Liu, and B. Huang (2020) A review on reinforcement learning: introduction and applications in industrial process control. Computers & Chemical Engineering, pp. 106886. Cited by: §2.
  • P. Petsagkourakis, I. O. Sandoval, E. Bradford, D. Zhang, and E. A. del Rio-Chanona (2020) Reinforcement learning for batch bioprocess optimization. Computers & Chemical Engineering 133, pp. 106649. Cited by: §1.
  • K. Rakelly, A. Zhou, D. Quillen, C. Finn, and S. Levine (2019) Efficient off-policy meta-reinforcement learning via probabilistic context variables. In International conference on machine learning, pp. 5331–5340. Cited by: §2, §3, §3, §3, §3, Algorithm 1.
  • J. Rothfuss, D. Lee, I. Clavera, T. Asfour, and P. Abbeel (2018) ProPM: proximal meta-policy search. arXiv preprint arXiv:1810.06784. Cited by: §2.
  • D. Silver, A. Huang, C. J. Maddison, A. Guez, L. Sifre, G. Van Den Driessche, J. Schrittwieser, I. Antonoglou, V. Panneershelvam, and M. Lanctot (2016) Mastering the game of go with deep neural networks and tree search. Nature 529, pp. 484–489. Cited by: §1.
  • D. Silver, G. Lever, N. Heess, T. Degris, D. Wierstra, and M. Riedmiller (2014) Deterministic policy gradient algorithms. In Proceedings of the 31st International Conference on Machine Learning, Beijing, China. Cited by: §2, §2.
  • S. Spielberg, A. Tulsyan, N. P. Lawrence, P. D. Loewen, and R. B. Gopaluni (2019) Toward self-driving processes: a deep reinforcement learning approach to control. AIChE Journal. External Links: Document Cited by: §2.
  • R. S. Sutton and A. G. Barto (2018) Reinforcement learning: an introduction. MIT press. Cited by: §1, §2, §2, §2.
  • R. S. Sutton, D. A. McAllester, S. P. Singh, and Y. Mansour (2000) Policy gradient methods for reinforcement learning with function approximation. In Proceedings of the Advances in Neural Information Processing Systems, pp. 1057–1063. Cited by: §2.
  • J. X. Wang, Z. Kurth-Nelson, D. Tirumala, H. Soyer, J. Z. Leibo, R. Munos, C. Blundell, D. Kumaran, and M. Botvinick (2016) Learning to reinforcement learn. arXiv preprint arXiv:1611.05763. Cited by: §2.