Pure Exploration under Mediators' Feedback
On the map: LLM alignment / RLHF / preference optimisation → Reward modeling and RLHF
Abstract (arXiv)
Stochastic multi-armed bandits are a sequential-decision-making framework, where, at each interaction step, the learner selects an arm and observes a stochastic reward. Within the context of best-arm identification (BAI) problems, the goal of the agent lies in finding the optimal arm, i.e., the one with highest expected reward, as accurately and efficiently as possible. Nevertheless, the sequential interaction protocol of classical BAI problems, where the agent has complete control over the arm being pulled at each round, does not effectively model several decision-making problems of interest (e.g., off-policy learning, partially controllable environments, and human feedback). For this reason, in this work, we propose a novel strict generalization of the classical BAI problem that we refer to as best-arm identification under mediators' feedback (BAI-MF). More specifically, we consider the scenario in which the learner has access to a set of mediators, each of which selects the arms on the agent's behalf according to a stochastic and possibly unknown policy. The mediator, then, communicates back to the agent the pulled arm together with the observed reward. In this setting, the agent's goal lies in sequentially choosing which mediator to query to identify with high probability the optimal arm while minimizing the identification time, i.e., the sample complexity. To this end, we first derive and analyze a statistical lower bound on the sample complexity specific to our general mediator feedback scenario. Then, we propose a sequential decision-making strategy for discovering the best arm under the assumption that the mediators' policies are known to the learner. As our theory verifies, this algorithm matches the lower bound both almost surely and in expectation. Finally, we extend these results to cases where the mediators' policies are unknown to the learner obtaining comparable results.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Placed in Reward modeling and RLHF, one of 10 directions in LLM alignment / RLHF / preference optimisation. Builds on 21 of its 35 references that are in the map. First posted on arXiv 29 Aug 2023.
Builds on
21 of 35 references.
References · 21
- 2020Offline Reinforcement Learning: Tutorial, Review, and Perspectives on Open Problemscited by 1,920
- 2014On the Complexity of Best Arm Identification in Multi-Armed Bandit Modelscited by 286
- 2016Optimal Best Arm Identification with Fixed Confidencecited by 239
- 2017Unifying PAC and Regret: Uniform PAC Bounds for Episodic Reinforcement Learningcited by 176
- 2013lil' UCB : An Optimal Exploration Algorithm for Multi-Armed Banditscited by 149
- 2020Optimal Best-arm Identification in Linear Banditscited by 71
- 2019Pure Exploration with Multiple Correct Answerscited by 64
- 2019Non-Asymptotic Pure Exploration by Solving Gamescited by 63
- 2020Gamification of Pure Exploration for Linear Banditscited by 63
- 2018Best Arm Identification for Contaminated Banditscited by 24
- 2021A/B/n Testing with Control in the Presence of Subpopulationscited by 17
- 2017Thresholding Bandit for Dose-ranging: The Impact of Monotonicitycited by 16
9 more
- 2019Optimal Best Markovian Arm Identification with Fixed Confidencecited by 12
- 2020Best Arm Identification in Spectral Banditscited by 11
- 2022On Elimination Strategies for Bandit Fixed-Confidence Identificationcited by 8
- 2023On the Minimax Regret for Online Learning with Feedback Graphscited by 8
- 2021Mean-based Best Arm Identification in Stochastic Bandits under Reward Contaminationcited by 6
- 2018Contextual Bandits with Stochastic Expertscited by 5
- 2020Policy Optimization as Online Learning with Mediator Feedbackcited by 4
- 2022Online Learning with Off-Policy Feedbackcited by 4
- 2023Learning to Select and Rank from Choice-Based Feedback: A Simple Nested Approachcited by 1
Built on by
Citing papers per year: 2024: 1.
1 paper.
Similar, not cited
Close in wording, no citation either way
- 2025Quantile Multi-Armed Bandits with 1-bit Feedback
- 2023ICQ: A Quantization Scheme for Best-Arm Identification Over Bit-Constrained Channels
- 2024Selective Reviews of Bandit Problems in AI via a Statistical View
- 2025An Exploration-free Method for a Linear Stochastic Bandit Driven by a Linear Gaussian Dynamical System
- 2024Reinforcement Learning from Human Feedback without Reward Inference: Model-Free Algorithm and Instance-Dependent Analysis
- 2025Multi-Armed Sampling Problem and the End of Exploration
- 2021Learning from an Exploring Demonstrator: Optimal Reward Estimation for Bandits
- 2026Pure Exploration for a Good Policy in Reinforcement Learning with Bandit Feedback