Transition Constrained Bayesian Optimization via Markov Decision Processes

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Folch, Jose Pablo, Tsay, Calvin, Lee, Robert M, Shafei, Behrang, Ormaniec, Weronika, Krause, Andreas, van der Wilk, Mark, Misener, Ruth, Mutný, Mojmír
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916462710489088
author Folch, Jose Pablo
Tsay, Calvin
Lee, Robert M
Shafei, Behrang
Ormaniec, Weronika
Krause, Andreas
van der Wilk, Mark
Misener, Ruth
Mutný, Mojmír
author_facet Folch, Jose Pablo
Tsay, Calvin
Lee, Robert M
Shafei, Behrang
Ormaniec, Weronika
Krause, Andreas
van der Wilk, Mark
Misener, Ruth
Mutný, Mojmír
contents Bayesian optimization is a methodology to optimize black-box functions. Traditionally, it focuses on the setting where you can arbitrarily query the search space. However, many real-life problems do not offer this flexibility; in particular, the search space of the next query may depend on previous ones. Example challenges arise in the physical sciences in the form of local movement constraints, required monotonicity in certain variables, and transitions influencing the accuracy of measurements. Altogether, such transition constraints necessitate a form of planning. This work extends classical Bayesian optimization via the framework of Markov Decision Processes. We iteratively solve a tractable linearization of our utility function using reinforcement learning to obtain a policy that plans ahead for the entire horizon. This is a parallel to the optimization of an acquisition function in policy space. The resulting policy is potentially history-dependent and non-Markovian. We showcase applications in chemical reactor optimization, informative path planning, machine calibration, and other synthetic examples.
format Preprint
id arxiv_https___arxiv_org_abs_2402_08406
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Transition Constrained Bayesian Optimization via Markov Decision Processes
Folch, Jose Pablo
Tsay, Calvin
Lee, Robert M
Shafei, Behrang
Ormaniec, Weronika
Krause, Andreas
van der Wilk, Mark
Misener, Ruth
Mutný, Mojmír
Machine Learning
Bayesian optimization is a methodology to optimize black-box functions. Traditionally, it focuses on the setting where you can arbitrarily query the search space. However, many real-life problems do not offer this flexibility; in particular, the search space of the next query may depend on previous ones. Example challenges arise in the physical sciences in the form of local movement constraints, required monotonicity in certain variables, and transitions influencing the accuracy of measurements. Altogether, such transition constraints necessitate a form of planning. This work extends classical Bayesian optimization via the framework of Markov Decision Processes. We iteratively solve a tractable linearization of our utility function using reinforcement learning to obtain a policy that plans ahead for the entire horizon. This is a parallel to the optimization of an acquisition function in policy space. The resulting policy is potentially history-dependent and non-Markovian. We showcase applications in chemical reactor optimization, informative path planning, machine calibration, and other synthetic examples.
title Transition Constrained Bayesian Optimization via Markov Decision Processes
topic Machine Learning
url https://arxiv.org/abs/2402.08406