SpectralUnmix: A Torch-Based Regularized Non-negative Matrix Factorization
Rafael S. de Souza, Paula Coelho, Niranjana P, Ana L. Chies-Santos, Rogério Riffel
Abstract
We present SpectralUnmix, an R package for regularized non-negative matrix factorization (NMF), implemented in torch with optional GPU acceleration. The package estimates low-rank non-negative representations through proximal-gradient updates and allows smoothness regularization along the spectral axis. As a compact demonstration, we apply the method to a subset of stellar spectra and compare the recovered NMF components with principal-component directions and representative stellar spectra. The package is released under the MIT license at \href{https://rafaelsdesouza.github.io/SpectralUnmix/}{this repository}.
