Active Learning with Safety Constraints
Romain Camilleri, Andrew Wagenmaker, Jamie H. Morgenstern, Lalit Jain, Kevin Jamieson
Abstract
Active learning methods have shown great promise in reducing the number of samples necessary for learning. As automated learning systems are adopted into real-time, real-world decision-making pipelines, it is increasingly important that such algorithms are designed with safety in mind. In this work we investigate the complexity of learning the best safe decision in interactive environments. We reduce this problem to a constrained linear bandits problem, where our goal is to find the best arm satisfying certain (unknown) safety constraints. We propose an adaptive experimental design-based algorithm, which we show efficiently trades off between the difficulty of showing an arm is unsafe vs suboptimal. To our knowledge, our results are the first on best-arm identification in linear bandits with safety constraints. In practice, we demonstrate that this approach performs well on synthetic and real world datasets.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers4
- Multitask Learning with No Regret: from Improved Confidence Bounds to Active LearningPier Giuseppe Sessa, Pierre Laforgue, Nicolò Cesa-Bianchi, Andreas KrauseNeurIPS 2023 · 5 citations
- Testing the Feasibility of Linear Programs with Bandit FeedbackAditya Gangrade, Aditya Gopalan, Venkatesh Saligrama, Clayton ScottICML 2024 · 3 citations
- Feasible Action Search for Bandit Linear Programs via Thompson SamplingAditya Gangrade, Aldo Pacchiano, Clayton Scott, Venkatesh SaligramaICML 2025
- Constrained Linear Thompson SamplingAditya Gangrade, Venkatesh SaligramaNeurIPS 2025
Builds on6
- An Empirical Process Approach to the Union Bound: Practical Algorithms for Combinatorial and Linear BanditsJulian Katz-Samuels, Lalit Jain, Zohar S. Karnin, Kevin JamiesonNeurIPS 2020 · 72 citations
- High-dimensional Experimental Design and Kernel BanditsRomain Camilleri, Kevin Jamieson, Julian Katz-SamuelsICML 2021 · 63 citations
- Reward-Free RL is No Harder Than Reward-Aware RL in Linear Markov Decision ProcessesAndrew J. Wagenmaker, Yifang Chen, Max Simchowitz, Simon S. Du et al.ICML 2022 · 61 citations
- Stage-wise Conservative Linear BanditsAhmadreza Moradipari, Christos Thrampoulidis, Mahnoosh AlizadehNeurIPS 2020 · 37 citations
- Improved Algorithms for Agnostic Pool-based Active ClassificationJulian Katz-Samuels, Jifan Zhang, Lalit Jain, Kevin JamiesonICML 2021 · 26 citations
Related papers
- Interactively Learning Preference Constraints in Linear BanditsDavid Lindner, Sebastian Tschiatschek, Katja Hofmann, Andreas KrauseICML 2022 · 15 citations
- Constrained Best Arm Identification with Tests for FeasibilityTing Cai, Kirthevasan KandasamyAAAI 2026
- Robust Pure Exploration in Linear Bandits with Limited BudgetAyya Alieva, Ashok Cutkosky, Abhimanyu DasICML 2021 · 27 citations
- Safe Linear Stochastic BanditsKia Khezeli, Eilyan BitarAAAI 2020 · 31 citations
- Optimal Best-arm Identification in Linear BanditsYassir Jedra, Alexandre ProutièreNeurIPS 2020 · 99 citations
