Negative Binomial Matrix Completion
Abstract
Matrix completion focuses on recovering missing or incomplete information in matrices. This problem arises in various applications, including image processing and network analysis. Previous research proposed Poisson matrix completion for count data with noise that follows a Poisson distribution, which assumes that the mean and variance are equal. Since overdispersed count data, whose variance is greater than the mean, is more likely to occur in realistic settings, we assume that the noise follows the negative binomial (NB) distribution, which can be more general than the Poisson distribution. In this paper, we introduce NB matrix completion by proposing a nuclear-norm regularized model that can be solved by proximal gradient descent. In our experiments, we demonstrate that the NB model outperforms Poisson matrix completion in various noise and missing data settings on real data.
Cite
@article{arxiv.2408.16113,
title = {Negative Binomial Matrix Completion},
author = {Yu Lu and Kevin Bui and Roummel F. Marcia},
journal= {arXiv preprint arXiv:2408.16113},
year = {2024}
}
Comments
6 pages, Accepted by the IEEE International Workshop on Machine Learning for Signal Processing (MLSP)