Cost-Aware Optimal Pairwise Pure Exploration
Di Wu, Chengshuai Shi, Ruida Zhou, Cong Shen
TL;DR
This work builds a general framework for pure exploration in multi-armed bandits with arm-dependent costs, introducing CAET (Cost-Aware Pairwise Exploration Task) to identify pairwise arm relationships while minimizing cumulative cost. It derives a minimax-style lower bound T^*(c, μ) and shows that CAET achieves asymptotic optimality (up to a tunable factor θ) by tracking an estimated optimal sampling proportion and balancing zero-cost arms through an adaptive α. The method extends naturally to regret minimization via an explore-then-commit scheme and provides practical insights through theoretical results and numerical experiments. Overall, the paper advances cost-aware pure exploration by unifying tasks under a single framework, addressing zero-cost challenges, and delivering provably efficient algorithms with strong asymptotic guarantees.
Abstract
Pure exploration is one of the fundamental problems in multi-armed bandits (MAB). However, existing works mostly focus on specific pure exploration tasks, without a holistic view of the general pure exploration problem. This work fills this gap by introducing a versatile framework to study pure exploration, with a focus on identifying the pairwise relationships between targeted arm pairs. Moreover, unlike existing works that only optimize the stopping time (i.e., sample complexity), this work considers that arms are associated with potentially different costs and targets at optimizing the cumulative cost that occurred during learning. Under the general framework of pairwise pure exploration with arm-specific costs, a performance lower bound is derived. Then, a novel algorithm, termed CAET (Cost-Aware Pairwise Exploration Task), is proposed. CAET builds on the track-and-stop principle with a novel design to handle the arm-specific costs, which can potentially be zero and thus represent a very challenging case. Theoretical analyses prove that the performance of CAET approaches the lower bound asymptotically. Special cases are further discussed, including an extension to regret minimization, which is another major focus of MAB. The effectiveness and efficiency of CAET are also verified through experimental results under various settings.
