使用GPU的非负矩阵分解（NMF）

Question

目前，我正在使用NMF的Python实现。我正在考虑改进NMF的方法，因为如果你有很多文件，它会变慢。由于NMF使用矩阵乘法，我想可能使用GPU（图形处理单元）。我找到了一个实现NMF on GPUs的解决方案。

问题是：为了加快NMF的性能，使用支持GPU的NMF是一个很好的解决方案吗？或者我应该采取不同的方法？

Answer 1

目前，具有块主旋转功能的交替非负最小二乘法是计算NMF的最快方法。
您可以在此处找到Python的实现：https://github.com/kimjingu/nonnegfac-python

如果你确定GPU实现正在使用最快的方法之一，那就去吧。较慢的方法（例如乘法更新）可能会慢几个数量级，并且可能不值得使用GPU。