Badr-MOUFAD / benchmark_nmf

Benchmark Repository for Nonnegative Matrix Factorization

Benchopt is a package to simplify and make more transparent and reproducible the comparisons of optimization algorithms. This benchmark is dedicated to solver of Nonnegative Matrix Factorization:

$$\min_{W\in \mathbb{R}^{m\times r}_+, H\in \mathbb{R}^{r\times n}_+} f(X, WH)$$

where $m, n$ stand for respectively for the number of rows and columns of the data matrix $X$ which may have negative entries,

$$X \in \mathbb{R}^{m \times n}$$

In short, matrix $X$ is approximated by a low rank matrix $WH$ where each low-rank factor $W$ and $H$ have nonnegative entries, which makes NMF a part-based decomposition.

The rank for the NMF must be provided in the dataset. Several values may be specified, but the responsability of chosing candidate rank values by default does not fall on the solvers, nor the objective.

Install

This benchmark can be run using the following commands:

$ pip install -U benchopt
$ git clone https://github.com/cohenjer/benchmark_nmf
$ benchopt run benchmark_nmf

Apart from the problem, options can be passed to benchopt run, to restrict the benchmarks to some solvers or datasets, e.g.:

$ benchopt run benchmark_nmf -s apg -d simulated --max-runs 10 --n-repetitions 10

Use benchopt run -h for more details about these options, or visit https://benchopt.github.io/api.html.

Todo:

• Use optimal permutations for Factor Match Score metric
• Fix Nimfa early stopping
• Standardise loss naming conventions
• Adding more dataset from various applications
• Improve support for various loss tracking vs loss/update options in solvers