English

LIBRA: Language Model Informed Bandit Recourse Algorithm for Personalized Treatment Planning

Artificial Intelligence 2026-01-21 v1 Machine Learning Statistics Theory Statistics Theory

Abstract

We introduce a unified framework that seamlessly integrates algorithmic recourse, contextual bandits, and large language models (LLMs) to support sequential decision-making in high-stakes settings such as personalized medicine. We first introduce the recourse bandit problem, where a decision-maker must select both a treatment action and a feasible, minimal modification to mutable patient features. To address this problem, we develop the Generalized Linear Recourse Bandit (GLRB) algorithm. Building on this foundation, we propose LIBRA, a Language Model-Informed Bandit Recourse Algorithm that strategically combines domain knowledge from LLMs with the statistical rigor of bandit learning. LIBRA offers three key guarantees: (i) a warm-start guarantee, showing that LIBRA significantly reduces initial regret when LLM recommendations are near-optimal; (ii) an LLM-effort guarantee, proving that the algorithm consults the LLM only O(log2T)O(\log^2 T) times, where TT is the time horizon, ensuring long-term autonomy; and (iii) a robustness guarantee, showing that LIBRA never performs worse than a pure bandit algorithm even when the LLM is unreliable. We further establish matching lower bounds that characterize the fundamental difficulty of the recourse bandit problem and demonstrate the near-optimality of our algorithms. Experiments on synthetic environments and a real hypertension-management case study confirm that GLRB and LIBRA improve regret, treatment quality, and sample efficiency compared with standard contextual bandits and LLM-only benchmarks. Our results highlight the promise of recourse-aware, LLM-assisted bandit algorithms for trustworthy LLM-bandits collaboration in personalized high-stakes decision-making.

Keywords

Cite

@article{arxiv.2601.11905,
  title  = {LIBRA: Language Model Informed Bandit Recourse Algorithm for Personalized Treatment Planning},
  author = {Junyu Cao and Ruijiang Gao and Esmaeil Keyvanshokooh and Jianhao Ma},
  journal= {arXiv preprint arXiv:2601.11905},
  year   = {2026}
}

Comments

50 pages. Previous version with human-AI collaboration: arXiv:2410.14640

R2 v1 2026-07-01T09:08:39.923Z