Modeling Attrition in Recommender Systems with Departing Bandits
Omer Ben-Porat, Lee Cohen, Liu Leqi, Zachary C. Lipton, Yishay Mansour
TL;DR
We introduce Departing Bandits, a multi-armed bandit variant where user horizons depend on the recommender's policy and where latent user types govern both click probabilities and departure risks via $P$ and $\Lambda$. When parameters are known, planning reduces to a POMDP with latent states; in the two-type, two-category setting the optimal policy is threshold-based and computable in constant time by analyzing three structural regimes of $P$. For unknown parameters, we develop a UCB-based framework for sub-exponential returns, achieving $\tilde{O}(\sqrt{T})$ regret in both the single-type and the two-type two-category cases by restricting to a compact policy class of threshold policies. These results enable principled online learning and planning for recommender systems where exploration can shorten user sessions, with potential practical impact on policy design under user churn.
Abstract
Traditionally, when recommender systems are formalized as multi-armed bandits, the policy of the recommender system influences the rewards accrued, but not the length of interaction. However, in real-world systems, dissatisfied users may depart (and never come back). In this work, we propose a novel multi-armed bandit setup that captures such policy-dependent horizons. Our setup consists of a finite set of user types, and multiple arms with Bernoulli payoffs. Each (user type, arm) tuple corresponds to an (unknown) reward probability. Each user's type is initially unknown and can only be inferred through their response to recommendations. Moreover, if a user is dissatisfied with their recommendation, they might depart the system. We first address the case where all users share the same type, demonstrating that a recent UCB-based algorithm is optimal. We then move forward to the more challenging case, where users are divided among two types. While naive approaches cannot handle this setting, we provide an efficient learning algorithm that achieves $\tilde{O}(\sqrt{T})$ regret, where $T$ is the number of users.
