Active multiple matrix completion with adaptive confidence sets

Andrea Locatelli (1) , Alexandra Carpentier (1) , Michal Valko (2, 3)

1 OVGU - Otto-von-Guericke-Universität Magdeburg = Otto-von-Guericke University [Magdeburg]
2 SEQUEL - Sequential Learning
3 DeepMind [Paris]

In this work, we formulate a new multi-task active learning setting in which the learner's goal is to solve multiple matrix completion problems simultaneously. At each round, the learner can choose from which matrix it receives a sample from an entry drawn uniformly at random. Our main practical motivation is market segmentation, where the matrices represent different regions with different preferences of the customers. The challenge in this setting is that each of the matrices can be of a different size and also of a different rank which is unknown. We provide and analyze a new algorithm, MALocate that is able to adapt to the unknown ranks of the different matrices. We then give a lower-bound showing that our strategy is minimax-optimal and demonstrate its performance with synthetic experiments.

Domaines: Machine Learning [stat.ML]

Andrea Locatelli, Alexandra Carpentier, Michal Valko. Active multiple matrix completion with adaptive confidence sets. International Conference on Artificial Intelligence and Statistics, 2019, Okinawa, Japan. ⟨hal-02387468⟩