ICML

2025

1. ICML

   Clustering Items through Bandit Feedback: Finding the Right Feature out of Many

   Maximilian Graf, Victor Thuot, and Nicolas Verzelen

   In Proceedings of The forty-second International Conference on Machine Learning , Jul 2025

   Abs Bib PDF Poster

   We study the problem of clustering a set of items based on bandit feedback. Each of the n items is characterized by a feature vector, with a possibly large dimension d. The items are partitioned into two unknown groups, such that items within the same group share the same feature vector. We consider a sequential and adaptive setting in which, at each round, the learner selects one item and one feature, then observes a noisy evaluation of the item’s feature. The learner’s objective is to recover the correct partition of the items, while keeping the number of observations as small as possible. We provide an algorithm which relies on finding a relevant feature for the clustering task, leveraging the Sequential Halving algorithm. With probability at least 1 − δ, we obtain an accurate recovery of the partition and derive an upper bound on the budget required. Furthermore, we obtain an instance-dependent lower bound, which is tight in some relevant cases.

   @inproceedings{graf2025clustering,
     title = {{Clustering Items through Bandit Feedback: Finding the Right Feature out of Many}},
     author = {Graf, Maximilian and Thuot, Victor and Verzelen, Nicolas},
     url = {https://openreview.net/forum?id=99zsyZpUqp},
     year = {2025},
     month = jul,
     booktitle = {Proceedings of The forty-second International Conference on Machine Learning},
   }