Extension of NMF to 3D
Are you looking for something like this?

"Introduction to Tensor Decompositions and their
Applications in Machine Learning"
https://arxiv.org/pdf/1711.10781.pdf

I'm not sure if there are better papers for it, but from what I can tell tensor decomposition seems like what you're describing.