Differential Good Arm Identification

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Tsai, Yun-Da, Tsai, Tzu-Hsien, Lin, Shou-De
Format: Preprint
Published: 2023
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866929244677865472
author Tsai, Yun-Da
Tsai, Tzu-Hsien
Lin, Shou-De
author_facet Tsai, Yun-Da
Tsai, Tzu-Hsien
Lin, Shou-De
contents This paper targets a variant of the stochastic multi-armed bandit problem called good arm identification (GAI). GAI is a pure-exploration bandit problem with the goal to output as many good arms using as few samples as possible, where a good arm is defined as an arm whose expected reward is greater than a given threshold. In this work, we propose DGAI - a differentiable good arm identification algorithm to improve the sample complexity of the state-of-the-art HDoC algorithm in a data-driven fashion. We also showed that the DGAI can further boost the performance of a general multi-arm bandit (MAB) problem given a threshold as a prior knowledge to the arm set. Extensive experiments confirm that our algorithm outperform the baseline algorithms significantly in both synthetic and real world datasets for both GAI and MAB tasks.
format Preprint
id arxiv_https___arxiv_org_abs_2303_07154
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Differential Good Arm Identification
Tsai, Yun-Da
Tsai, Tzu-Hsien
Lin, Shou-De
Machine Learning
This paper targets a variant of the stochastic multi-armed bandit problem called good arm identification (GAI). GAI is a pure-exploration bandit problem with the goal to output as many good arms using as few samples as possible, where a good arm is defined as an arm whose expected reward is greater than a given threshold. In this work, we propose DGAI - a differentiable good arm identification algorithm to improve the sample complexity of the state-of-the-art HDoC algorithm in a data-driven fashion. We also showed that the DGAI can further boost the performance of a general multi-arm bandit (MAB) problem given a threshold as a prior knowledge to the arm set. Extensive experiments confirm that our algorithm outperform the baseline algorithms significantly in both synthetic and real world datasets for both GAI and MAB tasks.
title Differential Good Arm Identification
topic Machine Learning
url https://arxiv.org/abs/2303.07154