Fixing the Loose Brake: Exponential-Tailed Stopping Time in Best Arm Identification

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Balagopalan, Kapilan, Nguyen, Tuan Ngo, Zhao, Yao, Jun, Kwang-Sung
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915342478999552
author Balagopalan, Kapilan
Nguyen, Tuan Ngo
Zhao, Yao
Jun, Kwang-Sung
author_facet Balagopalan, Kapilan
Nguyen, Tuan Ngo
Zhao, Yao
Jun, Kwang-Sung
contents The best arm identification problem requires identifying the best alternative (i.e., arm) in active experimentation using the smallest number of experiments (i.e., arm pulls), which is crucial for cost-efficient and timely decision-making processes. In the fixed confidence setting, an algorithm must stop data-dependently and return the estimated best arm with a correctness guarantee. Since this stopping time is random, we desire its distribution to have light tails. Unfortunately, many existing studies focus on high probability or in expectation bounds on the stopping time, which allow heavy tails and, for high probability bounds, even not stopping at all. We first prove that this never-stopping event can indeed happen for some popular algorithms. Motivated by this, we propose algorithms that provably enjoy an exponential-tailed stopping time, which improves upon the polynomial tail bound reported by Kalyanakrishnan et al. (2012). The first algorithm is based on a fixed budget algorithm called Sequential Halving along with a doubling trick. The second algorithm is a meta algorithm that takes in any fixed confidence algorithm with a high probability stopping guarantee and turns it into one that enjoys an exponential-tailed stopping time. Our results imply that there is much more to be desired for contemporary fixed confidence algorithms.
format Preprint
id arxiv_https___arxiv_org_abs_2411_01808
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Fixing the Loose Brake: Exponential-Tailed Stopping Time in Best Arm Identification
Balagopalan, Kapilan
Nguyen, Tuan Ngo
Zhao, Yao
Jun, Kwang-Sung
Machine Learning
The best arm identification problem requires identifying the best alternative (i.e., arm) in active experimentation using the smallest number of experiments (i.e., arm pulls), which is crucial for cost-efficient and timely decision-making processes. In the fixed confidence setting, an algorithm must stop data-dependently and return the estimated best arm with a correctness guarantee. Since this stopping time is random, we desire its distribution to have light tails. Unfortunately, many existing studies focus on high probability or in expectation bounds on the stopping time, which allow heavy tails and, for high probability bounds, even not stopping at all. We first prove that this never-stopping event can indeed happen for some popular algorithms. Motivated by this, we propose algorithms that provably enjoy an exponential-tailed stopping time, which improves upon the polynomial tail bound reported by Kalyanakrishnan et al. (2012). The first algorithm is based on a fixed budget algorithm called Sequential Halving along with a doubling trick. The second algorithm is a meta algorithm that takes in any fixed confidence algorithm with a high probability stopping guarantee and turns it into one that enjoys an exponential-tailed stopping time. Our results imply that there is much more to be desired for contemporary fixed confidence algorithms.
title Fixing the Loose Brake: Exponential-Tailed Stopping Time in Best Arm Identification
topic Machine Learning
url https://arxiv.org/abs/2411.01808