使用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。