Learning from an Exploring Demonstrator: Optimal Reward Estimation for Bandits
Not on the map. Why?
Abstract (arXiv)
We introduce the "inverse bandit" problem of estimating the rewards of a multi-armed bandit instance from observing the learning process of a low-regret demonstrator. Existing approaches to the related problem of inverse reinforcement learning assume the execution of an optimal policy, and thereby suffer from an identifiability issue. In contrast, we propose to leverage the demonstrator's behavior en route to optimality, and in particular, the exploration phase, for reward estimation. We begin by establishing a general information-theoretic lower bound under this paradigm that applies to any demonstrator algorithm, which characterizes a fundamental tradeoff between reward estimation and the amount of exploration of the demonstrator. Then, we develop simple and efficient reward estimators for upper-confidence-based demonstrator algorithms that attain the optimal tradeoff, showing in particular that consistent reward estimation -- free of identifiability issues -- is possible under our paradigm. Extensive simulations on both synthetic and semi-synthetic data corroborate our theoretical results.
Key facts. Cited by 8 papers in the map (data through 7 Oct 2026). Builds on 22 of its 58 references that are in the map. First posted on arXiv 28 Jun 2021.
Builds on
22 of 58 references.
References · 22
- 2016Concrete Problems in AI Safetycited by 1,582
- 2016Generative Adversarial Imitation Learningcited by 1,576
- 2016Hyperband: A Novel Bandit-Based Approach to Hyperparameter Optimizationcited by 688
- 2017Learning Robust Rewards with Adversarial Inverse Reinforcement Learningcited by 485
- 2019Introduction to Multi-Armed Banditscited by 475
- 2014On the Complexity of Best Arm Identification in Multi-Armed Bandit Modelscited by 280
- 2017One-Shot Visual Imitation Learning via Meta-Learningcited by 225
- 2013lil' UCB : An Optimal Exploration Algorithm for Multi-Armed Banditscited by 148
- 2020Reward-rational (implicit) choice: A unifying formalism for reward learningcited by 119
- 2019Imitation Learning from Imperfect Demonstrationcited by 112
- 2018Reinforcement Learning from Imperfect Demonstrationscited by 74
- 2017Why Adaptively Collected Data Have Negative Bias and How to Correct for Itcited by 36
10 more
- 2017Repeated Inverse Reinforcement Learningcited by 29
- 2018Best arm identification in multi-armed bandits with delayed feedbackcited by 29
- 2019Are sample means in multi-armed bandits positively or negatively biased?cited by 24
- 2019The Assistive Multi-Armed Banditcited by 19
- 2019On-Policy Robot Imitation Learning from a Converging Supervisorcited by 10
- 2020On conditional versus marginal bias in multi-armed banditscited by 9
- 2017Bandit Models of Human Behavior: Reward Processing in Mental Disorderscited by 8
- 2020Inverse Reinforcement Learning from a Gradient-based Learnercited by 7
- 2020Bandits with adversarial scalingcited by 5
- 2020Deep PQR: Solving Inverse Reinforcement Learning using Anchor Actionscited by 1
Built on by
Citing papers per year: 2021: 1; 2022: 3; 2023: 3; 2024: 0; 2025: 1.
8 papers.
- 2023Diffusion Models for Black-Box Optimizationcited by 75
- 2022Generative Pretraining for Black-Box Optimizationcited by 36
- 2021Inverse Contextual Bandits: Learning How Behavior Evolves over Timecited by 12
- 2022Reliable Conditioning of Behavioral Cloning for Offline Reinforcement Learningcited by 6
- 2022Integrating Reward Maximization and Population Estimation: Sequential Decision-Making for Internal Revenue Service Audit Selectioncited by 5
- 2025Choosing the Better Bandit Algorithm under Data Sharing: When Do A/B Experiments Work?cited by 4
- 2023Eliciting Risk Aversion with Inverse Reinforcement Learning via Interactive Questioningcited by 2
- 2023MERMAIDE: Learning to Align Learners using Model-Based Meta-Learningcited by 0
- 2025Choosing the Better Bandit Algorithm under Data Sharing: When Do A/B Experiments Work?16 Jul 2025
- 2023Eliciting Risk Aversion with Inverse Reinforcement Learning via Interactive Questioning16 Aug 2023
- 2023Diffusion Models for Black-Box Optimization12 Jun 2023
- 2023MERMAIDE: Learning to Align Learners using Model-Based Meta-Learning10 Apr 2023
- 2022Reliable Conditioning of Behavioral Cloning for Offline Reinforcement Learning11 Oct 2022
- 2022Generative Pretraining for Black-Box Optimization22 Jun 2022
- 2022Integrating Reward Maximization and Population Estimation: Sequential Decision-Making for Internal Revenue Service Audit Selection25 Apr 2022
- 2021Inverse Contextual Bandits: Learning How Behavior Evolves over Time13 Jul 2021
Similar, not cited
Close in wording, no citation either way
- 2021Identifiability in inverse reinforcement learning
- 2023Reward Learning as Doubly Nonparametric Bandits: Optimal Design and Scaling Laws
- 2026Inverse Reinforcement Learning without an Optimal Demonstrator: A Feasible Reward Set Approach
- 2025Reward Compatibility: A Framework for Inverse RL
- 2018A Survey of Inverse Reinforcement Learning: Challenges, Methods and Progress
- 2026Decoding Rewards in Competitive Games: Inverse Game Theory with Entropy Regularization
- 2025Multi-Armed Sampling Problem and the End of Exploration
- 2024The Value of Reward Lookahead in Reinforcement Learning