English

HELLINGER-UCB: A novel algorithm for stochastic multi-armed bandit problem and cold start problem in recommender system

Machine Learning 2024-04-17 v1 Machine Learning

Abstract

In this paper, we study the stochastic multi-armed bandit problem, where the reward is driven by an unknown random variable. We propose a new variant of the Upper Confidence Bound (UCB) algorithm called Hellinger-UCB, which leverages the squared Hellinger distance to build the upper confidence bound. We prove that the Hellinger-UCB reaches the theoretical lower bound. We also show that the Hellinger-UCB has a solid statistical interpretation. We show that Hellinger-UCB is effective in finite time horizons with numerical experiments between Hellinger-UCB and other variants of the UCB algorithm. As a real-world example, we apply the Hellinger-UCB algorithm to solve the cold-start problem for a content recommender system of a financial app. With reasonable assumption, the Hellinger-UCB algorithm has a convenient but important lower latency feature. The online experiment also illustrates that the Hellinger-UCB outperforms both KL-UCB and UCB1 in the sense of a higher click-through rate (CTR).

Keywords

Cite

@article{arxiv.2404.10207,
  title  = {HELLINGER-UCB: A novel algorithm for stochastic multi-armed bandit problem and cold start problem in recommender system},
  author = {Ruibo Yang and Jiazhou Wang and Andrew Mullhaupt},
  journal= {arXiv preprint arXiv:2404.10207},
  year   = {2024}
}
R2 v1 2026-06-28T15:55:16.666Z