Saved in:
Bibliographic Details
Main Authors: Li, Kevin, Laber, Eric
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2503.10836
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866929759583207424
author Li, Kevin
Laber, Eric
author_facet Li, Kevin
Laber, Eric
contents The contextual bandit framework is widely used to solve sequential optimization problems where the reward of each decision depends on auxiliary context variables. In settings such as medicine, business, and engineering, the decision maker often possesses additional structural information on the generative model that can potentially be used to improve the efficiency of bandit algorithms. We consider settings in which the mean reward is known to be a concave function of the action for each fixed context. Examples include patient-specific dose-response curves in medicine and expected profit in online advertising auctions. We propose a contextual bandit algorithm that accelerates optimization by conditioning the posterior of a Bayesian Gaussian Process model on this concavity information. We design a novel shape-constrained reward function estimator using a specially chosen regression spline basis and constrained Gaussian Process posterior. Using this model, we propose a UCB algorithm and derive corresponding regret bounds. We evaluate our algorithm on numerical examples and test functions used to study optimal dosing of Anti-Clotting medication.
format Preprint
id arxiv_https___arxiv_org_abs_2503_10836
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Exploiting Concavity Information in Gaussian Process Contextual Bandit Optimization
Li, Kevin
Laber, Eric
Machine Learning
The contextual bandit framework is widely used to solve sequential optimization problems where the reward of each decision depends on auxiliary context variables. In settings such as medicine, business, and engineering, the decision maker often possesses additional structural information on the generative model that can potentially be used to improve the efficiency of bandit algorithms. We consider settings in which the mean reward is known to be a concave function of the action for each fixed context. Examples include patient-specific dose-response curves in medicine and expected profit in online advertising auctions. We propose a contextual bandit algorithm that accelerates optimization by conditioning the posterior of a Bayesian Gaussian Process model on this concavity information. We design a novel shape-constrained reward function estimator using a specially chosen regression spline basis and constrained Gaussian Process posterior. Using this model, we propose a UCB algorithm and derive corresponding regret bounds. We evaluate our algorithm on numerical examples and test functions used to study optimal dosing of Anti-Clotting medication.
title Exploiting Concavity Information in Gaussian Process Contextual Bandit Optimization
topic Machine Learning
url https://arxiv.org/abs/2503.10836