Saved in:
Bibliographic Details
Main Authors: Zhu, Hao, Boedecker, Joschka
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2501.18945
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913913204899840
author Zhu, Hao
Boedecker, Joschka
author_facet Zhu, Hao
Boedecker, Joschka
contents We consider the inverse problem of multi-armed bandits (IMAB) that are widely used in neuroscience and psychology research for behavior modelling. We first show that the IMAB problem is not convex in general, but can be relaxed to a convex problem via variable transformation. Based on this result, we propose a two-step sequential heuristic for (approximately) solving the IMAB problem. We discuss a condition where our method provides global solution to the IMAB problem with certificate, as well as approximations to further save computing time. Numerical experiments indicate that our heuristic method is more robust than directly solving the IMAB problem via repeated local optimization, and can achieve the performance of Monte Carlo methods within a significantly decreased running time. We provide the implementation of our method based on CVXPY, which allows straightforward application by users not well versed in convex optimization.
format Preprint
id arxiv_https___arxiv_org_abs_2501_18945
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Solving Inverse Problem for Multi-armed Bandits via Convex Optimization
Zhu, Hao
Boedecker, Joschka
Computational Engineering, Finance, and Science
Machine Learning
Optimization and Control
Neurons and Cognition
We consider the inverse problem of multi-armed bandits (IMAB) that are widely used in neuroscience and psychology research for behavior modelling. We first show that the IMAB problem is not convex in general, but can be relaxed to a convex problem via variable transformation. Based on this result, we propose a two-step sequential heuristic for (approximately) solving the IMAB problem. We discuss a condition where our method provides global solution to the IMAB problem with certificate, as well as approximations to further save computing time. Numerical experiments indicate that our heuristic method is more robust than directly solving the IMAB problem via repeated local optimization, and can achieve the performance of Monte Carlo methods within a significantly decreased running time. We provide the implementation of our method based on CVXPY, which allows straightforward application by users not well versed in convex optimization.
title Solving Inverse Problem for Multi-armed Bandits via Convex Optimization
topic Computational Engineering, Finance, and Science
Machine Learning
Optimization and Control
Neurons and Cognition
url https://arxiv.org/abs/2501.18945